HHH-17116 Add test for issue
This commit is contained in:
Marco Belladelli
parent
e6391322e6
commit
8a89cf74e6
|
|
@ -0,0 +1,171 @@
|
||||||
|
/*
|
||||||
|
* 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.io.Serializable;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import org.hibernate.Session;
|
||||||
|
import org.hibernate.query.criteria.HibernateCriteriaBuilder;
|
||||||
|
import org.hibernate.query.criteria.JpaCriteriaQuery;
|
||||||
|
import org.hibernate.query.criteria.JpaCrossJoin;
|
||||||
|
import org.hibernate.query.criteria.JpaDerivedJoin;
|
||||||
|
import org.hibernate.query.criteria.JpaEntityJoin;
|
||||||
|
import org.hibernate.query.criteria.JpaJoin;
|
||||||
|
import org.hibernate.query.criteria.JpaPath;
|
||||||
|
import org.hibernate.query.criteria.JpaRoot;
|
||||||
|
import org.hibernate.query.criteria.JpaSubQuery;
|
||||||
|
import org.hibernate.query.sqm.tree.SqmJoinType;
|
||||||
|
|
||||||
|
import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
|
||||||
|
import org.hibernate.testing.orm.junit.Jira;
|
||||||
|
import org.hibernate.testing.orm.junit.Jpa;
|
||||||
|
import org.junit.jupiter.api.AfterAll;
|
||||||
|
import org.junit.jupiter.api.BeforeAll;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
import jakarta.persistence.Entity;
|
||||||
|
import jakarta.persistence.Id;
|
||||||
|
import jakarta.persistence.JoinColumn;
|
||||||
|
import jakarta.persistence.OneToOne;
|
||||||
|
import jakarta.persistence.Table;
|
||||||
|
import jakarta.persistence.Tuple;
|
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Marco Belladelli
|
||||||
|
*/
|
||||||
|
@Jpa( annotatedClasses = { CriteriaQualifiedJoinTest.Primary.class, CriteriaQualifiedJoinTest.Secondary.class } )
|
||||||
|
@Jira( "https://hibernate.atlassian.net/browse/HHH-17116" )
|
||||||
|
public class CriteriaQualifiedJoinTest {
|
||||||
|
@BeforeAll
|
||||||
|
public void setUp(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final Secondary secondaryA = new Secondary( 1, "a" );
|
||||||
|
final Secondary secondaryB = new Secondary( 2, "b" );
|
||||||
|
ArrayList<Object> entities = new ArrayList<Object>( List.of(
|
||||||
|
secondaryA,
|
||||||
|
secondaryB,
|
||||||
|
new Primary( 1, secondaryA ),
|
||||||
|
new Primary( 2, secondaryB )
|
||||||
|
) );
|
||||||
|
entities.forEach( entityManager::persist );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@AfterAll
|
||||||
|
public void tearDown(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
entityManager.createQuery( "delete from Primary" ).executeUpdate();
|
||||||
|
entityManager.createQuery( "delete from Secondary" ).executeUpdate();
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testJoinLateral(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final HibernateCriteriaBuilder cb = entityManager.unwrap( Session.class ).getCriteriaBuilder();
|
||||||
|
final JpaCriteriaQuery<Tuple> query = cb.createTupleQuery();
|
||||||
|
final JpaRoot<Primary> root = query.from( Primary.class );
|
||||||
|
final JpaSubQuery<Tuple> subquery = query.subquery( Tuple.class );
|
||||||
|
final JpaRoot<Primary> correlatedRoot = subquery.correlate( root );
|
||||||
|
final JpaJoin<Primary, Secondary> secondaryJoin = correlatedRoot.join( "secondary" );
|
||||||
|
subquery.multiselect( secondaryJoin.get( "name" ).alias( "name" ) );
|
||||||
|
final JpaDerivedJoin<Tuple> joinLateral = root.joinLateral( subquery );
|
||||||
|
final JpaPath<Integer> id = root.get( "id" );
|
||||||
|
final JpaPath<String> name = joinLateral.get( "name" );
|
||||||
|
query.multiselect( id, name ).orderBy( cb.asc( id ), cb.asc( name ) );
|
||||||
|
final List<Tuple> list = entityManager.createQuery( query ).getResultList();
|
||||||
|
assertQueryResults( list, List.of( 1, 2 ), List.of( "a", "b" ) );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testEntityJoin(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final HibernateCriteriaBuilder cb = entityManager.unwrap( Session.class ).getCriteriaBuilder();
|
||||||
|
final JpaCriteriaQuery<Tuple> query = cb.createTupleQuery();
|
||||||
|
final JpaRoot<Primary> root = query.from( Primary.class );
|
||||||
|
final JpaEntityJoin<Secondary> entityJoin = root.join( Secondary.class, SqmJoinType.INNER );
|
||||||
|
final JpaPath<Integer> id = root.get( "id" );
|
||||||
|
entityJoin.on( cb.equal( id, entityJoin.get( "id" ) ) );
|
||||||
|
final JpaPath<String> name = entityJoin.get( "name" );
|
||||||
|
query.multiselect( id, name ).orderBy( cb.asc( id ), cb.asc( name ) );
|
||||||
|
final List<Tuple> list = entityManager.createQuery( query ).getResultList();
|
||||||
|
assertQueryResults( list, List.of( 1, 2 ), List.of( "a", "b" ) );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testCrossJoin(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final HibernateCriteriaBuilder cb = entityManager.unwrap( Session.class ).getCriteriaBuilder();
|
||||||
|
final JpaCriteriaQuery<Tuple> query = cb.createTupleQuery();
|
||||||
|
final JpaRoot<Primary> root = query.from( Primary.class );
|
||||||
|
final JpaCrossJoin<Secondary> crossJoin = root.crossJoin( Secondary.class );
|
||||||
|
final JpaPath<Integer> id = root.get( "id" );
|
||||||
|
final JpaPath<String> name = crossJoin.get( "name" );
|
||||||
|
query.multiselect( id, name ).orderBy( cb.asc( id ), cb.asc( name ) );
|
||||||
|
final List<Tuple> list = entityManager.createQuery( query ).getResultList();
|
||||||
|
assertQueryResults( list, List.of( 1, 1, 2, 2 ), List.of( "a", "b", "a", "b" ) );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
private void assertQueryResults(List<Tuple> resultList, List<Integer> ids, List<String> names) {
|
||||||
|
assertThat( resultList ).hasSize( ids.size() );
|
||||||
|
for ( int i = 0; i < ids.size(); i++ ) {
|
||||||
|
assertThat( resultList.get( i ).get( 0, Integer.class ) ).isEqualTo( ids.get( i ) );
|
||||||
|
assertThat( resultList.get( i ).get( 1, String.class ) ).isEqualTo( names.get( i ) );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Entity( name = "Primary" )
|
||||||
|
@Table( name = "t_primary" )
|
||||||
|
public static class Primary implements Serializable {
|
||||||
|
@Id
|
||||||
|
private int id;
|
||||||
|
|
||||||
|
@OneToOne
|
||||||
|
@JoinColumn( name = "secondary_id" )
|
||||||
|
private Secondary secondary;
|
||||||
|
|
||||||
|
public Primary() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public Primary(int id, Secondary secondary) {
|
||||||
|
this.id = id;
|
||||||
|
this.secondary = secondary;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Secondary getSecondary() {
|
||||||
|
return secondary;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Entity( name = "Secondary" )
|
||||||
|
@Table( name = "t_secondary" )
|
||||||
|
public static class Secondary implements Serializable {
|
||||||
|
@Id
|
||||||
|
private int id;
|
||||||
|
|
||||||
|
private String name;
|
||||||
|
|
||||||
|
public Secondary() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public Secondary(int id, String name) {
|
||||||
|
this.id = id;
|
||||||
|
this.name = name;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getName() {
|
||||||
|
return name;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue