Added new test set for ATL.

1 files changed, 253 insertions, 0 deletions

diff --git a/tests/org.eclipse.m2m.atl.tests/src/org/eclipse/m2m/atl/tests/util/ModelUtils.java b/tests/org.eclipse.m2m.atl.tests/src/org/eclipse/m2m/atl/tests/util/ModelUtils.java
new file mode 100644
index 00000000..a7ed029a
--- /dev/null
+++ b/tests/org.eclipse.m2m.atl.tests/src/org/eclipse/m2m/atl/tests/util/ModelUtils.java
@@ -0,0 +1,253 @@
+/*******************************************************************************
+ * Copyright (c) 2007 Obeo.
+ * 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 - ATL tester
+ *******************************************************************************/
+package org.eclipse.m2m.atl.tests.util;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.StringWriter;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.Map;
+import java.util.Set;
+import java.util.concurrent.ConcurrentHashMap;
+
+import org.eclipse.core.runtime.NullProgressMonitor;
+import org.eclipse.emf.common.util.URI;
+import org.eclipse.emf.compare.diff.generic.DiffMaker;
+import org.eclipse.emf.compare.diff.metamodel.DiffFactory;
+import org.eclipse.emf.compare.diff.metamodel.DiffGroup;
+import org.eclipse.emf.compare.diff.metamodel.DiffModel;
+import org.eclipse.emf.compare.diff.metamodel.ModelInputSnapshot;
+import org.eclipse.emf.compare.match.api.MatchOptions;
+import org.eclipse.emf.compare.match.metamodel.MatchModel;
+import org.eclipse.emf.compare.match.service.MatchService;
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.emf.ecore.EPackage;
+import org.eclipse.emf.ecore.EStructuralFeature;
+import org.eclipse.emf.ecore.resource.Resource;
+import org.eclipse.emf.ecore.resource.ResourceSet;
+import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
+import org.eclipse.emf.ecore.xmi.XMLResource;
+import org.eclipse.emf.ecore.xmi.impl.EcoreResourceFactoryImpl;
+import org.eclipse.emf.ecore.xmi.impl.XMIResourceFactoryImpl;
+import org.eclipse.emf.ecore.xmi.impl.XMIResourceImpl;
+import org.eclipse.m2m.atl.tests.AtlTestPlugin;
+
+/**
+ * Utility class for models.
+ * 
+ * @author William Piers <a href="mailto:william.piers@obeo.fr">william.piers@obeo.fr</a>
+ */
+public final class ModelUtils {
+	/** Constant for the file encoding system property. */
+	private static final String ENCODING_PROPERTY = "file.encoding"; //$NON-NLS-1$
+
+	/**
+	 * Utility classes don't need to (and shouldn't) be instantiated.
+	 */
+	private ModelUtils() {
+		// prevents instantiation
+	}
+
+	/**
+	 * Loads a model from a {@link java.io.File File} in a given {@link ResourceSet}.
+	 * 
+	 * @param file
+	 *            {@link java.io.File File} containing the model to be loaded.
+	 * @param resourceSet
+	 *            The {@link ResourceSet} to load the model in.
+	 * @return The model loaded from the file.
+	 * @throws IOException
+	 *             If the given file does not exist.
+	 */
+	public static EObject load(File file, ResourceSet resourceSet) throws IOException {
+		return load(URI.createFileURI(file.getPath()), resourceSet);
+	}
+
+	/**
+	 * Loads a model from an {@link org.eclipse.emf.common.util.URI URI} in a given {@link ResourceSet}.
+	 * 
+	 * @param modelURI
+	 *            {@link org.eclipse.emf.common.util.URI URI} where the model is stored.
+	 * @param resourceSet
+	 *            The {@link ResourceSet} to load the model in.
+	 * @return The model loaded from the URI.
+	 * @throws IOException
+	 *             If the given file does not exist.
+	 */
+	public static EObject load(URI modelURI, ResourceSet resourceSet) throws IOException {
+		EObject result = null;
+
+		String fileExtension = modelURI.fileExtension();
+		if (fileExtension == null || fileExtension.length() == 0) {
+			fileExtension = Resource.Factory.Registry.DEFAULT_EXTENSION;
+		}
+
+		final Resource.Factory.Registry reg = Resource.Factory.Registry.INSTANCE;
+		final Object resourceFactory = reg.getExtensionToFactoryMap().get(fileExtension);
+		if (resourceFactory != null) {
+			resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put(fileExtension,
+					resourceFactory);
+		} else {
+			resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put(fileExtension,
+					new XMIResourceFactoryImpl());
+		}
+
+		final Resource modelResource = resourceSet.createResource(modelURI);
+		final Map options = new ConcurrentHashMap();
+		options.put(XMLResource.OPTION_ENCODING, System.getProperty(ENCODING_PROPERTY));
+		modelResource.load(options);
+		if (modelResource.getContents().size() > 0)
+			result = (EObject)modelResource.getContents().get(0);
+		return result;
+	}
+
+	private static Set getElementsByType(Resource extent,String type) {
+		Set ret = new HashSet();
+		for(Iterator i = extent.getAllContents(); i.hasNext(); ) {
+			EObject eo = (EObject)i.next();
+			if (eo.eClass().getName().equals(type))
+				ret.add(eo);
+		}
+		return ret;
+	}
+
+	/**
+	 * Metamodel register : allows to open/compare specific models
+	 * @param metamodelURI
+	 * @param resourceSet
+	 * @throws IOException
+	 */
+	public static void registerMetamodel(URI metamodelURI, ResourceSet resourceSet) throws IOException {
+		if (EPackage.Registry.INSTANCE.getEPackage(metamodelURI.toString()) != null) {
+			return;
+		}
+		Resource.Factory myEcoreFactory = new EcoreResourceFactoryImpl();
+		Resource mmExtent = myEcoreFactory.createResource(metamodelURI);
+		try {
+			mmExtent.load(new FileInputStream(metamodelURI.path()),Collections.EMPTY_MAP);
+		} catch (IOException e) {
+			e.printStackTrace();
+		} catch (Exception ex) {
+			ex.printStackTrace();
+		}
+
+		for(Iterator it = getElementsByType(mmExtent,"EPackage").iterator() ; it.hasNext() ; ) {
+			EPackage p = (EPackage)it.next();
+			String nsURI = p.getNsURI();
+			if(nsURI == null) {
+				nsURI = p.getName();
+				p.setNsURI(nsURI);
+			}
+			EPackage.Registry.INSTANCE.put(nsURI, p);
+		}
+
+		for(Iterator it = getElementsByType(mmExtent,"EDataType").iterator(); it.hasNext(); ) {
+			EObject eo = (EObject)it.next();
+			EStructuralFeature sf;
+			sf = eo.eClass().getEStructuralFeature("name");	 
+			String tname = (String)eo.eGet(sf);			 
+			String icn = null;
+			if(tname.equals("Boolean"))
+				icn = "java.lang.Boolean";
+			else if(tname.equals("Double"))
+				icn = "java.lang.Double";
+			else if(tname.equals("Float"))
+				icn = "java.lang.Float";
+			else if(tname.equals("Integer"))
+				icn = "java.lang.Integer";
+			else if(tname.equals("String"))
+				icn = "java.lang.String";
+
+			if(icn != null) {
+				sf = eo.eClass().getEStructuralFeature("instanceClassName");
+				eo.eSet(sf, icn);                
+			}
+		}
+	}
+
+	/**
+	 * Saves a model as a file to the given path.
+	 * 
+	 * @param root
+	 *            Root of the objects to be serialized in a file.
+	 * @param path
+	 *            File where the objects have to be saved.
+	 * @throws IOException
+	 *             Thrown if an I/O operation has failed or been interrupted during the saving process.
+	 */
+	public static void save(EObject root, String path) throws IOException {
+		final URI modelURI = URI.createURI(path);
+		final ResourceSet resourceSet = new ResourceSetImpl();
+		resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put(
+				Resource.Factory.Registry.DEFAULT_EXTENSION, new XMIResourceFactoryImpl());
+		final Resource newModelResource = resourceSet.createResource(modelURI);
+		newModelResource.getContents().add(root);
+		final Map options = new ConcurrentHashMap();
+		options.put(XMLResource.OPTION_ENCODING, System.getProperty(ENCODING_PROPERTY));
+		newModelResource.save(options);
+	}
+
+	/**
+	 * Serializes the given EObjet as a String.
+	 * 
+	 * @param root
+	 *            Root of the objects to be serialized.
+	 * @return The given EObjet serialized as a String.
+	 * @throws IOException
+	 *             Thrown if an I/O operation has failed or been interrupted during the saving process.
+	 */
+	public static String serialize(EObject root) throws IOException {
+		if (root == null)
+			throw new NullPointerException("ModelUtils.NullSaveRoot"); //$NON-NLS-1$
+
+		final XMIResourceImpl newResource = new XMIResourceImpl();
+		final StringWriter writer = new StringWriter();
+		newResource.getContents().add(root);
+		newResource.save(writer, Collections.EMPTY_MAP);
+		return writer.toString();
+	}
+
+	/**
+	 * Compare two ecore files as models.
+	 * 
+	 * @param leftUri
+	 * @param rightUri
+	 * @throws IOException
+	 * @throws InterruptedException
+	 */
+	public static void compareModels(File leftUri, File rightUri, boolean ignoreIds, boolean delete) throws Exception {
+		EObject leftModel = ModelUtils.load(leftUri, AtlTestPlugin.getResourceSet());
+		EObject rightModel = ModelUtils.load(rightUri,AtlTestPlugin.getResourceSet());
+
+		Map options = new HashMap();
+		if (ignoreIds) {
+			options.put(MatchOptions.OPTION_IGNORE_XMI_ID, Boolean.TRUE);	
+		}
+		final MatchModel inputMatch = MatchService.doMatch(leftModel, rightModel, new NullProgressMonitor(), options);
+		final DiffModel inputDiff = new DiffMaker().doDiff(inputMatch);
+
+		if (((DiffGroup) inputDiff.getOwnedElements().get(0)).getSubchanges() != 0){
+			ModelInputSnapshot snapshot = DiffFactory.eINSTANCE.createModelInputSnapshot();
+			snapshot.setDiff(inputDiff);
+			snapshot.setMatch(inputMatch);
+			ModelUtils.save(snapshot, "file:/"+leftUri.toString()+".emfdiff");
+			throw new Exception("test failed : there are differences.");
+		}
+		if (delete) {
+			leftUri.delete();
+		}
+	}
+
+}