HHH-18069 - Add test

Author: dreab8

hibernate-core/src/test/java/org/hibernate/orm/test/query/hql/set/UnionOfPartitionResultsTest.java

@@ -10,7 +10,6 @@
 
 import org.hibernate.testing.orm.junit.DialectFeatureChecks;
 import org.hibernate.testing.orm.junit.DomainModel;
-import org.hibernate.testing.orm.junit.FailureExpected;
 import org.hibernate.testing.orm.junit.RequiresDialectFeature;
 import org.hibernate.testing.orm.junit.SessionFactory;
 import org.hibernate.testing.orm.junit.SessionFactoryScope;
@@ -25,37 +24,125 @@
 /**
  * @author Jan Schatteman
  */
-@DomainModel (
-		annotatedClasses = { UnionOfPartitionResultsTest.Apple.class, UnionOfPartitionResultsTest.Pie.class }
+@DomainModel(
+		annotatedClasses = {UnionOfPartitionResultsTest.Apple.class, UnionOfPartitionResultsTest.Pie.class}
 )
 @SessionFactory
 public class UnionOfPartitionResultsTest {
 
 	@Test
-	@FailureExpected
 	@RequiresDialectFeature(feature = DialectFeatureChecks.SupportsUnion.class)
 	public void testUnionOfPartitionResults(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session -> {
+					String q = """
+							SELECT id
+							FROM
+							(
+								( SELECT id id, bakedPie bakedPie
+								FROM Apple c
+								)
+								UNION ALL
+								( SELECT id id, bakedPie bakedPie
+								FROM Apple a
+								)
+							)
+							""";
+
+					Query query = session.createQuery( q );
+
+					query.list();
+				}
+		);
+	}
+
+	@Test
+	@RequiresDialectFeature(feature = DialectFeatureChecks.SupportsUnion.class)
+	public void testUnionOfPartitionResults6(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session -> {
+					String q = """
+							SELECT id
+							FROM
+							(
+								SELECT id id, bakedPie bakedPie
+								FROM Apple c
+							)
+							""";
+
+					Query query = session.createQuery( q );
+
+					query.list();
+				}
+		);
+	}
+
+	@Test
+	@RequiresDialectFeature(feature = DialectFeatureChecks.SupportsUnion.class)
+	public void testUnionOfPartitionResults4(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session -> {
+					String q = """
+								( SELECT id id, bakedPie bakedPie
+								FROM Apple c
+								)
+								UNION ALL
+								( SELECT id id, bakedPie bakedPie
+								FROM Apple c
+								)
+							""";
+
+					Query query = session.createQuery( q );
+
+					query.list();
+				}
+		);
+	}
+
+	@Test
+	@RequiresDialectFeature(feature = DialectFeatureChecks.SupportsUnion.class)
+	public void testUnionOfPartitionResults2(SessionFactoryScope scope) {
+		scope.inTransaction(
+				session -> {
+					String q = """
+							SELECT id
+							FROM
+							(
+								( SELECT id id, bakedPie.id bakedPie
+								FROM Apple c
+								)
+								UNION ALL
+								( SELECT id id, bakedPie.id bakedPie
+								FROM Apple c
+								)
+							)
+							""";
+
+					Query query = session.createQuery( q );
+
+					query.list();
+				}
+		);
+	}
+
+	@Test
+	@RequiresDialectFeature(feature = DialectFeatureChecks.SupportsUnion.class)
+	public void testUnionOfPartitionResults3(SessionFactoryScope scope) {
 		scope.inTransaction(
 				session -> {
 					String q =
-							"SELECT new CurrentApple(id, bakedPie.id, dir) " +
-							"FROM (" +
-								"(" +
-									"SELECT id id, bakedPie bakedPie, bakedOn bakedOn, MAX(bakedOn) OVER (PARTITION BY bakedPie.id) mbo, -1 dir " +
-									"FROM Apple c " +
-									"WHERE bakedPie.id IN (1,2,3,4) AND bakedOn <= :now" +
-								") UNION ALL (" +
-									"SELECT id id, bakedPie bakedPie, bakedOn bakedOn, MIN(bakedOn) OVER (PARTITION BY bakedPie.id) mbo, 1 dir " +
-									"FROM Apple c " +
-									"WHERE bakedPie.id IN (1,2,3,4) AND bakedOn > :now" +
-								")" +
-							") " +
-							"WHERE bakedOn = mbo ORDER BY dir";
-
-					Query<CurrentApple> query = session.createQuery( q, CurrentApple.class );
-					query.setParameter( "now", LocalDate.now());
-
-					query.getSingleResult();
+							"SELECT id id, bakedPie bp, bakedOn bakedOn, MAX(bakedOn) OVER (PARTITION BY bakedPie.id) mbo, -1 dir " +
+							"FROM Apple c " +
+							"WHERE bakedPie.id IN (1,2,3,4) AND bakedOn <= :now" +
+							" UNION ALL " +
+							"SELECT id id, bakedPie bp, bakedOn bakedOn, MIN(bakedOn) OVER (PARTITION BY bakedPie.id) mbo, 1 dir " +
+							"FROM Apple c " +
+							"WHERE bakedPie.id IN (1,2,3,4) AND bakedOn > :now";
+
+					Query query = session.createQuery( q );
+					query.setParameter( "now", LocalDate.now() );
+
+					query.list();
 				}
 		);
 	}
@@ -70,12 +157,15 @@ public CurrentApple(int id, int pieId, int dir) {
 			this.pieId = pieId;
 			this.dir = dir;
 		}
+
 		public int getDir() {
 			return dir;
 		}
+
 		public int getId() {
 			return id;
 		}
+
 		public int getPieId() {
 			return pieId;
 		}
@@ -93,20 +183,25 @@ public static class Apple {
 		public Integer getId() {
 			return id;
 		}
+
 		public Apple setId(Integer id) {
 			this.id = id;
 			return this;
 		}
+
 		public LocalDate getBakedOn() {
 			return bakedOn;
 		}
+
 		public Apple setBakedOn(LocalDate bakedOn) {
 			this.bakedOn = bakedOn;
 			return this;
 		}
+
 		public Pie getBakedPie() {
 			return bakedPie;
 		}
+
 		public Apple setBakedPie(Pie bakedPie) {
 			this.bakedPie = bakedPie;
 			return this;
@@ -123,13 +218,16 @@ public static class Pie {
 		public Integer getId() {
 			return id;
 		}
+
 		public Pie setId(Integer id) {
 			this.id = id;
 			return this;
 		}
+
 		public String getTaste() {
 			return taste;
 		}
+
 		public Pie setTaste(String taste) {
 			this.taste = taste;
 			return this;