OoasTools

package org.momut.ooas.test.utils;

import java.io.ByteArrayInputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.nio.file.Files;

import java.nio.file.Paths;

import org.junit.Assert;

import org.momut.ooas.Compiler;

import org.momut.ooas.Compiler.ILogCallBack;

import org.momut.ooas.CompilerConfiguration.Backend;

/**

 * Base class for MoMuT Test classes

 * @author DemetzM

 * @author KrennW

 *

 */

public abstract class OoasTest {

	/// static stuff

	private static final String  s_COMPILER_TEMP_DIR = "ooasComp.temp.dir";

	private static final String  s_tempDir        = System.getProperty(OoasTest.s_COMPILER_TEMP_DIR);

	private static final boolean s_hasTempDir     = s_tempDir != null && !s_tempDir.isEmpty();

	protected static final String s_modelsDir = "org/momut/ooas/test/models";

	protected static final String s_standardDir = s_modelsDir + "/standard/";

	protected static final String s_complexDir = s_modelsDir + "/complex/";

	protected static File createTempDirectory(String dirName) throws IOException {

		return Files.createTempDirectory(dirName).toFile();

	}

	protected static File createTempDirectory(String prefix, String directoryName) throws IOException {

		final File file = new File(prefix);

		if (!file.exists())

			file.mkdirs();

		return Files.createTempDirectory(Paths.get(prefix), directoryName).toFile();

	}

	/**

	 * Creates an temporary directory either in the system's default

	 * temp directory or in the directory specified via the "momut.temp.dir"

	 * environment variable.

	 * @param clazz A class. Used to name the directory: "MoMuT_" + clazz.getSimpleName()

	 * @return file object pointing to the newly created temp. dir.

	 * @throws IOException

	 */

	protected static File createOutputDirectory(Class<?> clazz) throws IOException {

		final String dirName = "MoMuT_" + clazz.getSimpleName() + "_";

		if (s_hasTempDir)

			return createTempDirectory(s_tempDir, dirName);

		return createTempDirectory(dirName);

	}

	/// instance stuff

	public final String m_testName;

	public final String m_model;

	public OoasTest(String testName, String model) {

		m_testName = testName;

		m_model = model;

	}

	/** Run the OOAS Compiler and check whether there were any errors during compilation.

	 * @param outputDir

	 * @throws IOException */

	private static void runOoasCompiler(String testName, File ooasFile, Backend backend, File outputDir) throws IOException {

		final byte[] ooasInput = Files.readAllBytes(ooasFile.toPath());// ooasCode.getBytes("UTF-8");

		final FileOutputStream compilerFileOutput = new FileOutputStream(new File(outputDir, String.format("%s.compiler", testName.replace("/", "."))));

		final Compiler compiler = new Compiler(

				testName,

				new ByteArrayInputStream(ooasInput),

				compilerFileOutput,

				backend);

		final int errors = compiler.compile( new ILogCallBack(){

			@Override

			public void logError(boolean toConsole, String message) {

				System.out.println(message);

			}} );

		compilerFileOutput.flush();

		compilerFileOutput.close();

		if (errors != 0) {

			System.out.println("##Faulty OOAS produced for " + testName);

		}

		Assert.assertEquals(

				String.format("Errors during ooas compiler run: %s", compiler.toString()),

				0,

				errors);

	}

	public static void runOoasCompiler(String testName, File ooasFile, File outputDir, Backend backend) throws IOException {

		runOoasCompiler(testName, ooasFile, backend, outputDir);

	}

}