plugins/org.eclipse.sirius.tests.junit/src/org/eclipse/sirius/tests/unit/api/mappings/NodeMappingTest.java

/*******************************************************************************
 * Copyright (c) 2010, 2014 THALES GLOBAL SERVICES.
 * 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:
 *    Obeo - initial API and implementation
 *******************************************************************************/
package org.eclipse.sirius.tests.unit.api.mappings;

import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.sirius.diagram.description.DiagramDescription;
import org.eclipse.sirius.diagram.description.NodeMapping;
import org.eclipse.sirius.tests.support.api.SiriusDiagramTestCase;
import org.eclipse.sirius.viewpoint.description.RepresentationDescription;

/**
 * Tests for NodeMapping usage.
 * 
 * @author lredor
 */
public class NodeMappingTest extends SiriusDiagramTestCase implements MappingsReuseTestModeler {

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        genericSetUp(TEST_SEMANTIC_MODEL_PATH, MODELER_PATH);
        initViewpoint(DEFAULT_VIEWPOINT_NAME);
    }

    /**
     * Test getNodesCandidates() (avoid Stack overflow of #2121).
     * 
     * @throws Exception
     */
    public void testGetNodesCandidates() throws Exception {
        assertEquals("Bad number of Sirius", 1, viewpoints.size());
        assertEquals("Bad number of Representations", 6, viewpoints.iterator().next().getOwnedRepresentations().size());
        assertEquals("Bad name for the first representation", "Common", viewpoints.iterator().next().getOwnedRepresentations().get(0).getName());
        assertEquals("Bad name for the first representation", "Common", viewpoints.iterator().next().getOwnedRepresentations().get(0).getName());
        RepresentationDescription representationDesc = viewpoints.iterator().next().getOwnedRepresentations().get(0);
        assertTrue("Bad type for representation", representationDesc instanceof DiagramDescription);
        DiagramDescription diagDesc = (DiagramDescription) representationDesc;
        for (NodeMapping nodeMapping : diagDesc.getAllNodeMappings()) {
            if (COMMON_NODE_MAPPING_ON_ECLASS.equals(nodeMapping.getName())) {
                EList<EObject> nodesCandidates = nodeMapping.getNodesCandidates(semanticModel, semanticModel);
                assertEquals("Bad number of nodes candidates for mapping \"" + COMMON_NODE_MAPPING_ON_ECLASS + "\"", 1, nodesCandidates.size());
            }
        }
    }
}