Bug 528787 - [Sequence Diagram] Undo a message creation also delete Exec

specification

 - JUnit test that reproduce the issue.

Change-Id: I9b49fac7008368a0f66faa57ae59ec4408d2a208
Signed-off-by: Mickael ADAM <mickael.adam@all4tec.net>

2 files changed, 190 insertions, 1 deletions

diff --git a/tests/junit/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence.tests/src/org/eclipse/papyrus/uml/diagram/sequence/tests/bug/m7/BugTest_m7.java b/tests/junit/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence.tests/src/org/eclipse/papyrus/uml/diagram/sequence/tests/bug/m7/BugTest_m7.java
index 58bdd543031..1fd50617828 100644
--- a/tests/junit/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence.tests/src/org/eclipse/papyrus/uml/diagram/sequence/tests/bug/m7/BugTest_m7.java
+++ b/tests/junit/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence.tests/src/org/eclipse/papyrus/uml/diagram/sequence/tests/bug/m7/BugTest_m7.java
@@ -31,7 +31,8 @@ import org.junit.runners.Suite.SuiteClasses;
 		TestMakeSameHeightForLifelines_402978.class,
 		TestMoveMessageLostFound_403138.class,
 		TestMessageOccurrenceSpecification_477463.class,
-		TestWeakReferences.class
+		TestWeakReferences.class,
+		Test_528787.class
 })
 public class BugTest_m7 {
 }
