tests/org.eclipse.epsilon.egl.traceability.fine.test.acceptance/src/org/eclipse/epsilon/egl/engine/traceability/fine/test/acceptance/EglFineGrainedTraceabilityAcceptanceTest.java

/*******************************************************************************
 * Copyright (c) 2011 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/
 * 
 * Contributors:
 *     Louis Rose - initial API and implementation
 ******************************************************************************/
package org.eclipse.epsilon.egl.engine.traceability.fine.test.acceptance;

import static org.eclipse.epsilon.egl.util.FileUtil.FILE_SEP;
import java.io.File;
import java.util.Collections;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EcorePackage;
import org.eclipse.epsilon.common.util.FileUtil;
import org.eclipse.epsilon.egl.EglFileGeneratingTemplateFactory;
import org.eclipse.epsilon.egl.engine.traceability.fine.EglFineGrainedTraceContextAdaptor;
import org.eclipse.epsilon.egl.engine.traceability.fine.trace.Trace;
import org.eclipse.epsilon.egl.test.acceptance.AcceptanceTestUtil;
import org.eclipse.epsilon.emc.emf.EmfUtil;
import org.eclipse.epsilon.emc.emf.InMemoryEmfModel;
import org.eclipse.epsilon.eol.models.IModel;
import org.eclipse.epsilon.eol.models.java.JavaModel;
import org.eclipse.epsilon.test.util.ModelWithEolAssertions;

public class EglFineGrainedTraceabilityAcceptanceTest {

	protected static ModelWithEolAssertions trace;
	
	protected static void generateTrace(String egl, EObject root) throws Exception {
		generateTrace(egl, root, "Out.txt");
	}
	
	protected static void generateTrace(String egl, EObject root, String destination) throws Exception {
		runEgl(egl, root, destination);
	}
	
	protected static String getOutputPath() throws Exception {
		return FileUtil.createTempDir(EglFineGrainedTraceabilityAcceptanceTest.class.getSimpleName(), true).getAbsolutePath();
	}
	
	protected static boolean deleteOutputFile(String relativePath) throws Exception {
		return new File(getOutputPath() + FILE_SEP + relativePath).delete();
	}
	
	protected static String getAbsoluteOutputPathFor(String filename) throws Exception {
		final File generatedFile = new File(getOutputPath(), filename);
		return generatedFile.getAbsolutePath().replace("\\", "\\\\");
	}

	private static void runEgl(String egl, EObject root, String destination) throws Exception {
		deleteOutputFile(destination);
		AcceptanceTestUtil.generate(createTemplateFactory(), egl, destination, new InMemoryEmfModel("Ecore", EmfUtil.createResource(root), EcorePackage.eINSTANCE));
	}

	private static EglFileGeneratingTemplateFactory createTemplateFactory() throws Exception {
		final EglFileGeneratingTemplateFactory factory = new EglFileGeneratingTemplateFactory();
		factory.setOutputRoot(getOutputPath());
		createTraceWithAssertions(new EglFineGrainedTraceContextAdaptor().adapt(factory.getContext()));
		return factory;
	}
	
	private static void createTraceWithAssertions(Trace root) {
		IModel traceModel = new JavaModel("TraceModel", Collections.singleton(root), Collections.singleton(Trace.class));
		trace = new ModelWithEolAssertions(traceModel);
		trace.setVariable("trace", "Trace.all.first");
	}
}