plugins/cdo/org.eclipse.papyrus.cdo.core.tests/src/org/eclipse/papyrus/cdo/internal/core/exporter/tests/AbstractModelExportTest.java

/*****************************************************************************
 * Copyright (c) 2013, 2017 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 429242
 *
 *****************************************************************************/
package org.eclipse.papyrus.cdo.internal.core.exporter.tests;

import static org.hamcrest.CoreMatchers.containsString;
import static org.hamcrest.CoreMatchers.instanceOf;
import static org.hamcrest.CoreMatchers.not;
import static org.junit.Assert.assertThat;
import static org.junit.Assume.assumeThat;

import java.io.IOException;

import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.transaction.RecordingCommand;
import org.eclipse.emf.transaction.TransactionalEditingDomain;
import org.eclipse.papyrus.cdo.core.importer.IModelTransferConfiguration;
import org.eclipse.papyrus.cdo.core.importer.IModelTransferNode;
import org.eclipse.papyrus.cdo.core.resource.CDOAwareModelSet;
import org.eclipse.papyrus.cdo.core.tests.AbstractPapyrusCDOTest;
import org.eclipse.papyrus.infra.core.Activator;
import org.eclipse.papyrus.infra.core.resource.AbstractBaseModel;
import org.eclipse.papyrus.infra.core.resource.ModelSet;
import org.eclipse.papyrus.infra.core.resource.sasheditor.DiModel;
import org.eclipse.papyrus.infra.core.services.ExtensionServicesRegistry;
import org.eclipse.papyrus.infra.core.services.ServiceMultiException;
import org.eclipse.papyrus.infra.core.services.ServiceNotFoundException;
import org.eclipse.papyrus.infra.core.services.ServicesRegistry;
import org.eclipse.papyrus.infra.gmfdiag.common.model.NotationModel;
import org.eclipse.papyrus.uml.tools.model.UmlModel;
import org.eclipse.uml2.common.util.UML2Util;
import org.eclipse.uml2.uml.Class;
import org.eclipse.uml2.uml.Model;
import org.eclipse.uml2.uml.UMLPackage;

import com.google.common.collect.Iterables;


/**
 * This is the AbstractModelExportTest type. Enjoy.
 */
public abstract class AbstractModelExportTest extends AbstractPapyrusCDOTest {

	protected static final String DEPENDENT_MODEL_NAME = "dependent.uml";

	protected static final String DEPENDENCY_MODEL_NAME = "dependency.uml";

	protected static final String DEPENDENT_MODEL_DI = "dependent.di";

	protected static final String DEPENDENCY_MODEL_DI = "dependency.di";

	protected void createModels() throws Exception {
		// create the model that the other model depends on
		ServicesRegistry[] services1 = { null };
		final ModelSet modelSet1 = createModelSet(DEPENDENCY_MODEL_NAME, services1);

		UmlModel uml1 = (UmlModel) modelSet1.getModel(UmlModel.MODEL_ID);
		final Model model1 = (Model) uml1.lookupRoot();
		TransactionalEditingDomain ted1 = modelSet1.getTransactionalEditingDomain();
		RecordingCommand recording1 = new RecordingCommand(ted1) {

			@Override
			protected void doExecute() {
				model1.createOwnedClass("Superclass", false);

				try {
					modelSet1.save(new NullProgressMonitor());
				} catch (IOException e) {
					Activator.log.error("The modelSet of the fixture failed on save", e); // $NON-NLS-1$
				}
			}
		};
		ted1.getCommandStack().execute(recording1);

		// create the model that depends on the first one
		ServicesRegistry[] services2 = { null };
		final ModelSet modelSet2 = createModelSet(DEPENDENT_MODEL_NAME, services2);

		UmlModel uml2 = (UmlModel) modelSet2.getModel(UmlModel.MODEL_ID);
		final Model model2 = (Model) uml2.lookupRoot();
		TransactionalEditingDomain ted2 = modelSet2.getTransactionalEditingDomain();
		RecordingCommand recording2 = new RecordingCommand(ted2) {

			@Override
			protected void doExecute() {
				Class subclass = model2.createOwnedClass("Subclass", false);

				// add the dependency
				Model dependency = UML2Util.load(modelSet2, getTestResourceURI(DEPENDENCY_MODEL_NAME), UMLPackage.Literals.MODEL);
				model2.createPackageImport(dependency);
				subclass.createGeneralization((Class) dependency.getOwnedType("Superclass"));
				try {
					modelSet2.save(new NullProgressMonitor());
				} catch (IOException e) {
					Activator.log.error("The modelSet of the fixture failed on save", e); // $NON-NLS-1$
				}
			}
		};
		ted2.getCommandStack().execute(recording2);

		// dispose
		services1[0].disposeRegistry();
		services2[0].disposeRegistry();
	}

	protected ModelSet createModelSet(String name, ServicesRegistry[] outServices) throws Exception {

		outServices[0] = new ExtensionServicesRegistry(Activator.PLUGIN_ID);

		try {
			// start the ModelSet and its dependencies
			outServices[0].startServicesByClassKeys(ModelSet.class);
		} catch (ServiceMultiException e) {
			for (ServiceNotFoundException next : Iterables.filter(e.getExceptions(), ServiceNotFoundException.class)) {
				assertThat(next.getLocalizedMessage(), not(containsString("ModelSet")));
			}
		}

		ModelSet result = outServices[0].getService(ModelSet.class);

		assumeThat(result, instanceOf(CDOAwareModelSet.class));

		result.createModels(getTestResourceURI(name));

		// load blank model
		Resource uml = ((AbstractBaseModel) result.getModel(UmlModel.MODEL_ID)).getResource();
		Resource notation = ((AbstractBaseModel) result.getModel(NotationModel.MODEL_ID)).getResource();
		Resource di = ((AbstractBaseModel) result.getModel(DiModel.DI_MODEL_ID)).getResource();
		loadTemplate("empty", "model", uml, notation, di);

		return result;
	}

	protected IModelTransferNode getNode(IModelTransferConfiguration config, URI uri) {
		IModelTransferNode result = null;

		for (IModelTransferNode next : config.getModelsToTransfer()) {
			if (next.getPrimaryResourceURI().equals(uri)) {
				result = next;
				break;
			}
		}

		return result;
	}

}