Bug 475709 - [Tooling] Manage Protocol containment invariance

Change-Id: Ide1aa1f0345c8523561f823cca9c16c43cd2608f
Signed-off-by: Onder GURCAN <onder.gurcan@cea.fr>

1 files changed, 15 insertions, 0 deletions

diff --git a/tests/junit/umlrt/core/org.eclipse.papyrusrt.umlrt.core.tests/src/org/eclipse/papyrusrt/umlrt/core/tests/creation/CreateElementTest.java b/tests/junit/umlrt/core/org.eclipse.papyrusrt.umlrt.core.tests/src/org/eclipse/papyrusrt/umlrt/core/tests/creation/CreateElementTest.java
index c30cacc93..1fb33f920 100644
--- a/tests/junit/umlrt/core/org.eclipse.papyrusrt.umlrt.core.tests/src/org/eclipse/papyrusrt/umlrt/core/tests/creation/CreateElementTest.java
+++ b/tests/junit/umlrt/core/org.eclipse.papyrusrt.umlrt.core.tests/src/org/eclipse/papyrusrt/umlrt/core/tests/creation/CreateElementTest.java
@@ -217,6 +217,21 @@ public class CreateElementTest extends AbstractPapyrusTest {
 		// test not creation of EnumerationLiteral in that package
 		runNotCreationTest(newEnumeration, UMLElementTypes.OPERATION, true);
 	}
+	
+	@Test
+	public void testCreateProtocolInModel() throws Exception {
+		runCreationTest(rootModel, UMLRTElementTypesEnumerator.PROTOCOL, true);
+	}
+	
+	@Test
+	public void testNotCreateProtocolInClassifier() throws Exception {
+		// create a package in the root model
+		CreateElementCommand command = new CreateElementCommand(new CreateElementRequest(rootModel, UMLElementTypes.CLASS));
+		command.execute(null, null);
+		EObject newClassifier = command.getNewElement();
+		// test not creation of EnumerationLiteral in that package
+		runNotCreationTest(newClassifier, UMLRTElementTypesEnumerator.PROTOCOL, true);
+	}
 
 	protected void runNotCreationTest(EObject owner, IHintedType hintedType, boolean canCreate) throws Exception {
 		Assert.assertTrue("Editor should not be dirty before test", !openPapyrusEditor.isDirty());