From 7ca9d0aeeda7d507c6d24f08f6cf5b8d14bf40fa Mon Sep 17 00:00:00 2001
From: Jan Schatteman <jschatte@redhat.com>
Date: Mon, 27 May 2024 22:26:18 +0200
Subject: [PATCH] HHH-18069 - Add test

Signed-off-by: Jan Schatteman <jschatte@redhat.com>
---
 .../hql/set/UnionOfPartitionResultsTest.java  | 139 ++++++++++++++++++
 1 file changed, 139 insertions(+)
 create mode 100644 hibernate-core/src/test/java/org/hibernate/orm/test/query/hql/set/UnionOfPartitionResultsTest.java

diff --git a/hibernate-core/src/test/java/org/hibernate/orm/test/query/hql/set/UnionOfPartitionResultsTest.java b/hibernate-core/src/test/java/org/hibernate/orm/test/query/hql/set/UnionOfPartitionResultsTest.java
new file mode 100644
index 0000000000..5894908cf3
--- /dev/null
+++ b/hibernate-core/src/test/java/org/hibernate/orm/test/query/hql/set/UnionOfPartitionResultsTest.java
@@ -0,0 +1,139 @@
+/*
+ * SPDX-License-Identifier: LGPL-2.1-or-later
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+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) " +
+							"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();
+				}
+		);
+	}
+
+	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;
+		}
+	}
+
+}