orm derived id model tests

1 files changed, 128 insertions, 0 deletions

diff --git a/jpa/tests/org.eclipse.jpt.core.tests/src/org/eclipse/jpt/core/tests/internal/jpa2/context/orm/GenericOrmManyToOneMapping2_0Tests.java b/jpa/tests/org.eclipse.jpt.core.tests/src/org/eclipse/jpt/core/tests/internal/jpa2/context/orm/GenericOrmManyToOneMapping2_0Tests.java
new file mode 100644
index 0000000000..a0bb549284
--- /dev/null
+++ b/jpa/tests/org.eclipse.jpt.core.tests/src/org/eclipse/jpt/core/tests/internal/jpa2/context/orm/GenericOrmManyToOneMapping2_0Tests.java
@@ -0,0 +1,128 @@
+/*******************************************************************************
+ *  Copyright (c) 2009  Oracle. 
+ *  All rights reserved.  This program and the accompanying materials are 
+ *  made available under the terms of the Eclipse Public License v1.0 which 
+ *  accompanies this distribution, and is available at 
+ *  http://www.eclipse.org/legal/epl-v10.html
+ *  
+ *  Contributors: 
+ *  	Oracle - initial API and implementation
+ *******************************************************************************/
+package org.eclipse.jpt.core.tests.internal.jpa2.context.orm;
+
+import java.util.Iterator;
+import org.eclipse.jpt.core.MappingKeys;
+import org.eclipse.jpt.core.context.orm.OrmPersistentAttribute;
+import org.eclipse.jpt.core.context.orm.OrmPersistentType;
+import org.eclipse.jpt.core.jpa2.context.orm.OrmManyToOneMapping2_0;
+import org.eclipse.jpt.core.jpa2.context.orm.OrmOneToOneMapping2_0;
+import org.eclipse.jpt.core.jpa2.resource.orm.XmlDerivedId;
+import org.eclipse.jpt.core.jpa2.resource.orm.XmlEntity;
+import org.eclipse.jpt.core.resource.java.JPA;
+import org.eclipse.jpt.utility.internal.CollectionTools;
+import org.eclipse.jpt.utility.internal.iterators.ArrayIterator;
+
+public class GenericOrmManyToOneMapping2_0Tests
+	extends Generic2_0OrmContextModelTestCase
+{
+	public GenericOrmManyToOneMapping2_0Tests(String name) {
+		super(name);
+	}
+	
+	
+	private void createTestEntityWithDerivedId() throws Exception {
+		createTestType(new DefaultAnnotationWriter() {
+			@Override
+			public Iterator<String> imports() {
+				return new ArrayIterator<String>(JPA.ENTITY, JPA.MANY_TO_ONE, JPA.ID);
+			}
+			@Override
+			public void appendTypeAnnotationTo(StringBuilder sb) {
+				sb.append("@Entity").append(CR);
+			}
+			
+			@Override
+			public void appendIdFieldAnnotationTo(StringBuilder sb) {
+				sb.append(CR);
+				sb.append("    @ManyToOne @Id").append(CR);				
+				sb.append("    private " + TYPE_NAME + " manyToOne;").append(CR);
+				sb.append(CR);				
+			}
+		});
+		OrmPersistentType ormPersistentType = getEntityMappings().addPersistentType(MappingKeys.ENTITY_TYPE_MAPPING_KEY, FULLY_QUALIFIED_TYPE_NAME);
+		for (OrmPersistentAttribute each : CollectionTools.iterable(ormPersistentType.attributes())) {
+			each.makeSpecified();
+		}
+	}
+	
+	public void testUpdateDerivedId() throws Exception {
+		createTestEntityWithDerivedId();
+		OrmPersistentType contextType = getEntityMappings().getPersistentType(FULLY_QUALIFIED_TYPE_NAME);
+		OrmPersistentAttribute contextAttribute = contextType.getAttributeNamed("manyToOne");
+		OrmManyToOneMapping2_0 contextMapping = (OrmManyToOneMapping2_0) contextAttribute.getMapping();
+		XmlEntity resourceEntity = (XmlEntity) contextType.getMapping().getResourceTypeMapping();
+		XmlDerivedId resourceMapping = 
+				(XmlDerivedId) resourceEntity.getAttributes().getManyToOnes().get(0);
+		
+		assertNull(resourceMapping.getId());
+		assertFalse(contextMapping.getDerivedId().getValue());
+		
+		resourceMapping.setId(Boolean.FALSE);
+		assertEquals(Boolean.FALSE, resourceMapping.getId());
+		assertFalse(contextMapping.getDerivedId().getValue());
+		
+		resourceMapping.setId(Boolean.TRUE);
+		assertEquals(Boolean.TRUE, resourceMapping.getId());
+		assertTrue(contextMapping.getDerivedId().getValue());
+		
+		resourceMapping.setId(null);
+		assertNull(resourceMapping.getId());
+		assertFalse(contextMapping.getDerivedId().getValue());
+	}
+	
+	public void testSetDerivedId() throws Exception {
+		createTestEntityWithDerivedId();
+		OrmPersistentType contextType = getEntityMappings().getPersistentType(FULLY_QUALIFIED_TYPE_NAME);
+		OrmPersistentAttribute contextAttribute = contextType.getAttributeNamed("manyToOne");
+		OrmManyToOneMapping2_0 contextMapping = (OrmManyToOneMapping2_0) contextAttribute.getMapping();
+		XmlEntity resourceEntity = (XmlEntity) contextType.getMapping().getResourceTypeMapping();
+		XmlDerivedId resourceMapping = 
+				(XmlDerivedId) resourceEntity.getAttributes().getManyToOnes().get(0);
+		
+		assertNull(resourceMapping.getId());
+		assertFalse(contextMapping.getDerivedId().getValue());
+		
+		contextMapping.getDerivedId().setValue(true);
+		assertEquals(Boolean.TRUE, resourceMapping.getId());
+		assertTrue(contextMapping.getDerivedId().getValue());
+		
+		contextMapping.getDerivedId().setValue(false);
+		assertNull(resourceMapping.getId());
+		assertFalse(contextMapping.getDerivedId().getValue());
+	}
+	
+	public void testMorphMapping() throws Exception {
+		createTestEntityWithDerivedId();
+		OrmPersistentType contextType = getEntityMappings().getPersistentType(FULLY_QUALIFIED_TYPE_NAME);
+		OrmPersistentAttribute contextAttribute = contextType.getAttributeNamed("manyToOne");
+		XmlEntity resourceEntity = (XmlEntity) contextType.getMapping().getResourceTypeMapping();
+		
+		XmlDerivedId resourceMapping = 
+				(XmlDerivedId) resourceEntity.getAttributes().getManyToOnes().get(0);
+		resourceMapping.setId(Boolean.TRUE);
+		assertEquals(Boolean.TRUE, resourceMapping.getId());
+		assertTrue(((OrmManyToOneMapping2_0) contextAttribute.getMapping()).getDerivedId().getValue());
+		
+		contextAttribute.setSpecifiedMappingKey(MappingKeys.ONE_TO_ONE_ATTRIBUTE_MAPPING_KEY);
+		resourceMapping = 
+				(XmlDerivedId) resourceEntity.getAttributes().getOneToOnes().get(0);
+		assertEquals(Boolean.TRUE, resourceMapping.getId());
+		assertTrue(((OrmOneToOneMapping2_0) contextAttribute.getMapping()).getDerivedId().getValue());
+		
+		contextAttribute.setSpecifiedMappingKey(MappingKeys.MANY_TO_ONE_ATTRIBUTE_MAPPING_KEY);
+		resourceMapping = 
+				(XmlDerivedId) resourceEntity.getAttributes().getManyToOnes().get(0);
+		assertEquals(Boolean.TRUE, resourceMapping.getId());
+		assertTrue(((OrmManyToOneMapping2_0) contextAttribute.getMapping()).getDerivedId().getValue());
+	}
+}