tests/org.eclipse.ocl.examples.impactanalyzer.tests/src/org/eclipse/ocl/examples/impactanalyzer/tests/instanceScope/NavigationStepDebugGraphTest.java

/*******************************************************************************
 * Copyright (c) 2009, 2018 SAP AG and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v2.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v20.html
 *
 * Contributors:
 *     SAP AG - initial API and implementation
 ******************************************************************************/
package org.eclipse.ocl.examples.impactanalyzer.tests.instanceScope;

import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

import org.eclipse.emf.common.notify.Notification;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.util.EcoreUtil;
import org.eclipse.ocl.ecore.OCLExpression;
import org.eclipse.ocl.examples.impactanalyzer.ImpactAnalyzerFactory;
import org.eclipse.ocl.examples.impactanalyzer.benchmark.preparation.notifications.NotificationHelper;
import org.eclipse.ocl.examples.impactanalyzer.testutils.BaseDepartmentTestWithOCL;
import org.eclipse.ocl.examples.impactanalyzer.util.OCLFactory;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import company.CompanyFactory;
import company.Department;
import company.impl.DepartmentImpl;


public class NavigationStepDebugGraphTest extends BaseDepartmentTestWithOCL  {

	private Department dep1;
	private Department dep2;
	private Department dep3;

	private static final String recursiveBudgetCalculation = "context Department \n"+
			"inv: self.sumBudget() < 100000";

	protected OCLExpression recursiveBudgetCalculationAST = null;

	@Override
	@Before
	public void setUp() {
		super.setUp();
		createInstances();

		this.recursiveBudgetCalculationAST = (OCLExpression) parse( recursiveBudgetCalculation, this.comp).iterator()
				.next().getSpecification().getBodyExpression();

		if (this.comp.eResource() != null) {
			this.comp.eResource().getContents().add(this.recursiveBudgetCalculationAST);
		}

	}

	@Override
	@After
	public void tearDown() {
		EcoreUtil.delete(this.dep1);
		EcoreUtil.delete(this.dep2);
		EcoreUtil.delete(this.dep3);
		super.tearDown();
	}

	@Test
	public void testRecursiveOcl(){
		debugPrintln("Test recursive ocl operation call");
		Set<OCLExpression> stmts = new HashSet<OCLExpression>();

		stmts.add(this.recursiveBudgetCalculationAST);

		Notification noti = NotificationHelper.createAttributeChangeNotification(this.dep3, this.departmentBudget, this.dep3.getBudget(), this.dep3.getBudget() + 20);
		Collection<EObject> instances = ImpactAnalyzerFactory.INSTANCE.createImpactAnalyzer(this.recursiveBudgetCalculationAST, this.dep1.eClass(), /* notifyOnNewContextElements */ false,
				OCLFactory.getInstance()).getContextObjects(noti);

		compareInstances(instances, new EObject[] { this.dep1, this.dep2, this.dep3 });
	}

	private void createInstances() {

		this.dep1 = CompanyFactory.eINSTANCE.createDepartment();
		this.dep1.setName("Dep1");
		this.dep1.setBudget(1200);
		this.dep1.setMaxJuniors(5);

		this.dep2 = CompanyFactory.eINSTANCE.createDepartment();
		this.dep2.setName("Dep2");
		this.dep2.setBudget(1000);
		this.dep2.setMaxJuniors(5);

		dep2.setParentDepartment(dep1);
		dep1.getSubDepartment().add(dep2);

		this.dep3 = CompanyFactory.eINSTANCE.createDepartment();
		this.dep3.setName("Dep3");
		this.dep3.setBudget(1000);
		this.dep3.setMaxJuniors(5);

		this.dep3.setParentDepartment(dep2);
		this.dep2.getSubDepartment().add(dep3);

		this.allDepartments.add((DepartmentImpl)dep1);
		if (this.comp.eResource() != null) {
			this.comp.eResource().getContents().add(dep1);
		}
	}

	/**
	 * @param instances
	 * @param expectedInstances
	 */
	private void compareInstances(Collection<EObject> instances, EObject[] expectedInstances) {

		Set<EObject> expected = new HashSet<EObject>();
		for (int i = 0; i < expectedInstances.length; i++) {
			expected.add(expectedInstances[i]);
		}

		if (instances.containsAll(expected) && instances.size() == expected.size()) {
			debugPrintln(">> SUCCESS exact match\n");
		} else if (instances.containsAll(expected) && instances.size() != expected.size()) {
			debugPrintln(">> SUCCESS \n excess context instances:\n");
			instances.removeAll(expected);
			for (Iterator<EObject> i = instances.iterator(); i.hasNext();) {
				debugPrintln(i.next() + "\n");
			}
		} else {
			debugPrintln(">> FAILURE\n");
			expected.removeAll(instances);
			for (EObject o : expected) {
				debugPrint(o + "not in result");
			}
			debugPrintln("");
			fail("Returned instances does not match expected instances!");
		}
	}
}