HHH-16680 Add test for treating correlated from node

This commit is contained in:
Christian Beikov 2023-06-05 11:57:14 +02:00
parent 98c1267716
commit 46167d2698
1 changed files with 114 additions and 0 deletions

View File

@ -0,0 +1,114 @@
/*
* 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.jpa.criteria;
import java.util.List;
import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
import org.hibernate.testing.orm.junit.JiraKey;
import org.hibernate.testing.orm.junit.Jpa;
import org.junit.jupiter.api.Test;
import jakarta.persistence.CascadeType;
import jakarta.persistence.DiscriminatorValue;
import jakarta.persistence.Entity;
import jakarta.persistence.FetchType;
import jakarta.persistence.Id;
import jakarta.persistence.ManyToOne;
import jakarta.persistence.OneToMany;
import jakarta.persistence.criteria.CriteriaBuilder;
import jakarta.persistence.criteria.CriteriaQuery;
import jakarta.persistence.criteria.Join;
import jakarta.persistence.criteria.Root;
import jakarta.persistence.criteria.Subquery;
@Jpa(
annotatedClasses = {
TreatCorrelatedTest.BaseRoot.class,
TreatCorrelatedTest.DynamicRoot.class,
TreatCorrelatedTest.VersionedRoot.class,
TreatCorrelatedTest.Node.class,
TreatCorrelatedTest.Version.class
}
)
@JiraKey("HHH-16680")
public class TreatCorrelatedTest {
@Test
public void testCriteriaQuery(EntityManagerFactoryScope scope) {
scope.inTransaction(
entityManager -> {
CriteriaBuilder cb = entityManager.getCriteriaBuilder();
CriteriaQuery<BaseRoot> query = cb.createQuery( BaseRoot.class );
Root<BaseRoot> root = query.from( BaseRoot.class );
Subquery<Long> nodesQuery = query.subquery( Long.class );
Root<BaseRoot> subQueryRoot = nodesQuery.from( BaseRoot.class );
nodesQuery.select( subQueryRoot.get( "id" ) );
Root<VersionedRoot> treatedRoot = cb.treat( subQueryRoot, VersionedRoot.class );
Join<VersionedRoot, Version> joinTrees = treatedRoot.join( "versions" );
Join<Version, Node> joinNodes = joinTrees.join( "nodes" );
nodesQuery.where(
cb.equal( joinNodes.get( "version" ).get( "owner" ).get( "id" ), root.get( "id" ) )
);
query.where(
cb.and(
cb.equal( root.get( "type" ), "Dog" ),
cb.exists( nodesQuery )
)
);
query.select( root );
List<BaseRoot> result = entityManager.createQuery( query ).getResultList();
}
);
}
@Entity(name = "BaseRoot")
public static class BaseRoot {
@Id
private Long id;
private String name;
private String type;
}
@Entity
@DiscriminatorValue(value = "1")
public static class DynamicRoot extends BaseRoot {
}
@Entity(name = "Node")
public static class Node {
@Id
private Long id;
@ManyToOne(fetch = FetchType.LAZY)
private Version version;
}
@Entity(name = "Version")
public static class Version {
@Id
private Long id;
private String name;
@ManyToOne(fetch = FetchType.LAZY)
private VersionedRoot owner;
@OneToMany(fetch = FetchType.LAZY, mappedBy = "version", cascade = CascadeType.PERSIST)
private List<Node> nodes;
}
@Entity
@DiscriminatorValue(value = "2")
public static class VersionedRoot extends BaseRoot {
@OneToMany(fetch = FetchType.LAZY, mappedBy = "owner", cascade = CascadeType.ALL, orphanRemoval = true)
private List<Version> versions;
}
}