tests/org.eclipse.epsilon.flexmi.test/src/org/eclipse/epsilon/flexmi/test/FlexmiTests.java

/*********************************************************************
* Copyright (c) 2008 The University of York.
*
* This program and the accompanying materials are made
* available under the terms of the Eclipse Public License 2.0
* which is available at https://www.eclipse.org/legal/epl-2.0/
*
* SPDX-License-Identifier: EPL-2.0
**********************************************************************/
package org.eclipse.epsilon.flexmi.test;

import static org.junit.Assert.assertEquals;

import java.io.File;

import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EcorePackage;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.epsilon.common.util.FileUtil;
import org.eclipse.epsilon.emc.emf.InMemoryEmfModel;
import org.eclipse.epsilon.eol.EolEvaluator;
import org.eclipse.epsilon.flexmi.FlexmiResource;
import org.eclipse.epsilon.flexmi.FlexmiResourceFactory;
import org.junit.Before;

public abstract class FlexmiTests {
	
	@Before
	public void setup()  throws Exception {
		// Load imported files
		FileUtil.getFile("models/templates/subdir/template-importing-eol.flexmi", FlexmiTests.class);
		FileUtil.getFile("models/templates/imported-template.flexmi", FlexmiTests.class);
		FileUtil.getFile("models/templates/imported-templates.flexmi", FlexmiTests.class);
		FileUtil.getFile("models/templates/library.eol", FlexmiTests.class);
				
	}
	
	protected FlexmiResource loadResource(String filename) throws Exception {
		ResourceSet resourceSet = new ResourceSetImpl();
		resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put("flexmi", new FlexmiResourceFactory());
		resourceSet.getPackageRegistry().put(EcorePackage.eINSTANCE.getNsURI(), EcorePackage.eINSTANCE);
		File resourceFile;
		try {
			resourceFile = FileUtil.getFile("models/" + filename, FlexmiTestSuite.class);
		}
		catch (NullPointerException ex) {
			// Let tests for invalid paths have their joy
			resourceFile = new File("models/" + filename);
		}
		FlexmiResource resource = (FlexmiResource) resourceSet.createResource(URI.createURI(resourceFile.toURI().toString()));
		resource.load(null);
		return resource;
	}
	
	protected void assertEval(String expression, Object result, String modelFileName) throws Exception {
		InMemoryEmfModel model = new InMemoryEmfModel(loadResource(modelFileName));
		EolEvaluator evaluator = new EolEvaluator(model);
		assertEquals(result, evaluator.evaluate(expression));
	}
	
}