HHH-13619 : test cases
This commit is contained in:
Gail Badner
Steve Ebersole
parent
5bf772c59d
commit
24cedfa6ec
|
|
@ -0,0 +1,342 @@
|
||||||
|
/*
|
||||||
|
* 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.test.hql.size;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import javax.persistence.CascadeType;
|
||||||
|
import javax.persistence.Entity;
|
||||||
|
import javax.persistence.FetchType;
|
||||||
|
import javax.persistence.Id;
|
||||||
|
import javax.persistence.ManyToMany;
|
||||||
|
|
||||||
|
import org.hibernate.Hibernate;
|
||||||
|
|
||||||
|
import org.hibernate.testing.TestForIssue;
|
||||||
|
import org.hibernate.testing.junit4.BaseNonConfigCoreFunctionalTestCase;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import static org.hamcrest.MatcherAssert.assertThat;
|
||||||
|
import static org.hamcrest.core.Is.is;
|
||||||
|
import static org.hibernate.testing.transaction.TransactionUtil.doInHibernate;
|
||||||
|
|
||||||
|
@TestForIssue( jiraKey = "HHH-13619" )
|
||||||
|
public class ManyToManySizeTest extends BaseNonConfigCoreFunctionalTestCase {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpression() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.ManyToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( c.customers ) )" +
|
||||||
|
" from Company c" +
|
||||||
|
" group by c.id, c.name" +
|
||||||
|
" order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 3 ) );
|
||||||
|
final CompanyDto companyDto0 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto0.getId(), is( 0 ) );
|
||||||
|
assertThat( companyDto0.getName(), is( "Company 0") );
|
||||||
|
assertThat( companyDto0.getSizeCustomer(), is( 0 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 2 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpressionWithLeftJoin() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.ManyToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( c.customers ) )" +
|
||||||
|
" from Company c left join c.customers cu" +
|
||||||
|
" group by c.id, c.name" +
|
||||||
|
" order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 3 ) );
|
||||||
|
final CompanyDto companyDto0 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto0.getId(), is( 0 ) );
|
||||||
|
assertThat( companyDto0.getName(), is( "Company 0") );
|
||||||
|
assertThat( companyDto0.getSizeCustomer(), is( 0 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 2 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpressionWithInnerJoin() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.ManyToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( c.customers ) )" +
|
||||||
|
" from Company c inner join c.customers cu" +
|
||||||
|
" group by c.id, c.name" +
|
||||||
|
" order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 2 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpressionOfAliasWithInnerJoin() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.ManyToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( cu ) )" +
|
||||||
|
" from Company c inner join c.customers cu" +
|
||||||
|
" group by c.id, c.name" +
|
||||||
|
" order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 2 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpressionExcludeEmptyCollection() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.ManyToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( c.customers ) )" +
|
||||||
|
" from Company c" +
|
||||||
|
" where c.id != 0" +
|
||||||
|
" group by c.id, c.name order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 2 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsConditionalExpressionExcludeEmptyCollection() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List<Company> results = session.createQuery(
|
||||||
|
"from Company c" +
|
||||||
|
" where size( c.customers ) > 0" +
|
||||||
|
" group by c.id, c.name order by c.id",
|
||||||
|
Company.class
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 2 ) );
|
||||||
|
final Company company1 = results.get( 0 );
|
||||||
|
assertThat( company1.id, is( 1 ) );
|
||||||
|
assertThat( company1.name, is( "Company 1") );
|
||||||
|
assertThat( Hibernate.isInitialized( company1.customers ), is( true ) );
|
||||||
|
assertThat( company1.customers.size(), is( 1 ) );
|
||||||
|
final Company company2 = results.get( 1 );
|
||||||
|
assertThat( company2.id, is( 2 ) );
|
||||||
|
assertThat( company2.name, is( "Company 2") );
|
||||||
|
assertThat( Hibernate.isInitialized( company2.customers ), is( true ) );
|
||||||
|
assertThat( company2.customers.size(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsConditionalExpressionIncludeEmptyCollection() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List<Company> results = session.createQuery(
|
||||||
|
"from Company c" +
|
||||||
|
" where size( c.customers ) > -1" +
|
||||||
|
" group by c.id, c.name order by c.id",
|
||||||
|
Company.class
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 3 ) );
|
||||||
|
final Company company0 = results.get( 0 );
|
||||||
|
assertThat( company0.id, is( 0 ) );
|
||||||
|
assertThat( company0.name, is( "Company 0") );
|
||||||
|
assertThat( Hibernate.isInitialized(company0.customers), is( true ) );
|
||||||
|
assertThat( company0.customers.size(), is( 0 ) );
|
||||||
|
final Company company1 = results.get( 1 );
|
||||||
|
assertThat( company1.id, is( 1 ) );
|
||||||
|
assertThat( company1.name, is( "Company 1") );
|
||||||
|
assertThat( Hibernate.isInitialized(company1.customers), is( true ) );
|
||||||
|
assertThat( company1.customers.size(), is( 1 ) );
|
||||||
|
final Company company2 = results.get( 2 );
|
||||||
|
assertThat( company2.id, is( 2 ) );
|
||||||
|
assertThat( company2.name, is( "Company 2") );
|
||||||
|
assertThat( Hibernate.isInitialized(company2.customers), is( true ) );
|
||||||
|
assertThat( company2.customers.size(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void initDataBase(){
|
||||||
|
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
// Add a company with no customers
|
||||||
|
final Company companyWithNoCustomers = new Company( 0 );
|
||||||
|
companyWithNoCustomers.name = "Company 0";
|
||||||
|
session.persist( companyWithNoCustomers );
|
||||||
|
int k = 0;
|
||||||
|
for ( int i = 1; i <= 2; i++ ) {
|
||||||
|
final Company company = new Company( i );
|
||||||
|
company.name = "Company " + i;
|
||||||
|
|
||||||
|
for ( int j = 1; j <= i; j++ ) {
|
||||||
|
final Customer customer = new Customer( k );
|
||||||
|
customer.name = "Customer " + k;
|
||||||
|
company.customers.add( customer );
|
||||||
|
k++;
|
||||||
|
}
|
||||||
|
session.persist( company );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void cleanupDatabase() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
for ( Company company : session.createQuery( "from Company", Company.class ).list() ) {
|
||||||
|
session.delete( company );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected Class[] getAnnotatedClasses() {
|
||||||
|
return new Class[] { Company.class, Customer.class };
|
||||||
|
}
|
||||||
|
|
||||||
|
@Entity(name ="Company")
|
||||||
|
public static class Company {
|
||||||
|
|
||||||
|
@Id
|
||||||
|
private int id;
|
||||||
|
|
||||||
|
private String name;
|
||||||
|
|
||||||
|
@ManyToMany(cascade = CascadeType.ALL,fetch = FetchType.EAGER)
|
||||||
|
private List<Customer> customers = new ArrayList<>();
|
||||||
|
|
||||||
|
public Company() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public Company(int id) {
|
||||||
|
this.id = id;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Entity(name = "Customer")
|
||||||
|
public static class Customer {
|
||||||
|
|
||||||
|
@Id
|
||||||
|
private int id;
|
||||||
|
|
||||||
|
private String name;
|
||||||
|
|
||||||
|
public Customer() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public Customer(int id) {
|
||||||
|
this.id = id;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class CompanyDto {
|
||||||
|
|
||||||
|
public int id;
|
||||||
|
|
||||||
|
public String name;
|
||||||
|
|
||||||
|
public int sizeCustomer;
|
||||||
|
|
||||||
|
public int getId() {
|
||||||
|
return id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setId(int id) {
|
||||||
|
this.id = id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getName() {
|
||||||
|
return name;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setName(String name) {
|
||||||
|
this.name = name;
|
||||||
|
}
|
||||||
|
|
||||||
|
public int getSizeCustomer() {
|
||||||
|
return sizeCustomer;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setSizeCustomer(int sizeCustomer) {
|
||||||
|
this.sizeCustomer = sizeCustomer;
|
||||||
|
}
|
||||||
|
|
||||||
|
public CompanyDto(){}
|
||||||
|
|
||||||
|
public CompanyDto(int id, String name, int sizeCustomer){
|
||||||
|
this.id = id;
|
||||||
|
this.name = name;
|
||||||
|
this.sizeCustomer = sizeCustomer;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,344 @@
|
||||||
|
/*
|
||||||
|
* 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.test.hql.size;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import javax.persistence.CascadeType;
|
||||||
|
import javax.persistence.Entity;
|
||||||
|
import javax.persistence.FetchType;
|
||||||
|
import javax.persistence.Id;
|
||||||
|
import javax.persistence.JoinColumn;
|
||||||
|
import javax.persistence.OneToMany;
|
||||||
|
|
||||||
|
import org.hibernate.Hibernate;
|
||||||
|
|
||||||
|
import org.hibernate.testing.TestForIssue;
|
||||||
|
import org.hibernate.testing.junit4.BaseNonConfigCoreFunctionalTestCase;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
|
||||||
|
import static org.hamcrest.MatcherAssert.assertThat;
|
||||||
|
import static org.hamcrest.core.Is.is;
|
||||||
|
import static org.hibernate.testing.transaction.TransactionUtil.doInHibernate;
|
||||||
|
|
||||||
|
@TestForIssue( jiraKey = "HHH-13619" )
|
||||||
|
public class OneToManySizeTest extends BaseNonConfigCoreFunctionalTestCase {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpression() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.OneToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( c.customers ) )" +
|
||||||
|
" from Company c" +
|
||||||
|
" group by c.id, c.name" +
|
||||||
|
" order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 3 ) );
|
||||||
|
final CompanyDto companyDto0 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto0.getId(), is( 0 ) );
|
||||||
|
assertThat( companyDto0.getName(), is( "Company 0") );
|
||||||
|
assertThat( companyDto0.getSizeCustomer(), is( 0 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 2 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpressionWithLeftJoin() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.OneToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( c.customers ) )" +
|
||||||
|
" from Company c left join c.customers cu" +
|
||||||
|
" group by c.id, c.name" +
|
||||||
|
" order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 3 ) );
|
||||||
|
final CompanyDto companyDto0 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto0.getId(), is( 0 ) );
|
||||||
|
assertThat( companyDto0.getName(), is( "Company 0") );
|
||||||
|
assertThat( companyDto0.getSizeCustomer(), is( 0 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 2 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpressionWithInnerJoin() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.OneToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( c.customers ) )" +
|
||||||
|
" from Company c inner join c.customers cu" +
|
||||||
|
" group by c.id, c.name" +
|
||||||
|
" order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 2 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpressionOfAliasWithInnerJoin() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.OneToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( cu ) )" +
|
||||||
|
" from Company c inner join c.customers cu" +
|
||||||
|
" group by c.id, c.name" +
|
||||||
|
" order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 2 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsSelectExpressionExcludeEmptyCollection() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List results = session.createQuery(
|
||||||
|
"select new org.hibernate.test.hql.size.OneToManySizeTest$CompanyDto(" +
|
||||||
|
" c.id, c.name, size( c.customers ) )" +
|
||||||
|
" from Company c" +
|
||||||
|
" where c.id != 0" +
|
||||||
|
" group by c.id, c.name order by c.id"
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 2 ) );
|
||||||
|
final CompanyDto companyDto1 = (CompanyDto) results.get( 0 );
|
||||||
|
assertThat( companyDto1.getId(), is( 1 ) );
|
||||||
|
assertThat( companyDto1.getName(), is( "Company 1") );
|
||||||
|
assertThat( companyDto1.getSizeCustomer(), is( 1 ) );
|
||||||
|
final CompanyDto companyDto2 = (CompanyDto) results.get( 1 );
|
||||||
|
assertThat( companyDto2.getId(), is( 2 ) );
|
||||||
|
assertThat( companyDto2.getName(), is( "Company 2") );
|
||||||
|
assertThat( companyDto2.getSizeCustomer(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsConditionalExpressionExcludeEmptyCollection() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List<Company> results = session.createQuery(
|
||||||
|
"from Company c" +
|
||||||
|
" where size( c.customers ) > 0" +
|
||||||
|
" group by c.id, c.name order by c.id",
|
||||||
|
Company.class
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 2 ) );
|
||||||
|
final Company company1 = results.get( 0 );
|
||||||
|
assertThat( company1.id, is( 1 ) );
|
||||||
|
assertThat( company1.name, is( "Company 1") );
|
||||||
|
assertThat( Hibernate.isInitialized( company1.customers ), is( true ) );
|
||||||
|
assertThat( company1.customers.size(), is( 1 ) );
|
||||||
|
final Company company2 = results.get( 1 );
|
||||||
|
assertThat( company2.id, is( 2 ) );
|
||||||
|
assertThat( company2.name, is( "Company 2") );
|
||||||
|
assertThat( Hibernate.isInitialized( company2.customers ), is( true ) );
|
||||||
|
assertThat( company2.customers.size(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSizeAsConditionalExpressionIncludeEmptyCollection() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
final List<Company> results = session.createQuery(
|
||||||
|
"from Company c" +
|
||||||
|
" where size( c.customers ) > -1" +
|
||||||
|
" group by c.id, c.name order by c.id",
|
||||||
|
Company.class
|
||||||
|
).list();
|
||||||
|
assertThat( results.size(), is( 3 ) );
|
||||||
|
final Company company0 = results.get( 0 );
|
||||||
|
assertThat( company0.id, is( 0 ) );
|
||||||
|
assertThat( company0.name, is( "Company 0") );
|
||||||
|
assertThat( Hibernate.isInitialized(company0.customers), is( true ) );
|
||||||
|
assertThat( company0.customers.size(), is( 0 ) );
|
||||||
|
final Company company1 = results.get( 1 );
|
||||||
|
assertThat( company1.id, is( 1 ) );
|
||||||
|
assertThat( company1.name, is( "Company 1") );
|
||||||
|
assertThat( Hibernate.isInitialized(company1.customers), is( true ) );
|
||||||
|
assertThat( company1.customers.size(), is( 1 ) );
|
||||||
|
final Company company2 = results.get( 2 );
|
||||||
|
assertThat( company2.id, is( 2 ) );
|
||||||
|
assertThat( company2.name, is( "Company 2") );
|
||||||
|
assertThat( Hibernate.isInitialized(company2.customers), is( true ) );
|
||||||
|
assertThat( company2.customers.size(), is( 2 ) );
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void initDataBase(){
|
||||||
|
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
// Add a company with no customers
|
||||||
|
final Company companyWithNoCustomers = new Company( 0 );
|
||||||
|
companyWithNoCustomers.name = "Company 0";
|
||||||
|
session.persist( companyWithNoCustomers );
|
||||||
|
int k = 0;
|
||||||
|
for ( int i = 1; i <= 2; i++ ) {
|
||||||
|
final Company company = new Company( i );
|
||||||
|
company.name = "Company " + i;
|
||||||
|
|
||||||
|
for ( int j = 1; j <= i; j++ ) {
|
||||||
|
final Customer customer = new Customer( k );
|
||||||
|
customer.name = "Customer " + k;
|
||||||
|
company.customers.add( customer );
|
||||||
|
k++;
|
||||||
|
}
|
||||||
|
session.persist( company );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void cleanupDatabase() {
|
||||||
|
doInHibernate(
|
||||||
|
this::sessionFactory,
|
||||||
|
session -> {
|
||||||
|
for ( Company company : session.createQuery( "from Company", Company.class ).list() ) {
|
||||||
|
session.delete( company );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected Class[] getAnnotatedClasses() {
|
||||||
|
return new Class[] { Company.class, Customer.class };
|
||||||
|
}
|
||||||
|
|
||||||
|
@Entity(name ="Company")
|
||||||
|
public static class Company {
|
||||||
|
|
||||||
|
@Id
|
||||||
|
private int id;
|
||||||
|
|
||||||
|
private String name;
|
||||||
|
|
||||||
|
@OneToMany(cascade = CascadeType.ALL,fetch = FetchType.EAGER)
|
||||||
|
@JoinColumn
|
||||||
|
private List<Customer> customers = new ArrayList<>();
|
||||||
|
|
||||||
|
public Company() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public Company(int id) {
|
||||||
|
this.id = id;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Entity(name = "Customer")
|
||||||
|
public static class Customer {
|
||||||
|
|
||||||
|
@Id
|
||||||
|
private int id;
|
||||||
|
|
||||||
|
private String name;
|
||||||
|
|
||||||
|
public Customer() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public Customer(int id) {
|
||||||
|
this.id = id;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class CompanyDto {
|
||||||
|
|
||||||
|
public int id;
|
||||||
|
|
||||||
|
public String name;
|
||||||
|
|
||||||
|
public int sizeCustomer;
|
||||||
|
|
||||||
|
public int getId() {
|
||||||
|
return id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setId(int id) {
|
||||||
|
this.id = id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getName() {
|
||||||
|
return name;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setName(String name) {
|
||||||
|
this.name = name;
|
||||||
|
}
|
||||||
|
|
||||||
|
public int getSizeCustomer() {
|
||||||
|
return sizeCustomer;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setSizeCustomer(int sizeCustomer) {
|
||||||
|
this.sizeCustomer = sizeCustomer;
|
||||||
|
}
|
||||||
|
|
||||||
|
public CompanyDto(){}
|
||||||
|
|
||||||
|
public CompanyDto(int id, String name, int sizeCustomer){
|
||||||
|
this.id = id;
|
||||||
|
this.name = name;
|
||||||
|
this.sizeCustomer = sizeCustomer;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue