| /******************************************************************************* |
| * Copyright (c) 2016, 2019 Chalmers | University of Gothenburg, rt-labs and others. |
| * All rights reserved. This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License v2.0 |
| * which accompanies this distribution, and is available at |
| * http://www.eclipse.org/legal/epl-v20.html |
| * |
| * SPDX-License-Identifier: EPL-2.0 |
| * |
| * Contributors: |
| * Chalmers | University of Gothenburg and rt-labs - initial API and implementation and/or initial documentation |
| * Chalmers | University of Gothenburg - additional features, updated API |
| *******************************************************************************/ |
| package org.eclipse.capra.testsuite; |
| |
| import static org.eclipse.capra.testsuite.TestHelper.clearWorkspace; |
| import static org.eclipse.capra.testsuite.TestHelper.createEClassInEPackage; |
| import static org.eclipse.capra.testsuite.TestHelper.createEcoreModel; |
| import static org.eclipse.capra.testsuite.TestHelper.createSimpleProject; |
| import static org.eclipse.capra.testsuite.TestHelper.getProject; |
| import static org.eclipse.capra.testsuite.TestHelper.load; |
| import static org.eclipse.capra.testsuite.TestHelper.projectExists; |
| import static org.eclipse.capra.testsuite.TestHelper.resetSelectionView; |
| import static org.eclipse.capra.testsuite.TestHelper.save; |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertFalse; |
| import static org.junit.Assert.assertTrue; |
| import static org.junit.Assert.fail; |
| |
| import java.io.IOException; |
| import java.util.List; |
| import java.util.Optional; |
| |
| import org.eclipse.capra.core.adapters.TracePersistenceAdapter; |
| import org.eclipse.capra.core.helpers.ArtifactHelper; |
| import org.eclipse.capra.core.helpers.ExtensionPointHelper; |
| import org.eclipse.capra.core.helpers.TraceHelper; |
| import org.eclipse.capra.generic.tracemodel.TracemodelPackage; |
| import org.eclipse.capra.ui.operations.CreateTraceOperation; |
| import org.eclipse.capra.ui.views.SelectionView; |
| import org.eclipse.core.commands.ExecutionException; |
| import org.eclipse.core.commands.operations.IOperationHistory; |
| import org.eclipse.core.resources.IProject; |
| import org.eclipse.core.runtime.CoreException; |
| import org.eclipse.core.runtime.IAdaptable; |
| import org.eclipse.core.runtime.Status; |
| import org.eclipse.emf.ecore.EClass; |
| import org.eclipse.emf.ecore.EObject; |
| import org.eclipse.emf.ecore.EPackage; |
| import org.eclipse.emf.ecore.resource.ResourceSet; |
| import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl; |
| import org.eclipse.ui.IWorkbench; |
| import org.eclipse.ui.PlatformUI; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| public class TestCreateTraceOperation { |
| |
| private static final String CLASS_A_NAME = "A"; |
| private static final String CLASS_B_NAME = "B"; |
| |
| private static final String MODEL_A_FILENAME = "modelA.ecore"; |
| private static final String MODEL_B_FILENAME = "modelB.ecore"; |
| private static final String MODEL_A_NAME = "modelA"; |
| private static final String MODEL_B_NAME = "modelB"; |
| |
| private static final String TEST_PROJECT_NAME = "TestProject"; |
| |
| @Before |
| public void init() throws CoreException { |
| clearWorkspace(); |
| resetSelectionView(); |
| } |
| |
| @Test |
| public void testUndoTraceLinkCreation() throws CoreException, IOException { |
| // Create a project |
| createSimpleProject(TEST_PROJECT_NAME); |
| assertTrue(projectExists(TEST_PROJECT_NAME)); |
| |
| // Create two models and persist them |
| IProject testProject = getProject(TEST_PROJECT_NAME); |
| EPackage a = TestHelper.createEcoreModel(MODEL_A_NAME); |
| createEClassInEPackage(a, CLASS_A_NAME); |
| save(testProject, a); |
| |
| EPackage b = createEcoreModel(MODEL_B_NAME); |
| createEClassInEPackage(b, CLASS_B_NAME); |
| save(testProject, b); |
| |
| // Load them, choose two elements |
| ResourceSet rs = new ResourceSetImpl(); |
| |
| EPackage _a = load(testProject, MODEL_A_FILENAME, rs); |
| assertEquals(_a.getName(), MODEL_A_NAME); |
| EClass _A = (EClass) _a.getEClassifier(CLASS_A_NAME); |
| |
| EPackage _b = load(testProject, MODEL_B_FILENAME, rs); |
| assertEquals(_b.getName(), MODEL_B_NAME); |
| EClass _B = (EClass) _b.getEClassifier(CLASS_B_NAME); |
| |
| // Add them to the selection view |
| SelectionView.getOpenedView().dropToSelection(_A); |
| SelectionView.getOpenedView().dropToSelection(_B); |
| |
| EClass traceType = TracemodelPackage.eINSTANCE.getRelatedTo(); |
| |
| IAdaptable adapter = SelectionView.getOpenedView().getSite(); |
| |
| IWorkbench workbench = PlatformUI.getWorkbench(); |
| IOperationHistory operationHistory = workbench.getOperationSupport().getOperationHistory(); |
| |
| CreateTraceOperation createTraceOperation = new CreateTraceOperation("Create trace link", |
| SelectionView.getOpenedView().getSelection()); |
| createTraceOperation.setChooseTraceType((traceTypes, selection) -> { |
| if (traceTypes.contains(traceType)) { |
| return Optional.of(traceType); |
| } else { |
| return Optional.empty(); |
| } |
| }); |
| try { |
| assertEquals(operationHistory.execute(createTraceOperation, null, adapter), Status.OK_STATUS); |
| } catch (ExecutionException e) { |
| fail("Could not create trace: ExecutionException in operation"); |
| } |
| |
| TracePersistenceAdapter persistenceAdapter = ExtensionPointHelper.getTracePersistenceAdapter().get(); |
| TraceHelper traceHelper = new TraceHelper(persistenceAdapter.getTraceModel(new ResourceSetImpl())); |
| EObject artifactModel = persistenceAdapter.getArtifactWrappers(_A.eResource().getResourceSet()); |
| ArtifactHelper artifactHelper = new ArtifactHelper(artifactModel); |
| List<EObject> selection = artifactHelper.createWrappers(SelectionView.getOpenedView().getSelection()); |
| |
| // Check that the trace exists |
| selection = artifactHelper.createWrappers(SelectionView.getOpenedView().getSelection()); |
| assertTrue(traceHelper.traceExists(selection, traceType)); |
| |
| try { |
| assertEquals(operationHistory.undoOperation(createTraceOperation, null, adapter), Status.OK_STATUS); |
| } catch (ExecutionException e) { |
| fail("Could not undo trace creation: ExecutionException in operation"); |
| } |
| |
| // Check that the trace does not exist |
| traceHelper = new TraceHelper(persistenceAdapter.getTraceModel(new ResourceSetImpl())); |
| selection = artifactHelper.createWrappers(SelectionView.getOpenedView().getSelection()); |
| assertFalse(traceHelper.traceExists(selection, traceType)); |
| } |
| } |