1 files changed, 83 insertions, 0 deletions
diff --git a/extraplugins/bpmn/org.eclipse.papyrus.bpmn.tests/src/org/eclipse/papyrus/bpmn/test/FlowElementCustomTest.java b/extraplugins/bpmn/org.eclipse.papyrus.bpmn.tests/src/org/eclipse/papyrus/bpmn/test/FlowElementCustomTest.java
new file mode 100644
index 00000000000..fb3c3010dfb
--- /dev/null
+++ b/extraplugins/bpmn/org.eclipse.papyrus.bpmn.tests/src/org/eclipse/papyrus/bpmn/test/FlowElementCustomTest.java
@@ -0,0 +1,83 @@
+/*****************************************************************************
+ * Copyright (c) 2015 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
+ *
+ *****************************************************************************/
+package org.eclipse.papyrus.bpmn.test;
+
+import org.eclipse.papyrus.bpmn.BPMNProfile.BPMNProcess;
+import org.eclipse.papyrus.bpmn.BPMNProfile.BPMNProfilePackage;
+import org.eclipse.papyrus.bpmn.BPMNProfile.BoundaryEvent;
+import org.eclipse.papyrus.bpmn.BPMNProfile.FlowElement;
+import org.eclipse.papyrus.bpmn.BPMNProfile.FlowElementsContainer;
+import org.eclipse.papyrus.bpmn.BPMNProfile.Task;
+import org.eclipse.papyrus.bpmn.BPMNProfile.impl.FlowElementCustom;
+import org.eclipse.papyrus.bpmn.util.BPMNResource;
+import org.eclipse.uml2.uml.Activity;
+import org.eclipse.uml2.uml.ChangeEvent;
+import org.eclipse.uml2.uml.Model;
+import org.eclipse.uml2.uml.OpaqueAction;
+import org.eclipse.uml2.uml.UMLFactory;
+import org.eclipse.uml2.uml.util.UMLUtil;
+import org.eclipse.uml2.uml.util.UMLUtil.StereotypeApplicationHelper;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+
+public class FlowElementCustomTest {
+
+	OpaqueAction action;
+	BPMNProcess bpmnProcess;
+	BoundaryEvent boundaryEvent;
+	Task task;
+	
+	@Before	
+	public void setUp(){
+		Model model = BPMNResource.createBPMNModel();
+		
+		Activity activity = UMLFactory.eINSTANCE.createActivity();
+		model.getPackagedElements().add(activity);
+		StereotypeApplicationHelper.getInstance(null).applyStereotype(activity, BPMNProfilePackage.eINSTANCE.getBPMNProcess());
+		bpmnProcess = UMLUtil.getStereotypeApplication(activity, BPMNProcess.class);
+		
+		action = UMLFactory.eINSTANCE.createOpaqueAction();
+		activity.getOwnedNodes().add(action);		
+		task = (Task) StereotypeApplicationHelper.getInstance(null).applyStereotype(action, BPMNProfilePackage.eINSTANCE.getTask());
+		
+		ChangeEvent changeEvent = UMLFactory.eINSTANCE.createChangeEvent();
+		model.getPackagedElements().add(changeEvent);
+		boundaryEvent = (BoundaryEvent) StereotypeApplicationHelper.getInstance(null).applyStereotype(changeEvent, BPMNProfilePackage.eINSTANCE.getBoundaryEvent());
+		task.getBoundaryEventRefs().add(boundaryEvent);
+		boundaryEvent.setBase_ActivityNode(task.getBase_ActivityNode());
+	}
+	
+	@Test
+	public void basicGetContainerGeneratedTest(){
+		try{
+			FlowElement fe = UMLUtil.getStereotypeApplication(task.getBase_Element(), FlowElement.class);
+			fe.getContainer();
+		}
+		catch(UnsupportedOperationException e){
+			Assert.fail("missing custom code !");
+		}
+	}
+	
+	@Test
+	public void basicGetContainerCustomTest(){
+		Assert.assertEquals(bpmnProcess, FlowElementCustom.basicGetContainer(UMLUtil.getStereotypeApplication(task.getBase_Element(), FlowElement.class)));
+		Assert.assertEquals(bpmnProcess, FlowElementCustom.basicGetContainer(UMLUtil.getStereotypeApplication(boundaryEvent.getBase_Element(), FlowElement.class)));
+	}
+	
+	@Test
+	public void basicGetContainerTest(){
+		FlowElement fe1 = UMLUtil.getStereotypeApplication(task.getBase_Element(), FlowElement.class);
+		Assert.assertEquals(bpmnProcess, fe1.getContainer());
+		FlowElement fe2 = UMLUtil.getStereotypeApplication(boundaryEvent.getBase_Element(), FlowElement.class);
+		Assert.assertEquals(bpmnProcess, fe2.getContainer());
+	}
+}