Added fix + test code.

406128: ModelImpl.newElement() and .deleteElement() do not update
allInstances caches for supertypes
https://bugs.eclipse.org/bugs/show_bug.cgi?id=406128

1 files changed, 35 insertions, 8 deletions

diff --git a/plugins/org.eclipse.m2m.atl.emftvm/src/org/eclipse/m2m/atl/emftvm/impl/ModelImpl.java b/plugins/org.eclipse.m2m.atl.emftvm/src/org/eclipse/m2m/atl/emftvm/impl/ModelImpl.java
index 6411cd07..eaf13d2a 100644
--- a/plugins/org.eclipse.m2m.atl.emftvm/src/org/eclipse/m2m/atl/emftvm/impl/ModelImpl.java
+++ b/plugins/org.eclipse.m2m.atl.emftvm/src/org/eclipse/m2m/atl/emftvm/impl/ModelImpl.java
@@ -304,22 +304,35 @@ public class ModelImpl extends EObjectImpl implements Model {
 	 * {@inheritDoc}
 	 * <!-- end-user-doc -->
 	 */
-	public EObject newElement(EClass type) {
+	public EObject newElement(final EClass type) {
 		final EObject instance = type.getEPackage().getEFactoryInstance().create(type);
 		getResource().getContents().add(instance);
 		assert instance.eResource() == getResource();
+		addElement(instance, type);
+		return instance;
+	}
+
+	/**
+	 * Adds <code>element</code> to the "allInstances" list for the given type and all its supertypes.
+	 * 
+	 * @param element
+	 *            the element to add
+	 * @param type
+	 *            the current type
+	 */
+	private void addElement(final EObject element, final EClass type) {
 		if (allInstancesMap.containsKey(type)) {
-			allInstancesMap.get(type).add(instance);
+			allInstancesMap.get(type).add(element);
+		}
+		for (EClass superType : type.getESuperTypes()) {
+			addElement(element, superType);
 		}
-		return instance;
 	}
 
 	/**
-	 * <!-- begin-user-doc. -->
-	 * {@inheritDoc}
-	 * <!-- end-user-doc -->
+	 * <!-- begin-user-doc. --> {@inheritDoc} <!-- end-user-doc -->
 	 */
-	public void deleteElement(EObject element) {
+	public void deleteElement(final EObject element) {
 		assert getResource() == element.eResource();
 		final EList<EObject> resContents = getResource().getContents();
 		EcoreUtil.delete(element);
@@ -328,10 +341,24 @@ public class ModelImpl extends EObjectImpl implements Model {
 			// adding children to a container removes them from their previous container
 			resContents.add(child); 
 		}
-		final EClass type = element.eClass();
+		deleteElement(element, element.eClass());
+	}
+
+	/**
+	 * Deletes <code>element</code> from the "allInstances" list for the given type and all its supertypes.
+	 * 
+	 * @param element
+	 *            the element to delete
+	 * @param type
+	 *            the current type
+	 */
+	private void deleteElement(final EObject element, final EClass type) {
 		if (allInstancesMap.containsKey(type)) {
 			allInstancesMap.get(type).remove(element);
 		}
+		for (EClass superType : type.getESuperTypes()) {
+			deleteElement(element, superType);
+		}
 	}
 
 	/**