[ui.behavior.test] added test for state machine inheritance

2 files changed, 152 insertions, 0 deletions

diff --git a/tests/org.eclipse.etrice.ui.behavior.tests/src/org/eclipse/etrice/ui/behavior/commands/AbstractStateMachineTest.java b/tests/org.eclipse.etrice.ui.behavior.tests/src/org/eclipse/etrice/ui/behavior/commands/AbstractStateMachineTest.java
index 767be6c9a..ac8bf0cde 100644
--- a/tests/org.eclipse.etrice.ui.behavior.tests/src/org/eclipse/etrice/ui/behavior/commands/AbstractStateMachineTest.java
+++ b/tests/org.eclipse.etrice.ui.behavior.tests/src/org/eclipse/etrice/ui/behavior/commands/AbstractStateMachineTest.java
@@ -22,6 +22,7 @@ import org.eclipse.etrice.core.room.ChoicePoint;
 import org.eclipse.etrice.core.room.EntryPoint;

 import org.eclipse.etrice.core.room.ExitPoint;

 import org.eclipse.etrice.core.room.InitialTransition;

+import org.eclipse.etrice.core.room.RefinedState;

 import org.eclipse.etrice.core.room.State;

 import org.eclipse.etrice.core.room.StateGraph;

 import org.eclipse.etrice.core.room.StateGraphItem;

