tests/org.eclipse.papyrus.sysml.diagram.internalblock.tests/src/org/eclipse/papyrus/sysml/diagram/internalblock/tests/reorient/connector/TestLinkReorientSourceConnector.java

/*****************************************************************************
 * Copyright (c) 2012, 2014 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:
 *		
 *  CEA LIST - Initial API and implementation
 *  Christian W. Damus (CEA) - bug 434993
 *
 *****************************************************************************/
package org.eclipse.papyrus.sysml.diagram.internalblock.tests.reorient.connector;

import static org.eclipse.papyrus.sysml.diagram.internalblock.tests.utils.EditorUtils.getDiagramView;
import static org.eclipse.papyrus.sysml.diagram.internalblock.tests.utils.TestPrepareUtils.createGraphicalNode;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import org.eclipse.gmf.runtime.diagram.core.util.ViewUtil;
import org.eclipse.gmf.runtime.notation.View;
import org.eclipse.papyrus.sysml.diagram.common.utils.SysMLGraphicalTypes;
import org.eclipse.papyrus.sysml.diagram.internalblock.tests.utils.EditorUtils;
import org.eclipse.papyrus.sysml.diagram.internalblock.tests.utils.TestPrepareUtils;
import org.eclipse.papyrus.sysml.service.types.element.SysMLElementTypes;
import org.eclipse.papyrus.uml.diagram.common.utils.UMLGraphicalTypes;
import org.eclipse.papyrus.uml.service.types.element.UMLElementTypes;
import org.eclipse.uml2.uml.Property;
import org.junit.BeforeClass;

/**
 * JUnit tests for Connector re-orient.
 */
public class TestLinkReorientSourceConnector extends AbstractLinkReorientSourceConnectorTest {
	
	@BeforeClass
	public static void initRelationship() throws Exception {
		
		// Prepare source and target
		View container = ViewUtil.getChildBySemanticHint(getDiagramView(), SysMLGraphicalTypes.SHAPE_SYSML_BLOCK_AS_COMPOSITE_ID);
		View blockStructureView = ViewUtil.getChildBySemanticHint(container, SysMLGraphicalTypes.COMPARTMENT_SYSML_STRUCTURE_ID);
		View actorPartSourceView = createGraphicalNode(SysMLElementTypes.ACTOR_PART_PROPERTY, SysMLGraphicalTypes.SHAPE_SYSML_BLOCKPROPERTY_AS_COMPOSITE_ID, blockStructureView);
		View actorPartTargetView = createGraphicalNode(SysMLElementTypes.ACTOR_PART_PROPERTY, SysMLGraphicalTypes.SHAPE_SYSML_BLOCKPROPERTY_AS_COMPOSITE_ID, blockStructureView);

		// Prepare relationship
		relationshipView = TestPrepareUtils.createGraphicalLink(
			UMLElementTypes.CONNECTOR, 
			UMLGraphicalTypes.LINK_UML_CONNECTOR_ID, 
			container, 
			EditorUtils.getEditPart(actorPartSourceView), 
			EditorUtils.getEditPart(actorPartTargetView));
	}	
	
	@BeforeClass
	public static void initExpectedResults() throws Exception {
		
		Property part = (Property) partContainer1.getElement();
		List<Property> nestedPath = new ArrayList<Property>();
		nestedPath.add(part);
		
		Property nestedPart = (Property)nestedPartContainer1.getElement();
		List<Property> deepNestedPath = Arrays.asList(part, nestedPart);
				
		// Initialize source creation results

		isReorientAllowed.put(getDiagramView(), false);

		isReorientAllowed.put(blockSourceView, false);
		
		isReorientAllowed.put(actorPartSourceView, true);
		isReorientAllowed.put(nestedActorPartSourceView, true);
		
		isReorientAllowed.put(portOnBlockSourceView, true);
		isReorientAllowed.put(portOnPartSourceView, true);
		isReorientAllowed.put(portOnNestedPartSourceView, true);
		
		isReorientAllowed.put(flowportOnBlockSourceView, true);
		isReorientAllowed.put(flowportOnPartSourceView, true);
		isReorientAllowed.put(flowportOnNestedPartSourceView, true);
		
		isReorientAllowed.put(partSourceView, true);
		isReorientAllowed.put(nestedPartSourceView, true);
		
		isReorientAllowed.put(propertySourceView, true);
		isReorientAllowed.put(nestedPropertySourceView, true);
		
		isReorientAllowed.put(referenceSourceView, true);
		isReorientAllowed.put(nestedReferenceSourceView, true);
		
		isReorientAllowed.put(valueSourceView, true);
		isReorientAllowed.put(nestedValueSourceView, true);
		
		isReorientAllowed.put(commentSourceView, false);
		isReorientAllowed.put(commentCNSourceView, false);
		isReorientAllowed.put(constraintSourceView, false);
		isReorientAllowed.put(constraintCNSourceView, false);

		// Initialize nested property path for tests
		List<Property> emptyPath = Collections.emptyList();
		for (View view : isReorientAllowed.keySet()) {
			expectedNestedPath.put(view, emptyPath);
		}
		
		expectedNestedPath.put(nestedActorPartSourceView, nestedPath);
		expectedNestedPath.put(portOnNestedPartSourceView, deepNestedPath);
		expectedNestedPath.put(flowportOnNestedPartSourceView, deepNestedPath);
		expectedNestedPath.put(nestedPartSourceView, nestedPath);
		expectedNestedPath.put(nestedPropertySourceView, nestedPath);
		expectedNestedPath.put(nestedReferenceSourceView, nestedPath);
		expectedNestedPath.put(nestedValueSourceView, nestedPath);

	}
}