HHH-18069 - Add test

Signed-off-by: Jan Schatteman <[email protected]>

Author: jrenaat

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

@@ -0,0 +1,152 @@
+/*
+ * 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.query.hql.set;
+
+import java.time.LocalDate;
+
+import org.hibernate.query.Query;
+
+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;
+import org.junit.jupiter.api.Test;
+
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.GenerationType;
+import jakarta.persistence.Id;
+import jakarta.persistence.ManyToOne;
+
+/**
+ * @author Jan Schatteman
+ */
+@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 new CurrentApple(id, bakedPie.id, dir)\n" +
+							"FROM (\n" +
+							"(\n" +
+							"  SELECT\n" +
+							"    id id, bakedPie bakedPie, bakedOn bakedOn,\n" +
+							"    MAX(bakedOn) OVER (PARTITION BY bakedPie.id) mbo,\n" +
+							"    -1 dir\n" +
+							"  FROM Apple c\n" +
+							"  WHERE\n" +
+							"       bakedPie.id IN (1,2,3,4)\n" +
+							"   AND bakedOn <= :now\n" +
+							"\n" +
+							") UNION ALL (\n" +
+							"\n" +
+							"  SELECT\n" +
+							"    id id, bakedPie bakedPie, bakedOn bakedOn,\n" +
+							"    MIN(bakedOn) OVER (PARTITION BY bakedPie.id) mbo,\n" +
+							"    1 dir\n" +
+							"  FROM Apple c\n" +
+							"  WHERE\n" +
+							"       bakedPie.id IN (1,2,3,4)\n" +
+							"   AND bakedOn > :now\n" +
+							")\n" +
+							")\n" +
+							"WHERE bakedOn = mbo\n" +
+							"ORDER BY dir";
+					Query<CurrentApple> query = session.createQuery( q, CurrentApple.class );
+					query.setParameter( "now", LocalDate.now());
+
+					query.getSingleResult();
+				}
+		);
+	}
+
+	public static class CurrentApple {
+		private final int id;
+		private final int pieId;
+		private final int dir;
+
+		public CurrentApple(int id, int pieId, int dir) {
+			this.id = id;
+			this.pieId = pieId;
+			this.dir = dir;
+		}
+		public int getDir() {
+			return dir;
+		}
+		public int getId() {
+			return id;
+		}
+		public int getPieId() {
+			return pieId;
+		}
+	}
+
+	@Entity(name = "Apple")
+	public static class Apple {
+		@Id
+		@GeneratedValue(strategy = GenerationType.IDENTITY)
+		private Integer id;
+		private LocalDate bakedOn;
+		@ManyToOne
+		private Pie bakedPie;
+
+		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;
+		}
+	}
+
+	@Entity(name = "Pie")
+	public static class Pie {
+		@Id
+		@GeneratedValue(strategy = GenerationType.IDENTITY)
+		private Integer id;
+		private String taste;
+
+		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;
+		}
+	}
+
+}