Test illustrating TCK assertions regarding query results with fetches

hibernate-core/src/test/java/org/hibernate/test/jpa/compliance/tck2_2/QueryExecutionTest.java

@@ -0,0 +1,124 @@
+/*
+ * 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.test.jpa.compliance.tck2_2;
+
+import java.util.ArrayList;
+import java.util.List;
+import javax.persistence.CascadeType;
+import javax.persistence.Entity;
+import javax.persistence.Id;
+import javax.persistence.JoinColumn;
+import javax.persistence.ManyToOne;
+import javax.persistence.OneToMany;
+import javax.persistence.Table;
+
+import org.hibernate.boot.MetadataSources;
+import org.hibernate.transform.DistinctRootEntityResultTransformer;
+
+import org.hibernate.testing.junit4.BaseNonConfigCoreFunctionalTestCase;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import org.hamcrest.CoreMatchers;
+
+import static javax.persistence.CascadeType.*;
+import static javax.persistence.CascadeType.PERSIST;
+import static org.hamcrest.MatcherAssert.assertThat;
+
+/**
+ * @author Steve Ebersole
+ */
+public class QueryExecutionTest extends BaseNonConfigCoreFunctionalTestCase {
+	@Test
+	public void testCollectionFetch() {
+		inTransaction(
+				session -> {
+					final List nonDistinctResult = session.createQuery( "select c from Customer c join fetch c.orders" ).list();
+					assertThat( nonDistinctResult.size(), CoreMatchers.is( 2 ) );
+
+					final List distinctResult = session.createQuery( "select distinct c from Customer c join fetch c.orders" ).list();
+					assertThat( distinctResult.size(), CoreMatchers.is( 1 ) );
+
+					final List distinctViaTransformerResult = session.createQuery( "select c from Customer c join fetch c.orders" )
+							.setResultTransformer( DistinctRootEntityResultTransformer.INSTANCE ).list();
+					assertThat( distinctResult.size(), CoreMatchers.is( 1 ) );
+				}
+		);
+	}
+
+	@Entity( name = "Customer" )
+	@Table( name = "customers" )
+	public static class Customer {
+		@Id
+		public Integer id;
+		public String name;
+		@OneToMany( mappedBy = "customer", cascade = {PERSIST, REMOVE} )
+		public List<Order> orders = new ArrayList<>();
+
+		public Customer() {
+		}
+
+		public Customer(Integer id, String name) {
+			this.id = id;
+			this.name = name;
+		}
+	}
+
+	@Entity( name = "Order" )
+	@Table( name = "orders" )
+	public static class Order {
+		@Id
+		public Integer id;
+		@JoinColumn
+		@ManyToOne
+		public Customer customer;
+		public String receivableLocator;
+
+		public Order() {
+		}
+
+		public Order(Integer id, Customer customer, String receivableLocator) {
+			this.id = id;
+			this.customer = customer;
+			this.receivableLocator = receivableLocator;
+
+			customer.orders.add( this );
+		}
+	}
+
+	@Override
+	protected void applyMetadataSources(MetadataSources sources) {
+		super.applyMetadataSources( sources );
+		sources.addAnnotatedClass( Customer.class );
+		sources.addAnnotatedClass( Order.class );
+	}
+
+	@Before
+	public void createData() {
+		inTransaction(
+				session -> {
+					final Customer cust = new Customer( 1, "Acme Corp");
+					final Order order1 = new Order( 1, cust, "123" );
+					final Order order2 = new Order( 2, cust, "456" );
+					session.persist( cust );
+				}
+		);
+	}
+
+	@After
+	public void dropData() {
+		inTransaction(
+				session -> {
+					final Customer customer = session.byId( Customer.class ).load( 1 );
+					if ( customer != null ) {
+						session.delete( customer );
+					}
+				}
+		);
+	}
+}