HHH-17769 Add test for issue

This commit is contained in:
Marco Belladelli 2024-02-27 11:54:45 +01:00
parent 0d7a7087ff
commit e80329f962
1 changed files with 173 additions and 0 deletions

View File

@ -0,0 +1,173 @@
/*
* 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.criteria;
import java.util.ArrayList;
import java.util.List;
import org.hibernate.query.criteria.HibernateCriteriaBuilder;
import org.hibernate.query.criteria.JpaCriteriaQuery;
import org.hibernate.query.criteria.JpaCteCriteria;
import org.hibernate.query.criteria.JpaRoot;
import org.hibernate.query.spi.QueryImplementor;
import org.hibernate.testing.orm.junit.DomainModel;
import org.hibernate.testing.orm.junit.Jira;
import org.hibernate.testing.orm.junit.SessionFactory;
import org.hibernate.testing.orm.junit.SessionFactoryScope;
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.GeneratedValue;
import jakarta.persistence.Id;
import jakarta.persistence.Tuple;
import jakarta.persistence.criteria.CriteriaBuilder;
import jakarta.persistence.criteria.CriteriaQuery;
import jakarta.persistence.criteria.Path;
import jakarta.persistence.criteria.Root;
import static org.assertj.core.api.Assertions.assertThat;
/**
* @author Marco Belladelli
*/
@DomainModel( annotatedClasses = CriteriaCteOffsetFetchTest.Product.class )
@SessionFactory
@Jira( "https://hibernate.atlassian.net/browse/HHH-17769" )
public class CriteriaCteOffsetFetchTest {
@BeforeAll
public void setUp(SessionFactoryScope scope) {
scope.inTransaction( session -> {
session.persist( new Product( "product_1", 1d ) );
session.persist( new Product( "product_2", 100d ) );
session.persist( new Product( "product_3", 200d ) );
session.persist( new Product( "product_4", 30d ) );
session.persist( new Product( "product_5", 20d ) );
} );
}
@AfterAll
public void tearDown(SessionFactoryScope scope) {
scope.inTransaction( session -> session.createMutationQuery( "delete from Product" ).executeUpdate() );
}
@Test
public void testMainQueryOffset(SessionFactoryScope scope) {
executeQuery( scope, 1, null, false );
}
@Test
public void testSetFirstResult(SessionFactoryScope scope) {
executeQuery( scope, 1, null, true );
}
@Test
public void testMainQueryFetch(SessionFactoryScope scope) {
executeQuery( scope, null, 2, false );
}
@Test
public void testSetMaxResults(SessionFactoryScope scope) {
executeQuery( scope, null, 2, true );
}
@Test
public void testMainQueryOffsetAndFetch(SessionFactoryScope scope) {
executeQuery( scope, 1, 2, false );
}
@Test
public void testSetFirstAndMaxResults(SessionFactoryScope scope) {
executeQuery( scope, 1, 2, true );
}
private void executeQuery(
SessionFactoryScope scope,
Integer firstResult,
Integer maxResults,
boolean queryOptions) {
scope.inTransaction( session -> {
final HibernateCriteriaBuilder cb = session.getCriteriaBuilder();
final JpaCriteriaQuery<Product> cq = cb.createQuery( Product.class );
final JpaRoot<Product> root = cq.from( Product.class );
final JpaCteCriteria<Tuple> first = cq.with( cteQuery( cb, "id_for_first", 1.2d ) );
final JpaRoot<Tuple> fromFirst = cq.from( first );
final JpaCteCriteria<Tuple> second = cq.with( cteQuery( cb, "id_for_second", 150.0d ) );
final JpaRoot<Tuple> fromSecond = cq.from( second );
cq.select( root ).where( cb.and(
cb.equal( root.get( "id" ), fromFirst.get( "id_for_first" ) ),
cb.equal( root.get( "id" ), fromSecond.get( "id_for_second" ) )
) ).orderBy( cb.asc( root.get( "name" ) ) );
if ( !queryOptions ) {
if ( firstResult != null ) {
cq.offset( firstResult );
}
if ( maxResults != null ) {
cq.fetch( maxResults );
}
}
final QueryImplementor<Product> query = session.createQuery( cq );
if ( queryOptions ) {
if ( firstResult != null ) {
query.setFirstResult( firstResult );
}
if ( maxResults != null ) {
query.setMaxResults( maxResults );
}
}
final List<Product> resultList = query.getResultList();
assertThat( resultList ).hasSize( 2 );
final List<String> names = new ArrayList<>( 3 );
names.add( firstResult == null ? "product_2" : "product_5" );
names.add( "product_4" );
names.sort( String::compareTo );
assertThat( resultList.stream().map( Product::getName ) ).containsExactly( names.toArray( new String[0] ) );
} );
}
private CriteriaQuery<Tuple> cteQuery(CriteriaBuilder cb, String idAlias, double param) {
final CriteriaQuery<Tuple> cq = cb.createTupleQuery();
final Root<Product> root = cq.from( Product.class );
final Path<Double> price = root.get( "price" );
return cq.multiselect( root.get( "id" ).alias( idAlias ) ).where( cb.and(
cb.isNotNull( price ),
param < 100 ? cb.greaterThan( price, param ) : cb.lessThan( price, param )
) );
}
@Entity( name = "Product" )
public static class Product {
@Id
@GeneratedValue
private Long id;
private String name;
private Double price;
public Product() {
}
public Product(String name, Double price) {
this.name = name;
this.price = price;
}
public String getName() {
return name;
}
public Double getPrice() {
return price;
}
}
}