1 files changed, 363 insertions, 0 deletions
diff --git a/plugins/cdo/tests/org.eclipse.papyrus.cdo.ui.tests/src/org/eclipse/papyrus/cdo/ui/tests/AbstractPapyrusCDOUITest.java b/plugins/cdo/tests/org.eclipse.papyrus.cdo.ui.tests/src/org/eclipse/papyrus/cdo/ui/tests/AbstractPapyrusCDOUITest.java
new file mode 100755
index 00000000..f88d229c
--- /dev/null
+++ b/plugins/cdo/tests/org.eclipse.papyrus.cdo.ui.tests/src/org/eclipse/papyrus/cdo/ui/tests/AbstractPapyrusCDOUITest.java
@@ -0,0 +1,363 @@
+/*****************************************************************************
+ * Copyright (c) 2013, 2017 CEA LIST, Christian W. Damus, 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 422257
+ *   Christian W. Damus (CEA) - bug 432813
+ *   Eike Stepper (CEA) - bug 466520
+ *   Christian W. Damus - bug 434983
+ *
+ *****************************************************************************/
+package org.eclipse.papyrus.cdo.ui.tests;
+
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.CoreMatchers.notNullValue;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.fail;
+
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+
+import org.eclipse.emf.cdo.dawn.editors.IDawnEditor;
+import org.eclipse.emf.cdo.dawn.editors.IDawnEditorSupport;
+import org.eclipse.emf.cdo.dawn.gmf.util.DawnDiagramUpdater;
+import org.eclipse.emf.cdo.transaction.CDOTransaction;
+import org.eclipse.emf.common.util.ECollections;
+import org.eclipse.emf.common.util.URI;
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.emf.ecore.resource.Resource;
+import org.eclipse.emf.ecore.resource.ResourceSet;
+import org.eclipse.emf.ecore.util.EcoreUtil;
+import org.eclipse.emf.transaction.RecordingCommand;
+import org.eclipse.emf.transaction.TransactionalEditingDomain;
+import org.eclipse.gef.EditPart;
+import org.eclipse.gmf.runtime.diagram.ui.editparts.DiagramEditPart;
+import org.eclipse.gmf.runtime.diagram.ui.resources.editor.parts.DiagramDocumentEditor;
+import org.eclipse.gmf.runtime.notation.Diagram;
+import org.eclipse.gmf.runtime.notation.View;
+import org.eclipse.papyrus.cdo.core.tests.AbstractPapyrusCDOTest;
+import org.eclipse.papyrus.cdo.internal.ui.editors.PapyrusCDOEditorManager;
+import org.eclipse.papyrus.infra.core.resource.ModelSet;
+import org.eclipse.papyrus.infra.ui.editor.IMultiDiagramEditor;
+import org.eclipse.swt.widgets.Display;
+import org.eclipse.ui.IEditorPart;
+import org.eclipse.ui.IPerspectiveDescriptor;
+import org.eclipse.ui.IViewPart;
+import org.eclipse.ui.IViewReference;
+import org.eclipse.ui.IWorkbenchPage;
+import org.eclipse.ui.PlatformUI;
+import org.eclipse.uml2.uml.NamedElement;
+import org.eclipse.uml2.uml.Namespace;
+import org.eclipse.uml2.uml.Package;
+import org.eclipse.uml2.uml.UMLPackage;
+import org.junit.After;
+import org.junit.Before;
+
+import com.google.common.base.Splitter;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.Lists;
+import com.google.common.collect.Maps;
+
+/**
+ * This is the AbstractPapyrusCDOUITest type. Enjoy.
+ */
+public abstract class AbstractPapyrusCDOUITest extends AbstractPapyrusCDOTest {
+
+	public static final String PLUGIN_ID = "org.eclipse.papyrus.cdo.ui.tests";
+
+	protected static final String TEST_MODEL_NAME = "model.di";
+
+	protected static final String TEST_UML_NAME = "model.uml";
+
+	protected static final String TEST_NOTATION_NAME = "model.notation";
+
+	private IWorkbenchPage page;
+
+	private List<IEditorPart> openedEditors = Lists.newArrayList();
+
+	private IMultiDiagramEditor lastEditor;
+
+	private DiagramDocumentEditor lastDiagramEditor;
+
+	@SuppressWarnings("restriction")
+	private org.eclipse.papyrus.infra.ui.internal.preferences.YesNo initialSashStoragePreference;
+
+	public AbstractPapyrusCDOUITest() {
+		super();
+	}
+
+	@Before
+	public void initWorkbench() throws Exception {
+
+		page = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage();
+
+		// ensure the Papyrus perspective
+		IPerspectiveDescriptor perspective = page.getWorkbenchWindow().getWorkbench().getPerspectiveRegistry().findPerspectiveWithId("org.eclipse.papyrus.infra.core.perspective");
+		if (!perspective.getId().equals(page.getPerspective().getId())) {
+			page.setPerspective(perspective);
+		}
+
+		// minimize the Welcome view
+		for (IViewReference next : page.getViewReferences()) {
+			if ("org.eclipse.ui.internal.introview".equals(next.getId())) {
+				page.setPartState(next, IWorkbenchPage.STATE_MINIMIZED);
+				break;
+			}
+		}
+
+		// bring the Repository Explorer forward
+		IViewPart reposView = page.showView("org.eclipse.emf.cdo.explorer.ui.CDORepositoriesView");
+		page.activate(reposView);
+
+		page.setEditorAreaVisible(true);
+
+		flushDisplayEvents();
+	}
+
+	@Before
+	public void importTestModel() throws Exception {
+
+		CDOTransaction transaction = createTransaction();
+		Map<Resource, Resource> importMap = Maps.newHashMap();
+
+		importResource(transaction, TEST_MODEL_NAME, TEST_MODEL_NAME, importMap);
+		importResource(transaction, TEST_UML_NAME, TEST_UML_NAME, importMap);
+		importResource(transaction, TEST_NOTATION_NAME, TEST_NOTATION_NAME, importMap);
+
+		for (Map.Entry<Resource, Resource> next : importMap.entrySet()) {
+			Resource xml = next.getKey();
+			Resource cdo = next.getValue();
+			ECollections.setEList(cdo.getContents(), ImmutableList.copyOf(xml.getContents()));
+			xml.unload();
+			xml.eAdapters().clear();
+			xml.getResourceSet().getResources().remove(xml);
+		}
+
+		commit(transaction.getResourceSet());
+		close(getCheckout(), transaction);
+	}
+
+	@SuppressWarnings("restriction")
+	@Before
+	public void suppressSashLayoutMigrationDialog() {
+		initialSashStoragePreference = org.eclipse.papyrus.infra.ui.internal.preferences.EditorPreferences.getInstance().getConvertSharedPageLayoutToPrivate();
+		org.eclipse.papyrus.infra.ui.internal.preferences.EditorPreferences.getInstance().setConvertSharedPageLayoutToPrivate(org.eclipse.papyrus.infra.ui.internal.preferences.YesNo.NO);
+	}
+
+	@After
+	public void closeEditors() {
+		flushDisplayEvents();
+
+		for (IEditorPart next : openedEditors) {
+			try {
+				page.closeEditor(next, false);
+			} catch (Exception e) {
+				e.printStackTrace();
+			}
+
+			flushDisplayEvents();
+		}
+
+		openedEditors.clear();
+
+		lastEditor = null;
+		lastDiagramEditor = null;
+
+		flushDisplayEvents();
+	}
+
+	@SuppressWarnings("restriction")
+	@After
+	public void restoreSashLayoutMigrationDialog() {
+		org.eclipse.papyrus.infra.ui.internal.preferences.EditorPreferences.getInstance().setConvertSharedPageLayoutToPrivate(initialSashStoragePreference);
+	}
+
+	private void importResource(CDOTransaction transaction, String srcPath, String dstPath, Map<? super Resource, ? super Resource> importMap) {
+		ResourceSet rset = transaction.getResourceSet();
+
+		Resource xml = rset.getResource(URI.createPlatformPluginURI(PLUGIN_ID + "/resources/" + srcPath, true), true);
+
+		EcoreUtil.resolveAll(xml);
+
+		Resource cdo = transaction.getOrCreateResource(getResourcePath(dstPath));
+
+		importMap.put(xml, cdo);
+	}
+
+	protected IWorkbenchPage getWorkbenchPage() {
+		return page;
+	}
+
+	protected IViewPart getRepositoryExplorer() {
+		return page.findView("org.eclipse.emf.cdo.explorer.ui.CDORepositoriesView");
+	}
+
+	protected IEditorPart openEditor() {
+		URI uri = getTestResourceURI(TEST_MODEL_NAME);
+
+		try {
+			IEditorPart result = PapyrusCDOEditorManager.INSTANCE.openEditor(getWorkbenchPage(), uri, TEST_MODEL_NAME);
+
+			openedEditors.add(result);
+
+			assertThat(result, instanceOf(IMultiDiagramEditor.class));
+
+			lastEditor = (IMultiDiagramEditor) result;
+			if (lastEditor.getActiveEditor() instanceof DiagramDocumentEditor) {
+				lastDiagramEditor = (DiagramDocumentEditor) lastEditor.getActiveEditor();
+			}
+
+			flushDisplayEvents();
+
+			return result;
+		} catch (Exception e) {
+			e.printStackTrace();
+			fail("Failed to open editor on test model.");
+			return null; // make the compiler happy
+		}
+	}
+
+	protected void closeEditor() {
+		closeEditor(lastEditor);
+
+		lastDiagramEditor = null;
+		lastEditor = null;
+	}
+
+	protected void closeEditor(IEditorPart editor) {
+		page.closeEditor(editor, false);
+		openedEditors.remove(editor);
+
+		flushDisplayEvents();
+	}
+
+	protected void flushDisplayEvents() {
+		while (Display.getCurrent().readAndDispatch()) {
+			// pass
+		}
+	}
+
+	protected void sleep(int seconds) {
+		for (int i = 0; i < seconds; i++) {
+			try {
+				Thread.sleep(1000);
+			} catch (Exception e) {
+				// fine. So, we don't sleep so long
+			}
+
+			flushDisplayEvents();
+		}
+	}
+
+	protected IDawnEditor getDawnEditor() {
+		return lastEditor.getAdapter(IDawnEditor.class);
+	}
+
+	protected IDawnEditorSupport getDawnEditorSupport() {
+		return getDawnEditor().getDawnEditorSupport();
+	}
+
+	protected void executeEdit(final Runnable edit) {
+		TransactionalEditingDomain domain = getModelSet().getTransactionalEditingDomain();
+
+		domain.getCommandStack().execute(new RecordingCommand(domain, "Edit") {
+
+			@Override
+			protected void doExecute() {
+				edit.run();
+			}
+		});
+	}
+
+	protected ModelSet getModelSet() {
+		ModelSet result = null;
+
+		try {
+			result = lastEditor.getServicesRegistry().getService(ModelSet.class);
+		} catch (Exception e) {
+			e.printStackTrace();
+			fail("Failed to get ModelSet from current editor.");
+		}
+
+		return result;
+	}
+
+	protected Package getUMLModel() {
+		Resource resource = getModelSet().getResource(getTestResourceURI(TEST_UML_NAME), true);
+		assertThat("UML model not found.", resource, notNullValue());
+		return (Package) EcoreUtil.getObjectByType(resource.getContents(), UMLPackage.Literals.PACKAGE);
+	}
+
+	protected <N extends NamedElement> N findElement(String relativeName, Class<N> type) {
+
+		N result = null;
+		Namespace namespace = getUMLModel();
+
+		for (Iterator<String> iter = Splitter.on(NamedElement.SEPARATOR).split(relativeName).iterator(); iter.hasNext();) {
+
+			NamedElement next = namespace.getMember(iter.next());
+			if (next instanceof Namespace) {
+				namespace = (Namespace) next;
+			}
+
+			if (!iter.hasNext()) {
+				if (type.isInstance(next)) {
+					result = type.cast(next);
+				}
+			}
+		}
+
+		return result;
+	}
+
+	protected <N extends NamedElement> N expectElement(String relativeName, Class<N> type) {
+
+		N result = findElement(relativeName, type);
+		assertThat("UML element not found.", result, notNullValue());
+		return result;
+	}
+
+	protected IMultiDiagramEditor getEditor() {
+		return lastEditor;
+	}
+
+	protected DiagramDocumentEditor getDiagramEditor() {
+		return lastDiagramEditor;
+	}
+
+	protected DiagramEditPart getDiagramEditPart() {
+		return getDiagramEditor().getDiagramEditPart();
+	}
+
+	protected Diagram getDiagram() {
+		return getDiagramEditPart().getDiagramView();
+	}
+
+	protected View findView(EObject element) {
+		return DawnDiagramUpdater.findViewForModel(element, getDiagramEditor());
+	}
+
+	protected View expectView(EObject element) {
+		View result = findView(element);
+		assertThat("Required view not found.", result, notNullValue());
+		return result;
+	}
+
+	protected EditPart findEditPart(EObject element) {
+		View view = findView(element);
+		return (view == null) ? null : DawnDiagramUpdater.findEditPart(view, getDiagramEditor());
+	}
+
+	protected EditPart expectEditPart(EObject element) {
+		EditPart result = findEditPart(element);
+		assertThat("Required edit-part not found.", result, notNullValue());
+		return result;
+	}
+}