package testing.simple.mappings; | |
import static junit.framework.Assert.assertEquals; | |
import static junit.framework.Assert.assertNotNull; | |
import java.util.Collection; | |
import javax.persistence.EntityManager; | |
import javax.persistence.EntityManagerFactory; | |
import javax.persistence.Persistence; | |
import junit.framework.Assert; | |
import org.eclipse.persistence.descriptors.ClassDescriptor; | |
import org.eclipse.persistence.dynamic.DynamicClassLoader; | |
import org.eclipse.persistence.dynamic.DynamicEntity; | |
import org.eclipse.persistence.dynamic.DynamicHelper; | |
import org.eclipse.persistence.dynamic.DynamicType; | |
import org.eclipse.persistence.dynamic.DynamicTypeBuilder; | |
import org.eclipse.persistence.exceptions.DatabaseException; | |
import org.eclipse.persistence.jpa.JpaHelper; | |
import org.eclipse.persistence.jpa.dynamic.JPADynamicHelper; | |
import org.eclipse.persistence.jpa.dynamic.JPADynamicTypeBuilder; | |
import org.eclipse.persistence.mappings.DirectToFieldMapping; | |
import org.eclipse.persistence.mappings.OneToManyMapping; | |
import org.junit.After; | |
import org.junit.AfterClass; | |
import org.junit.BeforeClass; | |
import org.junit.Test; | |
public class SimpleTypes_OneToMany { | |
private static EntityManagerFactory emf; | |
@Test | |
public void verifyConfig() throws Exception { | |
DynamicHelper helper = new JPADynamicHelper(emf); | |
ClassDescriptor descriptorA = helper.getSession().getClassDescriptorForAlias("SimpleA"); | |
assertNotNull("No descriptor found for alias='SimpleA'", descriptorA); | |
DynamicType simpleTypeA = helper.getType("SimpleA"); | |
assertNotNull("'SimpleA' EntityType not found", simpleTypeA); | |
assertEquals(descriptorA, simpleTypeA.getDescriptor()); | |
DirectToFieldMapping a_id = (DirectToFieldMapping) descriptorA.getMappingForAttributeName("id"); | |
assertEquals(int.class, a_id.getAttributeClassification()); | |
DirectToFieldMapping a_value1 = (DirectToFieldMapping) descriptorA.getMappingForAttributeName("value1"); | |
assertEquals(String.class, a_value1.getAttributeClassification()); | |
ClassDescriptor descriptorB = helper.getSession().getClassDescriptorForAlias("SimpleB"); | |
assertNotNull("No descriptor found for alias='SimpleB'", descriptorB); | |
DynamicType simpleTypeB = helper.getType("SimpleB"); | |
assertNotNull("'SimpleB' EntityType not found", simpleTypeB); | |
assertEquals(descriptorB, simpleTypeB.getDescriptor()); | |
DirectToFieldMapping b_id = (DirectToFieldMapping) descriptorB.getMappingForAttributeName("id"); | |
assertEquals(int.class, b_id.getAttributeClassification()); | |
DirectToFieldMapping b_value1 = (DirectToFieldMapping) descriptorB.getMappingForAttributeName("value1"); | |
assertEquals(String.class, b_value1.getAttributeClassification()); | |
OneToManyMapping a_b = (OneToManyMapping) descriptorA.getMappingForAttributeName("b"); | |
assertEquals(descriptorB, a_b.getReferenceDescriptor()); | |
} | |
@Test | |
public void createSimpleA() { | |
DynamicHelper helper = new JPADynamicHelper(emf); | |
DynamicType simpleTypeA = helper.getType("SimpleA"); | |
Assert.assertNotNull(simpleTypeA); | |
EntityManager em = emf.createEntityManager(); | |
DynamicEntity simpleInstance = simpleTypeA.newDynamicEntity(); | |
simpleInstance.set("id", 1); | |
simpleInstance.set("value1", "A1"); | |
em.getTransaction().begin(); | |
em.persist(simpleInstance); | |
em.getTransaction().commit(); | |
int simpleCount = ((Number) em.createQuery("SELECT COUNT(s) FROM SimpleA s").getSingleResult()).intValue(); | |
Assert.assertEquals(1, simpleCount); | |
em.close(); | |
} | |
@Test | |
public void createSimpleB() { | |
DynamicHelper helper = new JPADynamicHelper(emf); | |
DynamicType simpleTypeB = helper.getType("SimpleB"); | |
Assert.assertNotNull(simpleTypeB); | |
EntityManager em = emf.createEntityManager(); | |
DynamicEntity simpleInstance = simpleTypeB.newDynamicEntity(); | |
simpleInstance.set("id", 1); | |
simpleInstance.set("value1", "B1"); | |
em.getTransaction().begin(); | |
em.persist(simpleInstance); | |
em.getTransaction().commit(); | |
int simpleCount = ((Number) em.createQuery("SELECT COUNT(s) FROM SimpleB s").getSingleResult()).intValue(); | |
Assert.assertEquals(1, simpleCount); | |
em.close(); | |
} | |
@Test | |
public void createAwithB() { | |
DynamicHelper helper = new JPADynamicHelper(emf); | |
DynamicType simpleTypeA = helper.getType("SimpleA"); | |
Assert.assertNotNull(simpleTypeA); | |
DynamicType simpleTypeB = helper.getType("SimpleB"); | |
Assert.assertNotNull(simpleTypeB); | |
EntityManager em = emf.createEntityManager(); | |
Assert.assertNotNull(JpaHelper.getServerSession(emf).getDescriptorForAlias("SimpleB")); | |
DynamicEntity simpleInstanceB = simpleTypeB.newDynamicEntity(); | |
simpleInstanceB.set("id", 1); | |
simpleInstanceB.set("value1", "B2"); | |
DynamicEntity simpleInstanceA = simpleTypeA.newDynamicEntity(); | |
simpleInstanceA.set("id", 1); | |
simpleInstanceA.set("value1", "A2"); | |
simpleInstanceA.<Collection<DynamicEntity>> get("b").add(simpleInstanceB); | |
simpleInstanceB.set("a", simpleInstanceA); | |
em.getTransaction().begin(); | |
em.persist(simpleInstanceB); | |
em.persist(simpleInstanceA); | |
em.getTransaction().commit(); | |
int simpleCountB = ((Number) em.createQuery("SELECT COUNT(s) FROM SimpleB s").getSingleResult()).intValue(); | |
Assert.assertEquals(1, simpleCountB); | |
int simpleCountA = ((Number) em.createQuery("SELECT COUNT(s) FROM SimpleA s").getSingleResult()).intValue(); | |
Assert.assertEquals(1, simpleCountA); | |
em.close(); | |
} | |
@Test | |
public void removeAwithB_PrivateOwned() { | |
createAwithB(); | |
DynamicHelper helper = new JPADynamicHelper(emf); | |
DynamicType simpleAType = helper.getType("SimpleA"); | |
((OneToManyMapping) simpleAType.getDescriptor().getMappingForAttributeName("b")).setIsPrivateOwned(true); | |
EntityManager em = emf.createEntityManager(); | |
em.getTransaction().begin(); | |
DynamicEntity a = (DynamicEntity) em.find(simpleAType.getJavaClass(), 1); | |
assertNotNull(a); | |
assertEquals(1, a.<Collection> get("b").size()); | |
em.remove(a); | |
// em.remove(a.get("b", List.class).get(0)); | |
em.getTransaction().commit(); | |
} | |
@Test | |
public void createAwithExistingB() { | |
// TODO Assert.fail("Not Yet Implemented"); | |
} | |
@Test | |
public void removeBfromA() { | |
// TODO Assert.fail("Not Yet Implemented"); | |
} | |
@Test | |
public void addAtoB() { | |
// TODO Assert.fail("Not Yet Implemented"); | |
} | |
@BeforeClass | |
public static void setUp() { | |
emf = Persistence.createEntityManagerFactory("empty"); | |
DynamicHelper helper = new JPADynamicHelper(emf); | |
DynamicClassLoader dcl = helper.getDynamicClassLoader(); | |
Class<?> simpleTypeA = dcl.createDynamicClass("model.SimpleA"); | |
DynamicTypeBuilder aTypeBuilder = new JPADynamicTypeBuilder(simpleTypeA, null, "SIMPLE_TYPE_A"); | |
aTypeBuilder.setPrimaryKeyFields("SID"); | |
Class<?> simpleTypeB = dcl.createDynamicClass("model.SimpleB"); | |
DynamicTypeBuilder bTypeBuilder = new JPADynamicTypeBuilder(simpleTypeB, null, "SIMPLE_TYPE_B"); | |
bTypeBuilder.setPrimaryKeyFields("SID"); | |
bTypeBuilder.addDirectMapping("id", int.class, "SID"); | |
bTypeBuilder.addDirectMapping("value1", String.class, "VAL_1"); | |
bTypeBuilder.addOneToOneMapping("a", aTypeBuilder.getType(), "A_FK"); | |
aTypeBuilder.addDirectMapping("id", int.class, "SID"); | |
aTypeBuilder.addDirectMapping("value1", String.class, "VAL_1"); | |
aTypeBuilder.addOneToManyMapping("b", bTypeBuilder.getType(), "A_FK"); | |
helper.addTypes(true, true, aTypeBuilder.getType(), bTypeBuilder.getType()); | |
} | |
@After | |
public void clearDynamicTables() { | |
EntityManager em = emf.createEntityManager(); | |
em.getTransaction().begin(); | |
em.createQuery("DELETE FROM SimpleB").executeUpdate(); | |
em.createQuery("DELETE FROM SimpleA").executeUpdate(); | |
em.getTransaction().commit(); | |
em.close(); | |
} | |
@AfterClass | |
public static void shutdown() { | |
EntityManager em = emf.createEntityManager(); | |
em.getTransaction().begin(); | |
try { | |
em.createNativeQuery("DROP TABLE SIMPLE_TYPE_A CASCADE CONSTRAINTS").executeUpdate(); | |
em.createNativeQuery("DROP TABLE SIMPLE_TYPE_B CASCADE CONSTRAINTS").executeUpdate(); | |
} catch (DatabaseException dbe) { | |
// ignore | |
} | |
em.getTransaction().commit(); | |
em.close(); | |
emf.close(); | |
} | |
} |