tests/junit/plugins/uml/org.eclipse.papyrus.sirius.uml.diagram.clazz.tests/src/org/eclipse/papyrus/sirius/uml/diagram/clazz/tests/delete/view/nodes/TopNode_PackageWithSubNodes_DeleteViewTest.java

/******************************************************************************
 * Copyright (c) 2021 CEA LIST, Artal Technologies
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *  Rengin Battal (ARTAL) - rengin.battal@artal.fr - Initial API and implementation
 *****************************************************************************/
package org.eclipse.papyrus.sirius.uml.diagram.clazz.tests.delete.view.nodes;

import org.eclipse.emf.ecore.EObject;
import org.eclipse.gmf.runtime.diagram.ui.editparts.DiagramEditPart;
import org.eclipse.gmf.runtime.notation.Diagram;
import org.eclipse.gmf.runtime.notation.View;
import org.eclipse.papyrus.junit.utils.rules.ActiveDiagram;
import org.eclipse.papyrus.junit.utils.rules.PluginResource;
import org.eclipse.papyrus.sirius.uml.diagram.clazz.tests.AbstractSiriusDiagramTests;
import org.eclipse.sirius.diagram.DNodeContainer;
import org.eclipse.uml2.uml.Model;
import org.eclipse.uml2.uml.internal.impl.PackageImpl;
import org.junit.Assert;
import org.junit.Test;

/**
 * Delete Package element and it's sub nodes from diagram test
 */
@PluginResource("resources/delete/nodes/package/TopNode_PackageWithSubNodes_DeleteSemanticTest.di")
@SuppressWarnings("nls")
public class TopNode_PackageWithSubNodes_DeleteViewTest extends AbstractSiriusDiagramTests {

	private static final String ELEMENT_TO_DESTROY_NAME = "PackageToDelete";

	private static final String SUB_ELEMENT_TO_DESTROY_NAME = "InterfaceToDelete";

	private static final String CLASS_DIAGRAM_NAME = "TopNode_Package_Delete_ClassDiagram";

	@SuppressWarnings("restriction")
	@Test
	@ActiveDiagram(CLASS_DIAGRAM_NAME)
	public void Delete_Package_View() {
		checkSiriusDiagramSynchronization(false);

		// setup
		Assert.assertTrue(fixture.getModel() instanceof Model);
		Model rootModel = (Model) fixture.getModel();
		PackageImpl packageElement = (PackageImpl) rootModel.getMember(ELEMENT_TO_DESTROY_NAME);
		Assert.assertNotNull("We can't find the element to destroy", packageElement);
		Assert.assertNotNull("The UML model must not contain the destroyed sub element", packageElement.getPackagedElement(SUB_ELEMENT_TO_DESTROY_NAME));

		DiagramEditPart diagramEditpart = fixture.getActiveDiagram();
		Diagram classDiagram = diagramEditpart.getDiagramView();
		Assert.assertNotNull("We can't find the class diagram", classDiagram);

		Assert.assertEquals("The diagram must contains one element", 1, classDiagram.getChildren().size());
		Object elementToBeDeleted = classDiagram.getChildren().get(0);
		Assert.assertTrue(elementToBeDeleted instanceof View);
		EObject siriusNewRepresentation = ((View) elementToBeDeleted).getElement();
		Assert.assertTrue(siriusNewRepresentation instanceof DNodeContainer);
		Assert.assertEquals("The found view doesn't represent the element to destroy", packageElement, ((DNodeContainer) siriusNewRepresentation).getTarget());

		fixture.applyGraphicalDeletionTool((DNodeContainer) siriusNewRepresentation);
		fixture.flushDisplayEvents();

		// the semantic elements has not been destroyed
		packageElement = (PackageImpl) rootModel.getMember(ELEMENT_TO_DESTROY_NAME);
		Assert.assertNotNull("The UML model must not contain the destroyed element", packageElement);
		Assert.assertNotNull("The UML model must not contain the destroyed sub element", packageElement.getPackagedElement(SUB_ELEMENT_TO_DESTROY_NAME));

		// the view has been destroyed in the diagram
		Assert.assertEquals("The class diagram must not contain view after destruction of the class", 0, classDiagram.getChildren().size());

		// undo
		fixture.getEditingDomain().getCommandStack().undo();
		fixture.flushDisplayEvents();
		// the semantic and the view elements have been reset
		packageElement = (PackageImpl) rootModel.getMember(ELEMENT_TO_DESTROY_NAME);
		Assert.assertNotNull("The UML model must contain the destroyed element after undo", packageElement);
		Assert.assertNotNull("The UML model must not contain the destroyed sub element", packageElement.getPackagedElement(SUB_ELEMENT_TO_DESTROY_NAME));
		Assert.assertEquals("The class diagram must contain view after destruction of the class after undo", 1, classDiagram.getChildren().size());

		// redo
		fixture.getEditingDomain().getCommandStack().redo();
		fixture.flushDisplayEvents();
		packageElement = (PackageImpl) rootModel.getMember(ELEMENT_TO_DESTROY_NAME);
		Assert.assertNotNull("The UML model must contain the destroyed element after undo", packageElement);
		Assert.assertNotNull("The UML model must not contain the destroyed sub element", packageElement.getPackagedElement(SUB_ELEMENT_TO_DESTROY_NAME));
		Assert.assertEquals("The class diagram must not contain view after destruction of the class", 0, classDiagram.getChildren().size());
	}
}