diff --git a/hibernate-core/src/test/java/org/hibernate/orm/test/entitygraph/EntityGraphAndJoinTest.java b/hibernate-core/src/test/java/org/hibernate/orm/test/entitygraph/EntityGraphAndJoinTest.java new file mode 100644 index 0000000000..74f8c51c93 --- /dev/null +++ b/hibernate-core/src/test/java/org/hibernate/orm/test/entitygraph/EntityGraphAndJoinTest.java @@ -0,0 +1,160 @@ +/* + * Hibernate, Relational Persistence for Idiomatic Java + * + * License: GNU Lesser General Public License (LGPL), version 2.1 or later + * See the lgpl.txt file in the root directory or http://www.gnu.org/licenses/lgpl-2.1.html + */ +package org.hibernate.orm.test.entitygraph; + +import java.util.List; + +import org.hibernate.testing.jdbc.SQLStatementInspector; +import org.hibernate.testing.orm.junit.DomainModel; +import org.hibernate.testing.orm.junit.Jira; +import org.hibernate.testing.orm.junit.SessionFactory; +import org.hibernate.testing.orm.junit.SessionFactoryScope; +import org.junit.jupiter.api.BeforeAll; +import org.junit.jupiter.api.Test; + +import jakarta.persistence.Entity; +import jakarta.persistence.EntityGraph; +import jakarta.persistence.FetchType; +import jakarta.persistence.GeneratedValue; +import jakarta.persistence.Id; +import jakarta.persistence.JoinColumn; +import jakarta.persistence.ManyToOne; +import jakarta.persistence.NamedAttributeNode; +import jakarta.persistence.NamedEntityGraph; +import jakarta.persistence.TypedQuery; +import jakarta.persistence.criteria.CriteriaBuilder; +import jakarta.persistence.criteria.CriteriaQuery; +import jakarta.persistence.criteria.Join; +import jakarta.persistence.criteria.JoinType; +import jakarta.persistence.criteria.Root; + +import static org.assertj.core.api.Assertions.assertThat; +import static org.hibernate.jpa.SpecHints.HINT_SPEC_FETCH_GRAPH; + +/** + * @author Marco Belladelli + */ +@DomainModel( annotatedClasses = { + EntityGraphAndJoinTest.Person.class, + EntityGraphAndJoinTest.Address.class, +} ) +@SessionFactory( useCollectingStatementInspector = true ) +@Jira( "https://hibernate.atlassian.net/browse/HHH-17629" ) +public class EntityGraphAndJoinTest { + @BeforeAll + public void setUp(SessionFactoryScope scope) { + scope.inTransaction( session -> { + final Address a1 = new Address( 1L, "test" ); + final Address a2 = new Address( 2L, "test" ); + session.persist( a1 ); + session.persist( a2 ); + session.persist( new Person( "Marco", a1 ) ); + session.persist( new Person( "Andrea", a2 ) ); + } ); + } + + @Test + public void testHqlJoin(SessionFactoryScope scope) { + executeQuery( scope, false, false ); + } + + @Test + public void testHqlLeftJoin(SessionFactoryScope scope) { + executeQuery( scope, false, true ); + } + + @Test + public void testCriteriaJoin(SessionFactoryScope scope) { + executeQuery( scope, true, false ); + } + + @Test + public void testCriteriaLeftJoin(SessionFactoryScope scope) { + executeQuery( scope, true, true ); + } + + private void executeQuery(SessionFactoryScope scope, boolean criteria, boolean leftJoin) { + final SQLStatementInspector inspector = scope.getCollectingStatementInspector(); + inspector.clear(); + + scope.inTransaction( session -> { + final TypedQuery query; + if ( criteria ) { + final CriteriaBuilder cb = session.getCriteriaBuilder(); + final CriteriaQuery cq = cb.createQuery( Person.class ); + final Root root = cq.from( Person.class ); + final Join join = root.join( "address", leftJoin ? JoinType.LEFT : JoinType.INNER ); + cq.distinct( true ).where( cb.equal( join.get( "description" ), "test" ) ).orderBy( cb.asc( join.get( "id" ) ) ); + query = session.createQuery( cq.distinct( true ) ); + } + else { + query = session.createQuery( String.format( + "select distinct p from Person p %s p.address a where a.description = 'test' order by a.id", + leftJoin ? "left join" : "join" + ), Person.class ); + } + final EntityGraph entityGraph = session.getEntityGraph( "test-graph" ); + final List resultList = query.setHint( HINT_SPEC_FETCH_GRAPH, entityGraph ).getResultList(); + assertThat( resultList ).hasSize( 2 ); + assertThat( resultList.stream().map( p -> p.getAddress().getId() ) ).containsExactly( 1L, 2L ); + inspector.assertExecutedCount( 1 ); + inspector.assertNumberOfOccurrenceInQuery( 0, "join", 1 ); + } ); + } + + @Entity( name = "Address" ) + public static class Address { + @Id + private Long id; + + private String description; + + public Address() { + } + + public Address(Long id, String description) { + this.id = id; + this.description = description; + } + + public Long getId() { + return id; + } + + public String getDescription() { + return description; + } + } + + @Entity( name = "Person" ) + @NamedEntityGraph( name = "test-graph", attributeNodes = { + @NamedAttributeNode( "address" ), + } ) + public static class Person { + @Id + @GeneratedValue + private Integer id; + + private String name; + + @ManyToOne( fetch = FetchType.LAZY ) + @JoinColumn( name = "address_id" ) + private Address address; + + public Person() { + } + + public Person(String name, Address address) { + this.name = name; + this.address = address; + } + + public Address getAddress() { + return address; + } + } +}