tests/org.eclipse.ocl.examples.xtext.tests/src/org/eclipse/ocl/examples/pivot/tests/PivotFruitTestSuite.java

/*******************************************************************************
 * Copyright (c) 2011, 2018 Willink Transformations 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:
 *   L.Goubet, E.D.Willink - Initial API and implementation
 *******************************************************************************/

package org.eclipse.ocl.examples.pivot.tests;

import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EAttribute;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EClassifier;
import org.eclipse.emf.ecore.EEnum;
import org.eclipse.emf.ecore.EEnumLiteral;
import org.eclipse.emf.ecore.EFactory;
import org.eclipse.emf.ecore.EOperation;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.EReference;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.jdt.annotation.NonNull;
import org.eclipse.ocl.pivot.AssociationClass;
import org.eclipse.ocl.pivot.Operation;
import org.eclipse.ocl.pivot.Property;
import org.eclipse.ocl.pivot.internal.ecore.es2as.Ecore2AS;
import org.eclipse.ocl.pivot.internal.utilities.EnvironmentFactoryInternal;
import org.eclipse.ocl.pivot.utilities.NameUtil;
import org.eclipse.ocl.pivot.utilities.OCL;
import org.eclipse.uml2.uml.Association;

/**
 * Test suite with a simple Fruit model.
 */
public abstract class PivotFruitTestSuite extends PivotTestSuite
{
	protected EPackage fruitEPackage;
	protected EFactory fruitEFactory;

	protected org.eclipse.ocl.pivot.Package fruitPackage;

	protected EClass fruit;
	protected EOperation fruit_ripen;
	protected EOperation fruit_preferredColor;
	protected EOperation fruit_newFruit;
	protected EOperation fruit_setColor;
	protected EAttribute fruit_color;
	protected EAttribute fruit_name;
	//    protected EReference fruit_friends;

	protected EClass apple;
	protected Property apple_label;
	protected Property apple_tree;
	protected Property apple_appleFriends;
	protected Operation apple_labelOper;
	protected Operation apple_newApple;

	protected AssociationClass stem;
	protected Property stem_length;

	protected EClass tree;
	protected EReference tree_fruits;
	//	protected Property tree_apples;
	//    protected Property tree_height;
	protected EAttribute tree_name;

	protected EEnum color;
	protected EEnumLiteral color_black;
	protected EEnumLiteral color_red;
	protected EEnumLiteral color_green;
	protected EEnumLiteral color_yellow;
	protected EEnumLiteral color_orange;
	protected EEnumLiteral color_brown;
	protected EEnumLiteral color_pink;

	protected org.eclipse.ocl.pivot.Class forest;
	protected Property forest_trees;
	protected Property forest_trees_zoneQualifier;
	protected Property forest_trees_indexQualifier;
	protected Property forest_area;

	protected Association a_forest_tree;
	protected Property a_forest_tree_forest;

	protected org.eclipse.ocl.pivot.Class util;
	protected Property util_orderedSet;
	protected Property util_set;
	protected Property util_bag;
	protected Property util_sequence;
	protected Operation util_processOrderedSet;
	protected Operation util_processSet;
	protected Operation util_processBag;
	protected Operation util_processSequence;

	public PivotFruitTestSuite(boolean useCodeGen) {
		super(useCodeGen);
	}

	private EEnumLiteral getELiteral(EEnum eEnum, String name) {
		return eEnum.getEEnumLiteral(name);
	}

	private EOperation getEOperation(EClass eClass, String name, Object object, Object object2) {
		return NameUtil.getENamedElement(eClass.getEOperations(), name);
	}

	private EAttribute getEAttribute(EClass cls, String name, Object object) {
		return (EAttribute) cls.getEStructuralFeature(name);
	}

	private EReference getEReference(EClass cls, String name, Object object) {
		return (EReference) cls.getEStructuralFeature(name);
	}

	private EClassifier getEClassifier(EPackage ePackage, String name) {
		return ePackage.getEClassifier(name);
	}

