Hibernate/hibernate-orm
1
0
Fork 0
mirror of https://github.com/hibernate/hibernate-orm synced 2025-02-26 21:35:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

HHH-18069 - Add test

Browse Source
This commit is contained in:
Andrea Boriero 2024-11-27 11:49:21 +01:00 committed by Andrea Boriero
parent 7ca9d0aeed
commit 9ed11194a2
1 changed files with 96 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -10,7 +10,7 @@
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.JiraKey;
import org.hibernate.testing.orm.junit.RequiresDialectFeature;
import org.hibernate.testing.orm.junit.SessionFactory;
import org.hibernate.testing.orm.junit.SessionFactoryScope;
@ -25,41 +25,112 @@
/**
* @author Jan Schatteman
*/
@DomainModel (
annotatedClasses = { UnionOfPartitionResultsTest.Apple.class, UnionOfPartitionResultsTest.Pie.class }
@DomainModel(
annotatedClasses = {UnionOfPartitionResultsTest.Apple.class, UnionOfPartitionResultsTest.Pie.class}
)
@SessionFactory
@JiraKey( "HHH-18069" )
public class UnionOfPartitionResultsTest {
@Test
@FailureExpected
@RequiresDialectFeature(feature = DialectFeatureChecks.SupportsUnion.class)
public void testSubqueryWithUnion(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
public void testSubquery(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 testUnionQuery(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)
@RequiresDialectFeature(feature = DialectFeatureChecks.SupportPartitionBy.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" +
")" +
"(" +
"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();
query.list();
}
);
}
public static class CurrentApple {
private final int id;
private final int pieId;
@ -70,12 +141,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 +167,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 +202,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;