diff --git a/tests/junit/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence.tests/src/org/eclipse/papyrus/uml/diagram/sequence/tests/bug/m7/Test_528787.java b/tests/junit/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence.tests/src/org/eclipse/papyrus/uml/diagram/sequence/tests/bug/m7/Test_528787.java
new file mode 100644
index 00000000000..d049874d544
--- /dev/null
+++ b/tests/junit/plugins/uml/diagram/org.eclipse.papyrus.uml.diagram.sequence.tests/src/org/eclipse/papyrus/uml/diagram/sequence/tests/bug/m7/Test_528787.java
@@ -0,0 +1,188 @@
+/*****************************************************************************
+ * Copyright (c) 2017 CEA LIST and others.
+ *
+ * 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:
+ *   Mickaël ADAM (ALL4TEC) - mickael.adam@all4tec.net - Initial API and implementation
+ *
+ *****************************************************************************/
+
+package org.eclipse.papyrus.uml.diagram.sequence.tests.bug.m7;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+import org.eclipse.draw2d.geometry.Dimension;
+import org.eclipse.draw2d.geometry.Point;
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.gef.EditPart;
+import org.eclipse.gef.RequestConstants;
+import org.eclipse.gef.commands.Command;
+import org.eclipse.gef.commands.UnexecutableCommand;
+import org.eclipse.gmf.runtime.diagram.ui.editparts.GraphicalEditPart;
+import org.eclipse.gmf.runtime.diagram.ui.editparts.IGraphicalEditPart;
+import org.eclipse.gmf.runtime.diagram.ui.requests.CreateConnectionViewRequest;
+import org.eclipse.gmf.runtime.diagram.ui.requests.CreateViewRequest;
+import org.eclipse.gmf.runtime.diagram.ui.requests.CreateViewRequestFactory;
+import org.eclipse.gmf.runtime.emf.type.core.IElementType;
+import org.eclipse.papyrus.commands.ICreationCommand;
+import org.eclipse.papyrus.uml.diagram.sequence.edit.parts.ActionExecutionSpecificationEditPart;
+import org.eclipse.papyrus.uml.diagram.sequence.edit.parts.LifelineEditPart;
+import org.eclipse.papyrus.uml.diagram.sequence.edit.parts.MessageAsyncEditPart;
+import org.eclipse.papyrus.uml.diagram.sequence.providers.UMLElementTypes;
+import org.eclipse.papyrus.uml.diagram.sequence.tests.ISequenceDiagramTestsConstants;
+import org.eclipse.papyrus.uml.diagram.sequence.tests.canonical.CreateSequenceDiagramCommand;
+import org.eclipse.papyrus.uml.diagram.sequence.tests.canonical.TestLink;
+import org.eclipse.uml2.uml.ActionExecutionSpecification;
+import org.eclipse.uml2.uml.Interaction;
+import org.junit.Before;
+import org.junit.Test;
+
+/**
+ * Test the regression bug 528787:
+ * 
+ * Use case:
+ * <li>create 2 Lifline
+ * <li>create an action execution specification in lifeline2
+ * <li>create above it an AsyncMessage between LL1 and LL2
+ * <li>do an undo
+ * 
+ * <br>
+ * => action execution specification have been deleted from model, the exec spec node still exist and have now a arrow decorator.
+ * <br>
+ * =>Expected: action execution specification don't have to change.
+ */
+@SuppressWarnings({ "unchecked", "rawtypes" })
+public class Test_528787 extends TestLink {
+
+	/** Execution specification semantic element */
+	private EObject executionSpecification;
+
+	/** the initial container of the container */
+	private EObject initialExecSpecContainer;
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	protected ICreationCommand getDiagramCommandCreation() {
+		return new CreateSequenceDiagramCommand();
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	protected String getProjectName() {
+		return ISequenceDiagramTestsConstants.PROJECT_NAME;
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	protected String getFileName() {
+		return ISequenceDiagramTestsConstants.FILE_NAME;
+	}
+
+	/**
+	 * Initialize model. Create the usecase.
+	 */
+	@Before
+	public void initializeModel() {
+		// Create 2 lifeline
+		createNode(UMLElementTypes.Lifeline_Shape, getRootEditPart(), new Point(100, 100), new Dimension(62, 600));
+		createNode(UMLElementTypes.Lifeline_Shape, getRootEditPart(), new Point(200, 100), new Dimension(62, 600));
+		waitForComplete();
+
+		// Get lifeline editpart
+		assertEquals("Root editpart children size is not correct", 2, getRootEditPart().getChildren().size());
+		assertTrue("Root editpart child 0 must be LifelineEditPart", getRootEditPart().getChildren().get(0) instanceof LifelineEditPart);
+		LifelineEditPart lifeline1EditPart = (LifelineEditPart) getRootEditPart().getChildren().get(0);
+		assertTrue("Root editpart child 1  must be LifelineEditPart", getRootEditPart().getChildren().get(1) instanceof LifelineEditPart);
+		LifelineEditPart lifeline2EditPart = (LifelineEditPart) getRootEditPart().getChildren().get(1);
+		waitForComplete();
+
+		// Create execution specification
+		createNode(UMLElementTypes.ActionExecutionSpecification_Shape, lifeline2EditPart, new Point(getAbsoluteCenter(lifeline2EditPart).x, 200), null);
+		waitForComplete();
+		assertTrue("lifeline2 child 1 must be must be ActionExecutionSpecificationEditPart", lifeline2EditPart.getChildren().get(1) instanceof ActionExecutionSpecificationEditPart);
+
+		// get semantic element
+		executionSpecification = (EObject) ((GraphicalEditPart) lifeline2EditPart.getChildren().get(1)).resolveSemanticElement();
+		assertTrue("semantic element must bu instance of ActionExecutionSpecification", executionSpecification instanceof ActionExecutionSpecification);
+
+		// execution specification initial container
+		initialExecSpecContainer = executionSpecification.eContainer();
+		assertTrue("container semantic element must bu instance of Model", initialExecSpecContainer instanceof Interaction);
+
+		// Create message
+		createLink(UMLElementTypes.Message_AsynchEdge, lifeline1EditPart, lifeline2EditPart, new Point(getAbsoluteCenter(lifeline1EditPart).x, 100), new Point(getAbsoluteCenter(lifeline2EditPart).x, 100));
+		waitForComplete();
+		// test the link
+		assertEquals("Lifeline1 source connection size is not correct", 1, lifeline1EditPart.getSourceConnections().size());
+		assertEquals("Lifeline1 target connection size is not correct", 1, lifeline2EditPart.getTargetConnections().size());
+		assertTrue("message1 must be MessageAsyncEditPart", lifeline1EditPart.getSourceConnections().get(0) instanceof MessageAsyncEditPart);
+	}
+
+
+	/**
+	 * Test to create severals messages and execution specification and delete the first one. Weak reference have to move up from a delta.
+	 */
+	@Test
+	public void testExecutionChangeAfterMessageCreationUndo() {
+		// Undo
+		getEMFCommandStack().undo();
+		waitForComplete();
+		assertTrue("executionSpecification eContainer must not changed after undo", executionSpecification.eContainer().equals(initialExecSpecContainer));
+	}
+
+	private CreateConnectionViewRequest createConnectionViewRequest(final IElementType type, final EditPart source, final EditPart target, final Point sourcePoint, final Point targetPoint) {
+		CreateConnectionViewRequest connectionRequest = CreateViewRequestFactory.getCreateConnectionRequest(type, ((IGraphicalEditPart) getDiagramEditPart()).getDiagramPreferencesHint());
+		connectionRequest.setLocation(sourcePoint);
+
+		connectionRequest.setSourceEditPart(null);
+		connectionRequest.setTargetEditPart(source);
+		connectionRequest.setType(RequestConstants.REQ_CONNECTION_START);
+		source.getCommand(connectionRequest);
+		// Now, setup the request in preparation to get the connection end command.
+		connectionRequest.setSourceEditPart(source);
+		connectionRequest.setTargetEditPart(target);
+		connectionRequest.setType(RequestConstants.REQ_CONNECTION_END);
+		connectionRequest.setLocation(targetPoint);
+
+		EObject container = getRootEditPart().resolveSemanticElement();
+		connectionRequest.getExtendedData().put(SOURCE_MODEL_CONTAINER, container);
+		connectionRequest.getExtendedData().put(TARGET_MODEL_CONTAINER, container);
+
+		return connectionRequest;
+	}
+
+	private void createLink(final IElementType linkType, final EditPart source, final EditPart target, final Point sourcePoint, final Point targetPoint) {
+		CreateConnectionViewRequest req = createConnectionViewRequest(linkType, source, target, sourcePoint, targetPoint);
+		Command command = target.getCommand(req);
+		assertNotNull(CREATION + COMMAND_NULL, command);
+		assertTrue(CONTAINER_CREATION + TEST_IF_THE_COMMAND_CAN_BE_EXECUTED, command.canExecute());
+
+		getDiagramCommandStack().execute(command);
+	}
+
+	private void createNode(final IElementType type, final EditPart parentPart, final Point location, final Dimension size) {
+		// CREATION
+		CreateViewRequest requestcreation = CreateViewRequestFactory.getCreateShapeRequest(type, getRootEditPart().getDiagramPreferencesHint());
+		requestcreation.setLocation(location);
+		requestcreation.setSize(size);
+		Command command = parentPart.getCommand(requestcreation);
+		assertNotNull(CREATION + COMMAND_NULL, command);
+		assertTrue(CREATION + TEST_IF_THE_COMMAND_IS_CREATED, command != UnexecutableCommand.INSTANCE);
+		assertTrue(CREATION + TEST_IF_THE_COMMAND_CAN_BE_EXECUTED, command.canExecute());
+
+		getDiagramCommandStack().execute(command);
+	}
+
+}