	protected void initFruitPackage(@NonNull OCL ocl) {
		URI uri = getTestModelURI("models/ecore/Fruit.ecore");
		Resource ecoreResource = ocl.getResourceSet().getResource(uri, true);
		fruitEPackage = (EPackage)ecoreResource.getContents().get(0);
		fruitEFactory = fruitEPackage.getEFactoryInstance();
		Ecore2AS ecore2as = Ecore2AS.getAdapter(ecoreResource, (EnvironmentFactoryInternal) ocl.getEnvironmentFactory());
		fruitPackage = NameUtil.getNameable(ecore2as.getASModel().getOwnedPackages(), "fruit");

		fruit = (EClass) getEClassifier(fruitEPackage, "Fruit");
		fruit_ripen = getEOperation(fruit, "ripen", null, null);
		fruit_preferredColor = getEOperation(fruit, "preferredColor", null, null);
		fruit_newFruit = getEOperation(fruit, "newFruit", null, null);
		fruit_setColor = getEOperation(fruit, "setColor", null, null);
		fruit_color = getEAttribute(fruit, "color", null);
		fruit_name = getEAttribute(fruit, "name", null);
		//        fruit_friends = getEReference(fruit, "friends", null);

		apple = (EClass) getEClassifier(fruitEPackage, "Apple");
		/*		apple_label = getOwnedAttribute(apple, "label", null);
		apple_tree = getOwnedAttribute(apple, "tree", null);
        apple_appleFriends = getOwnedAttribute(apple, "appleFriends", null);
		apple_labelOper = getOwnedOperation(apple, "label", null, null);
		apple_newApple = getOwnedOperation(apple, "newApple", null, null);

//		stem = (AssociationClass) getOwnedType(fruitPackage, "Stem");
//		stem_length = getOwnedAttribute(stem, "length", null);
		 */
		tree = (EClass) getEClassifier(fruitEPackage, "Tree");
		tree_fruits = getEReference(tree, "fruits", null);
		//		tree_apples = getOwnedAttribute(tree, "apples", null);
		//		tree_height = getOwnedAttribute(tree, "height", null);
		tree_name = getEAttribute(tree, "name", null);

		color = (EEnum) getEClassifier(fruitEPackage, "Color");
		color_black = getELiteral(color, "black");
		color_red = getELiteral(color, "red");
		color_green = getELiteral(color, "green");
		color_yellow = getELiteral(color, "yellow");
		color_orange = getELiteral(color, "orange");
		color_brown = getELiteral(color, "brown");
		color_pink = getELiteral(color, "pink");
		/*
//		forest = (org.eclipse.ocl.pivot.Class) getOwnedType(fruitPackage, "Forest");
//		forest_trees = getOwnedAttribute(forest, "trees", null);
//		forest_trees_zoneQualifier = forest_trees.getQualifier("zone", null);
//		forest_trees_indexQualifier = forest_trees.getQualifier("index", null);
//        forest_area = getOwnedAttribute(forest, "area", null);

//		a_forest_tree = (Association) getOwnedType(fruitPackage, "A_Forest_Tree");
//		a_forest_tree_forest = a_forest_tree.getOwnedEnd("forest", null);

		util = (org.eclipse.ocl.pivot.Class) getOwnedType(fruitPackage, "FruitUtil");
		util_orderedSet = getOwnedAttribute(util, "orderedSet", null);
		util_set = getOwnedAttribute(util, "set", null);
		util_bag = getOwnedAttribute(util, "bag", null);
		util_sequence = getOwnedAttribute(util, "sequence", null);
		util_processOrderedSet = getOwnedOperation(util, "processOrderedSet", null, null);
		util_processSet = getOwnedOperation(util, "processSet", null, null);
		util_processBag = getOwnedOperation(util, "processBag", null, null);
		util_processSequence = getOwnedOperation(util, "processSequence", null, null);
		 */

		// convert the Package to Ecore for evaluation on instances
		//		fruitEPackage = UMLUtil.convertToEcore(fruitPackage, null).iterator().next();
		//		resourceSet.getPackageRegistry().put(fruitEPackage.getNsURI(), fruitEPackage);
		//		fruitEFactory = fruitEPackage.getEFactoryInstance();
		//		res.setTrackingModification(true);

		//		assertSame(
		//			fruitPackage,
		//			OCLUMLUtil.findPackage(
		//					Collections.singletonList(fruitPackage.getName()),
		//					resourceSet));
	}

	@Override
	protected void setUp() throws Exception {
		super.setUp();
		//		metamodelManager.addGlobalNamespace(PivotConstants.OCL_NAME, metamodelManager.getASmetamodel());
	}
}