plugins/org.eclipse.emf.henshin.tests/src/org/eclipse/emf/henshin/tests/interpreter/DebuggerTests.java

package org.eclipse.emf.henshin.tests.interpreter;

import static org.junit.Assert.assertEquals;

import java.util.List;

import org.eclipse.emf.ecore.EAttribute;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.henshin.interpreter.EGraph;
import org.eclipse.emf.henshin.interpreter.InterpreterFactory;
import org.eclipse.emf.henshin.interpreter.Match;
import org.eclipse.emf.henshin.interpreter.impl.Debugger;
import org.eclipse.emf.henshin.interpreter.impl.EGraphImpl;
import org.eclipse.emf.henshin.model.Node;
import org.eclipse.emf.henshin.model.Rule;
import org.eclipse.emf.henshin.model.compact.CModule;
import org.eclipse.emf.henshin.model.resource.HenshinResourceSet;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;

public class DebuggerTests {

	static String path;

	HenshinResourceSet res;
	EPackage pack;
	EClass account;
	EAttribute credit, iD;
	CModule mod;
	Debugger debugger;
	EGraph graph;
	EGraph expectedResult;

	@BeforeClass
	public static void globalSetUp() {
		path = "src/org/eclipse/emf/henshin/tests/interpreter/";

	}

	@Before
	public void localSetUp() {
		res = new HenshinResourceSet(path);
		mod = new CModule(res.getModule("bank.henshin"));
//		mod = CModule.loadFromFile(path+"bank.henshin");
		graph = new EGraphImpl(res.getResource("example-bank.xmi"));

		debugger = new Debugger(path);
		expectedResult = new EGraphImpl(res.getResource("example-result.xmi"));

		pack = res.registerDynamicEPackages("bank.ecore").get(0);
		account = (EClass) pack.getEClassifier("Account");
		credit = account.getEAllAttributes().get(1);
		iD = account.getEAllAttributes().get(0);
	}

	@Test
	public void undoRedoTest() {
		// Using a modified version of
		// https://www.eclipse.org/henshin/examples.php?example=bank for Testing
		// Creating a new account for Alice...
		String expectedAfterUndo;
		expectedAfterUndo = graph.toString().split("\\(")[1];

		graph = debugger.executeUnit(graph, mod.getModule(), "createAccount", "Alice", 5);
		// Transferring some Money...
		graph = debugger.executeUnit(graph, mod.getModule(), "transferMoney", "Alice", 1, 2, 50.0d);
		// Deleting all accounts of Charles...
		graph = debugger.executeUnit(graph, mod.getModule(), "deleteAllAccounts", "Charles");

		String graphString, expectedString;
		graphString = graph.toString().split("\\(")[1];
		expectedString = expectedResult.toString().split("\\(")[1];
		// Can only assert same Structure. Same Content needs to be checked.
		assertEquals(graphString, expectedString);

		graph = debugger.undo();
		graph = debugger.undo();
		graph = debugger.undo();
		graphString = graph.toString().split("\\(")[1];
		assertEquals(graphString, expectedAfterUndo);

		graph = debugger.redo();
		graph = debugger.redo();
		graph = debugger.redo();

		graphString = graph.toString().split("\\(")[1];
		assertEquals(graphString, expectedString);
	}

	@Test(expected = RuntimeException.class)
	public void undoRedoFailTest() {
		debugger.undo();
		debugger.redo();
	}

	@Test
	public void executeRuleOnCompleteMatchTest() {
		List<EObject> list = graph.getDomain(account, true);
		EObject eNode = null;
		for (EObject e : list) {
			int iDVal = (int) e.eGet(iD);
			if (iDVal == 2) {
				eNode = e;// This should be Account 2
				break;
			}
		}

		Rule r = (Rule) mod.getModule().getUnit("addCreditToRandomAccount2");
		Match match = InterpreterFactory.INSTANCE.createMatch(r, false);
		Node n = r.getLhs().getNodes().get(0); // The Rule only has this node.

		match.setNodeTarget(n, eNode);

		EGraph transformedGraph = debugger.executeRuleOnMatch(graph, mod.getModule(), "addCreditToRandomAccount2",
				match, 400.0);
		debugger.saveGraph(graph, "debuggerTest-result.xmi");

		list = transformedGraph.getDomain(account, true);
		eNode = null;
		for (EObject e : list) {
			double value = (double) e.eGet(credit);

			int iDVal = (int) e.eGet(iD);
			if (iDVal == 2) {
				assertEquals(value, 400.0, 0.01);
			}
		}
	}

	@Test
	public void executeRuleOnPartialMatchTest() {
		List<EObject> list = graph.getDomain(account, true);
		EObject eNode = null;
		for (EObject e : list) {
			int iDVal = (int) e.eGet(iD);
			if (iDVal == 2) {
				eNode = e;// This should be Account 2
				break;
			}
		}

		Rule r = (Rule) mod.getModule().getUnit("addCreditToRandomAccount");
		Match match = InterpreterFactory.INSTANCE.createMatch(r, false);
		Node n = r.getLhs().getNodes().get(0); // The Rule only has this node.

		match.setNodeTarget(n, eNode);

		EGraph transformedGraph = debugger.executeRuleOnMatch(graph, mod.getModule(), "addCreditToRandomAccount", match,
				400.0);
		debugger.saveGraph(graph, "debuggerTest-result.xmi");

		list = transformedGraph.getDomain(account, true);
		eNode = null;
		for (EObject e : list) {
			double value = (double) e.eGet(credit);

			int iDVal = (int) e.eGet(iD);
			if (iDVal == 2) {
				assertEquals(value, 400.0, 0.01);
			}
		}
	}

	@Test(expected = RuntimeException.class)
	public void executeRuleOnMatchStringGraphTest() {
		graph = debugger.executeRuleOnMatch("example-bank.xmi", mod.getModule(), "addCreditToRandomAccount", null,
				400.0);
	}

	@Test(expected = RuntimeException.class)
	public void executeRuleOnMatchStringModuleTest() {
		graph = debugger.executeRuleOnMatch(graph, "bank.henshin", "createAccount", null, "Alice", 5);
	}

	@Test
	public void executeRuleOnMatchAllStringsTest() {
		graph = debugger.executeRuleOnMatch("example-bank.xmi", "bank.henshin", "createAccount", null, "Alice", 5);
	}

	@Test(expected = RuntimeException.class)
	public void executeRuleOnMatchNoRuleTest() {
		graph = debugger.executeRuleOnMatch("example-bank.xmi", "bank.henshin", "cheesecake", null, "Alice", 5);
	}

	@Test(expected = RuntimeException.class)
	public void executeRuleOnMatchNotARuleTest() {
		graph = debugger.executeRuleOnMatch("example-bank.xmi", "bank.henshin", "independent", null, "Alice", 5);
	}
}