@@ -150,6 +151,9 @@ public abstract class AbstractStateMachineTest extends TestBase {
 	 * @return true if state graph item is not owned by the given actor class

 	 */

 	private boolean isInherited(ActorClass ac, StateGraphItem item) {

+		if (item instanceof RefinedState)

+			return true;

+		

 		EObject owner = item.eContainer();

 		while (owner!=null) {

 			if (owner instanceof ActorClass)

diff --git a/tests/org.eclipse.etrice.ui.behavior.tests/src/org/eclipse/etrice/ui/behavior/commands/TestInheritedStateMachine.java b/tests/org.eclipse.etrice.ui.behavior.tests/src/org/eclipse/etrice/ui/behavior/commands/TestInheritedStateMachine.java
new file mode 100644
index 000000000..b095eaa45
--- /dev/null
+++ b/tests/org.eclipse.etrice.ui.behavior.tests/src/org/eclipse/etrice/ui/behavior/commands/TestInheritedStateMachine.java
@@ -0,0 +1,148 @@
+/*******************************************************************************

+ * Copyright (c) 2010 protos software gmbh (http://www.protos.de).

+ * 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:

+ * 		Thomas Schuetz and Henrik Rentz-Reichert (initial contribution)

+ * 

+ *******************************************************************************/

+

+package org.eclipse.etrice.ui.behavior.commands;

+

+import static org.junit.Assert.assertEquals;

+import static org.junit.Assert.assertTrue;

+

+

+import org.eclipse.emf.ecore.EObject;

+import org.eclipse.etrice.core.room.ActorClass;

+import org.eclipse.etrice.core.room.State;

+import org.eclipse.etrice.core.room.StateGraph;

+import org.eclipse.etrice.ui.behavior.DiagramAccess;

+import org.eclipse.graphiti.mm.pictograms.ContainerShape;

+import org.eclipse.graphiti.mm.pictograms.Diagram;

+import org.eclipse.graphiti.mm.pictograms.Shape;

+import org.eclipse.graphiti.services.Graphiti;

+import org.junit.After;

+import org.junit.Before;

+import org.junit.Test;

+

+/**

+ * unit test of a hierarchical state machine (defined by HierarchicalStateMachine.room)

+ * 

+ * @author Henrik Rentz-Reichert initial contribution and API

+ *

+ */

+public class TestInheritedStateMachine extends AbstractStateMachineTest {

+

+	@Before

+	public void setUp() {

+		loadModelFile();

+	}

+	

+	@After

+	public void tearDown() {

+		removeDiagramsDirectory();

+	}

+	

+	@Override

+	protected String getModelFileName() {

+		return "InheritedStateMachine.room";

+	}

+	

+	@Test

+	public void checkModel() {

+		assertEquals("models read", 1, getModels().size());

+		assertEquals("actor classes in our model", 3, getModels().get(0).getActorClasses().size());

+	}

+	

+	@Test

+	public void testBaseClass() {

+		ActorClass ac = getActorClass("Base");

+		assertTrue("actor class Base present in model", ac!=null);

+		Diagram diagram = new DiagramAccess().getDiagram(ac);

+		

+		// have to use the actor class from the diagram since this is another instance

+		ac = (ActorClass) Graphiti.getLinkService().getBusinessObjectForLinkedPictogramElement(diagram);

+		

+		testSGItems(ac.getStateMachine(), diagram);

+		

+		// top and one sub state graph

+		assertEquals("diagram children", 2, diagram.getChildren().size());

+

+		for (Shape shape : diagram.getChildren()) {

+			assertTrue("top level shape is container shape", shape instanceof ContainerShape);

+			

+			EObject bo = Graphiti.getLinkService().getBusinessObjectForLinkedPictogramElement(shape);

+			assertTrue("top level business object is StateGraph", bo instanceof StateGraph);

+		}

+	}

+	

+	@Test

+	public void testDerivedClass() {

+		ActorClass ac = getActorClass("Derived");

+		assertTrue("actor class Derived present in model", ac!=null);

+		Diagram diagram = new DiagramAccess().getDiagram(ac);

+		

+		// have to use the actor class from the diagram since this is another instance

+		ac = (ActorClass) Graphiti.getLinkService().getBusinessObjectForLinkedPictogramElement(diagram);

+		assertTrue("is derived", ac.getBase()!=null);

+		assertTrue("extends Base", ac.getBase().getName().equals("Base"));

+		

+		testSGItems(ac.getStateMachine(), diagram);

+		

+		// top and two sub state graph

+		assertEquals("diagram children", 3, diagram.getChildren().size());

+

+		for (Shape shape : diagram.getChildren()) {

+			assertTrue("top level shape is container shape", shape instanceof ContainerShape);

+			

+			EObject bo = Graphiti.getLinkService().getBusinessObjectForLinkedPictogramElement(shape);

+			assertTrue("top level business object is StateGraph", bo instanceof StateGraph);

+		}

+	}

+	

+	@Test

+	public void testDoubleDerivedClass() {

+		ActorClass ac = getActorClass("DoubleDerived");

+		assertTrue("actor class DoubleDerived present in model", ac!=null);

+		Diagram diagram = new DiagramAccess().getDiagram(ac);

+		

+		// have to use the actor class from the diagram since this is another instance

+		ac = (ActorClass) Graphiti.getLinkService().getBusinessObjectForLinkedPictogramElement(diagram);

+		assertTrue("is derived", ac.getBase()!=null);

+		assertTrue("extends Base", ac.getBase().getName().equals("Derived"));

+		

+		testSGItems(ac.getStateMachine(), diagram);

+		

+		// top and four sub state graph

+		assertEquals("diagram children", 5, diagram.getChildren().size());

+

+		for (Shape shape : diagram.getChildren()) {

+			assertTrue("top level shape is container shape", shape instanceof ContainerShape);

+			

+			EObject bo = Graphiti.getLinkService().getBusinessObjectForLinkedPictogramElement(shape);

+			assertTrue("top level business object is StateGraph", bo instanceof StateGraph);

+		}

+	}

+

+	public void testSGItems(StateGraph sg, Diagram diagram) {

+		

+		testStateGraph(diagram, sg);

+		

+		for (State s : sg.getStates()) {

+			if (s.getSubgraph()!=null)

+				testSGItems(s.getSubgraph(), diagram);

+		}

+	}

+	

+	private ActorClass getActorClass(String name) {

+		for (ActorClass ac : getModels().get(0).getActorClasses()) {

+			if (ac.getName().equals(name))

+				return ac;

+		}

+		return null;

+	}

+}