Hibernate
/
hibernate-orm
mirror of https://github.com/hibernate/hibernate-orm
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

HHH-8285 Adding more tests

This commit is contained in:
Strong Liu 2013-11-01 14:46:39 +01:00
parent 90ee768f45
commit 99ae2a7be5
1 changed files with 119 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

141
hibernate-entitymanager/src/test/java/org/hibernate/jpa/test/graphs/EntityGraphLoadPlanBuilderTest.java
View File

@ -26,6 +26,7 @@ package org.hibernate.jpa.test.graphs;
import java.util.Iterator; import java.util.Iterator;
import java.util.Set; import java.util.Set;
import javax.persistence.ElementCollection;
import javax.persistence.Embeddable; import javax.persistence.Embeddable;
import javax.persistence.Embedded; import javax.persistence.Embedded;
import javax.persistence.Entity; import javax.persistence.Entity;
@ -66,7 +67,15 @@ import org.hibernate.persister.entity.EntityPersister;
public class EntityGraphLoadPlanBuilderTest extends BaseEntityManagerFunctionalTestCase { public class EntityGraphLoadPlanBuilderTest extends BaseEntityManagerFunctionalTestCase {
@Override @Override
protected Class<?>[] getAnnotatedClasses() { protected Class<?>[] getAnnotatedClasses() {
return new Class[] { Cat.class, Person.class, Country.class }; return new Class[] { Cat.class, Person.class, Country.class, Dog.class, ExpressCompany.class };
}
@Entity
public static class Dog {
@Id
String name;
@ElementCollection
Set<String> favorites;
} }
@Entity @Entity
@ -92,7 +101,14 @@ public class EntityGraphLoadPlanBuilderTest extends BaseEntityManagerFunctionalT
public static class Address { public static class Address {
@ManyToOne @ManyToOne
Country country; Country country;
}
@Entity
public static class ExpressCompany {
@Id
String name;
@ElementCollection
Set<Address> shipAddresses;
} }
@Entity @Entity
@ -100,33 +116,34 @@ public class EntityGraphLoadPlanBuilderTest extends BaseEntityManagerFunctionalT
@Id @Id
String name; String name;
} }
/** /**
* EntityGraph(1): * EntityGraph(1):
* *
* Cat * Cat
* *
* LoadPlan: * LoadPlan:
* *
* Cat * Cat
* *
* --------------------- * ---------------------
* *
* EntityGraph(2): * EntityGraph(2):
* *
* Cat * Cat
* owner -- Person * owner -- Person
* *
* LoadPlan: * LoadPlan:
* *
* Cat * Cat
* owner -- Person * owner -- Person
* address --- Address * address --- Address
*/ */
@Test @Test
public void testBasicFetchLoadPlanBuilding() { public void testBasicFetchLoadPlanBuilding() {
EntityManager em = getOrCreateEntityManager(); EntityManager em = getOrCreateEntityManager();
EntityGraph eg = em.createEntityGraph( Cat.class ); EntityGraph eg = em.createEntityGraph( Cat.class );
LoadPlan plan = buildLoadPlan( eg, Mode.FETCH ); LoadPlan plan = buildLoadPlan( eg, Mode.FETCH, Cat.class );
LoadPlanTreePrinter.INSTANCE.logTree( plan, new AliasResolutionContextImpl( sfi() ) ); LoadPlanTreePrinter.INSTANCE.logTree( plan, new AliasResolutionContextImpl( sfi() ) );
QuerySpace rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get( 0 ); QuerySpace rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get( 0 );
assertFalse( assertFalse(
@ -136,7 +153,7 @@ public class EntityGraphLoadPlanBuilderTest extends BaseEntityManagerFunctionalT
// -------------------------------------------------- another a little more complicated case // -------------------------------------------------- another a little more complicated case
eg = em.createEntityGraph( Cat.class ); eg = em.createEntityGraph( Cat.class );
eg.addSubgraph( "owner", Person.class ); eg.addSubgraph( "owner", Person.class );
plan = buildLoadPlan( eg, Mode.FETCH ); plan = buildLoadPlan( eg, Mode.FETCH, Cat.class );
LoadPlanTreePrinter.INSTANCE.logTree( plan, new AliasResolutionContextImpl( sfi() ) ); LoadPlanTreePrinter.INSTANCE.logTree( plan, new AliasResolutionContextImpl( sfi() ) );
rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get( 0 ); rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get( 0 );
Iterator<Join> iterator = rootQuerySpace.getJoins().iterator(); Iterator<Join> iterator = rootQuerySpace.getJoins().iterator();
@ -167,31 +184,31 @@ public class EntityGraphLoadPlanBuilderTest extends BaseEntityManagerFunctionalT
/** /**
* EntityGraph(1): * EntityGraph(1):
* *
* Cat * Cat
* *
* LoadPlan: * LoadPlan:
* *
* Cat * Cat
* *
* --------------------- * ---------------------
* *
* EntityGraph(2): * EntityGraph(2):
* *
* Cat * Cat
* owner -- Person * owner -- Person
* *
* LoadPlan: * LoadPlan:
* *
* Cat * Cat
* owner -- Person * owner -- Person
* address --- Address * address --- Address
* country -- Country * country -- Country
*/ */
@Test @Test
public void testBasicLoadLoadPlanBuilding() { public void testBasicLoadLoadPlanBuilding() {
EntityManager em = getOrCreateEntityManager(); EntityManager em = getOrCreateEntityManager();
EntityGraph eg = em.createEntityGraph( Cat.class ); EntityGraph eg = em.createEntityGraph( Cat.class );
LoadPlan plan = buildLoadPlan( eg, Mode.LOAD ); LoadPlan plan = buildLoadPlan( eg, Mode.LOAD, Cat.class );
LoadPlanTreePrinter.INSTANCE.logTree( plan, new AliasResolutionContextImpl( sfi() ) ); LoadPlanTreePrinter.INSTANCE.logTree( plan, new AliasResolutionContextImpl( sfi() ) );
QuerySpace rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get( 0 ); QuerySpace rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get( 0 );
assertFalse( assertFalse(
@ -201,7 +218,7 @@ public class EntityGraphLoadPlanBuilderTest extends BaseEntityManagerFunctionalT
// -------------------------------------------------- another a little more complicated case // -------------------------------------------------- another a little more complicated case
eg = em.createEntityGraph( Cat.class ); eg = em.createEntityGraph( Cat.class );
eg.addSubgraph( "owner", Person.class ); eg.addSubgraph( "owner", Person.class );
plan = buildLoadPlan( eg, Mode.LOAD ); plan = buildLoadPlan( eg, Mode.LOAD, Cat.class );
LoadPlanTreePrinter.INSTANCE.logTree( plan, new AliasResolutionContextImpl( sfi() ) ); LoadPlanTreePrinter.INSTANCE.logTree( plan, new AliasResolutionContextImpl( sfi() ) );
rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get( 0 ); rootQuerySpace = plan.getQuerySpaces().getRootQuerySpaces().get( 0 );
Iterator<Join> iterator = rootQuerySpace.getJoins().iterator(); Iterator<Join> iterator = rootQuerySpace.getJoins().iterator();
@ -234,11 +251,91 @@ public class EntityGraphLoadPlanBuilderTest extends BaseEntityManagerFunctionalT
em.close(); em.close();
} }
@Test
public void testBasicElementCollections() {
EntityManager em = getOrCreateEntityManager();
EntityGraph eg = em.createEntityGraph( Dog.class );
eg.addAttributeNodes( "favorites" );
LoadPlan loadLoadPlan = buildLoadPlan( eg, Mode.LOAD, Dog.class ); //WTF name!!!
LoadPlanTreePrinter.INSTANCE.logTree( loadLoadPlan, new AliasResolutionContextImpl( sfi() ) );
QuerySpace querySpace = loadLoadPlan.getQuerySpaces().getRootQuerySpaces().iterator().next();
Iterator<Join> iterator = querySpace.getJoins().iterator();
assertTrue( iterator.hasNext() );
Join collectionJoin = iterator.next();
assertEquals( QuerySpace.Disposition.COLLECTION, collectionJoin.getRightHandSide().getDisposition() );
assertFalse( iterator.hasNext() );
//----------------------------------------------------------------
LoadPlan fetchLoadPlan = buildLoadPlan( eg, Mode.FETCH, Dog.class );
LoadPlanTreePrinter.INSTANCE.logTree( fetchLoadPlan, new AliasResolutionContextImpl( sfi() ) );
querySpace = fetchLoadPlan.getQuerySpaces().getRootQuerySpaces().iterator().next();
iterator = querySpace.getJoins().iterator();
assertTrue( iterator.hasNext() );
collectionJoin = iterator.next();
assertEquals( QuerySpace.Disposition.COLLECTION, collectionJoin.getRightHandSide().getDisposition() );
assertFalse( iterator.hasNext() );
em.close();
}
@Test
public void testEmbeddedCollection() {
EntityManager em = getOrCreateEntityManager();
EntityGraph eg = em.createEntityGraph( ExpressCompany.class );
eg.addAttributeNodes( "shipAddresses" );
LoadPlan loadLoadPlan = buildLoadPlan( eg, Mode.LOAD, ExpressCompany.class ); //WTF name!!!
LoadPlanTreePrinter.INSTANCE.logTree( loadLoadPlan, new AliasResolutionContextImpl( sfi() ) );
QuerySpace querySpace = loadLoadPlan.getQuerySpaces().getRootQuerySpaces().iterator().next();
Iterator<Join> iterator = querySpace.getJoins().iterator();
assertTrue( iterator.hasNext() );
Join collectionJoin = iterator.next();
assertEquals( QuerySpace.Disposition.COLLECTION, collectionJoin.getRightHandSide().getDisposition() );
assertFalse( iterator.hasNext() );
iterator = collectionJoin.getRightHandSide().getJoins().iterator();
assertTrue( iterator.hasNext() );
Join collectionElementJoin = iterator.next();
assertFalse( iterator.hasNext() );
assertEquals( QuerySpace.Disposition.COMPOSITE, collectionElementJoin.getRightHandSide().getDisposition() );
iterator = collectionElementJoin.getRightHandSide().getJoins().iterator();
assertTrue( iterator.hasNext() );
Join countryJoin = iterator.next();
assertFalse( iterator.hasNext() );
assertEquals( QuerySpace.Disposition.ENTITY, countryJoin.getRightHandSide().getDisposition() );
//----------------------------------------------------------------
LoadPlan fetchLoadPlan = buildLoadPlan( eg, Mode.FETCH, ExpressCompany.class );
LoadPlanTreePrinter.INSTANCE.logTree( fetchLoadPlan, new AliasResolutionContextImpl( sfi() ) );
querySpace = fetchLoadPlan.getQuerySpaces().getRootQuerySpaces().iterator().next();
iterator = querySpace.getJoins().iterator();
assertTrue( iterator.hasNext() );
collectionJoin = iterator.next();
assertEquals( QuerySpace.Disposition.COLLECTION, collectionJoin.getRightHandSide().getDisposition() );
assertFalse( iterator.hasNext() );
iterator = collectionJoin.getRightHandSide().getJoins().iterator();
assertTrue( iterator.hasNext() );
collectionElementJoin = iterator.next();
assertFalse( iterator.hasNext() );
assertEquals( QuerySpace.Disposition.COMPOSITE, collectionElementJoin.getRightHandSide().getDisposition() );
iterator = collectionElementJoin.getRightHandSide().getJoins().iterator();
assertFalse( iterator.hasNext() );
//----------------------------------------------------------------
em.close();
}
private SessionFactoryImplementor sfi() { private SessionFactoryImplementor sfi() {
return entityManagerFactory().unwrap( SessionFactoryImplementor.class ); return entityManagerFactory().unwrap( SessionFactoryImplementor.class );
} }
private LoadPlan buildLoadPlan(EntityGraph entityGraph, Mode mode) { private LoadPlan buildLoadPlan(EntityGraph entityGraph, Mode mode, Class clazz) {
LoadQueryInfluencers loadQueryInfluencers = new LoadQueryInfluencers( sfi() ); LoadQueryInfluencers loadQueryInfluencers = new LoadQueryInfluencers( sfi() );
if ( Mode.FETCH == mode ) { if ( Mode.FETCH == mode ) {
@ -247,7 +344,7 @@ public class EntityGraphLoadPlanBuilderTest extends BaseEntityManagerFunctionalT
else { else {
loadQueryInfluencers.setLoadGraph( entityGraph ); loadQueryInfluencers.setLoadGraph( entityGraph );
} }
EntityPersister ep = (EntityPersister) sfi().getClassMetadata( Cat.class ); EntityPersister ep = (EntityPersister) sfi().getClassMetadata( clazz );
AbstractLoadPlanBuildingAssociationVisitationStrategy strategy = Mode.FETCH == mode ? new FetchGraphLoadPlanBuildingStrategy( AbstractLoadPlanBuildingAssociationVisitationStrategy strategy = Mode.FETCH == mode ? new FetchGraphLoadPlanBuildingStrategy(
sfi(), loadQueryInfluencers, LockMode.NONE sfi(), loadQueryInfluencers, LockMode.NONE
) : new LoadGraphLoadPlanBuildingStrategy( sfi(), loadQueryInfluencers, LockMode.NONE ); ) : new LoadGraphLoadPlanBuildingStrategy( sfi(), loadQueryInfluencers, LockMode.NONE );