tests/org.eclipse.ocl.examples.testutils/src/org/eclipse/ocl/examples/testutils/BaseDepartmentTest.java - ocl/org.eclipse.ocl - Git at Google

 /*******************************************************************************
  * 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.testutils;

 import java.util.HashSet;
 import java.util.Set;

 import org.eclipse.emf.ecore.EAttribute;
 import org.eclipse.emf.ecore.EClass;
 import org.eclipse.emf.ecore.EPackage;
 import org.eclipse.emf.ecore.EReference;
 import org.eclipse.emf.ecore.resource.ResourceSet;
 import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
 import org.eclipse.emf.ecore.util.ECrossReferenceAdapter;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Ignore;

 import company.CompanyFactory;
 import company.CompanyPackage;
 import company.Employee;
 import company.Freelance;
 import company.impl.DepartmentImpl;
 import company.impl.DivisionImpl;
 import company.impl.EmployeeImpl;
 import company.impl.FreelanceImpl;


 /**
  * This is the super class for all tests based on the Department model.
  */
 @Ignore
 public class BaseDepartmentTest extends BaseTest {

 	protected ResourceSet rs = null;
 	/**
 	 *
 	 */
 	protected EPackage companyPackage = null;

 	/**
 	 * the package containing the Company/Department meta model
 	 */
 	protected CompanyPackage comp = null;

 	/**
 	 * the set for all instances of {@link DepartmentImpl}
 	 */
 	protected Set<DepartmentImpl> allDepartments = new HashSet<DepartmentImpl>();

 	/**
 	 * the set of all instances of {@link FreelanceImpl}
 	 */
 	protected Set<FreelanceImpl> allFreelances = new HashSet<FreelanceImpl>();

 	/**
 	 * the set of all instances of {@link EmployeeImpl}
 	 */
 	protected Set<EmployeeImpl> allEmployees = new HashSet<EmployeeImpl>();

 	/**
 	 * a ID used to create unique names for employees and freelances
 	 */
 	protected int curImployeeID = 0;

 	/**
 	 * a ID used to create unique names for departments
 	 */
 	protected int curDepartmentID = 0;

 	/**
 	 * a instances of {@link EmployeeImpl}
 	 */
 	protected EmployeeImpl aEmployee = null;

 	/**
 	 * a instance of {@link DepartmentImpl}
 	 */
 	protected DepartmentImpl aDepartment = null;

 	/**
 	 * an instance of {@link DivisionImpl}
 	 */
 	protected DivisionImpl aDivision = null;

 	/**
 	 * a instance of {@link FreelanceImpl}
 	 */
 	protected FreelanceImpl aFreelance = null;

 	/*
 	 * for easy access to the model
 	 */
 	protected EClass companyClass = null;

 	protected EClass division = null;

 	protected EAttribute divisionBudget = null;

 	protected EClass department = null;

 	protected EAttribute departmentName = null;

 	protected EAttribute departmentMaxJuniors = null;

 	protected EAttribute departmentBudget = null;

 	protected EClass employee = null;

 	protected EAttribute employeeName = null;

 	protected EAttribute employeeAge = null;

 	protected EAttribute employeeSalary = null;

 	protected EReference employeeSecretary = null;

 	protected EAttribute employeeIsSecretary = null;

 	protected EClass freelance = null;

 	protected EClass student = null;

 	protected EAttribute freelanceAssignment = null;

 	protected EReference departmentRef = null;

 	protected EReference departmentEmployeeOfTheMonth = null;

 	protected EReference divisionEmployeesOfTheMonth = null;

 	protected EAttribute numberEmployeesOfTheMonth = null;

 	protected EAttribute eotmDelta = null;

 	protected EReference bossRef = null;

 	protected EReference managedRef = null;

 	protected EReference employerRef = null;

 	protected EReference employeeRef = null;

 	protected EReference directedRef = null;

 	protected EReference internRef = null;

 	@Override
 	@Before
 	public void setUp() {
 		beforeTestMethod();
 	}

 	@Override
 	@After
 	public void tearDown() {
 		this.resetInstances();
 	}

 	protected void beforeTestMethod() {
 		// build up the test model
 		buildModel();
 	}

 	/**
 	 * creates a whole bunch of instances
 	 *
 	 * @param numDepartments
 	 *            the number of departments
 	 * @param numEmployees
 	 *            the number of employees (not freelance) per department
 	 * @param numFreelance
 	 *            the number of freelance per department
 	 */
 	protected void createInstances(double numDepartments, int numEmployees, int numFreelance) {

 		int maxNumJuniors = 3;
 		int budget = 50000;

 		this.aDivision = (DivisionImpl) CompanyFactory.eINSTANCE.createDivision();
 		this.aDivision.setName("The super Division");
 		this.aDivision.setBudget(2000000);
 		for (double i = 0; i < numDepartments; i++) {
 			createDepartment(numEmployees, numFreelance, maxNumJuniors, budget);
 		}
 		// pick some instances to which the events will be related
 		this.aDepartment = this.allDepartments.iterator().next();
 		this.aDivision.getDepartment().add(this.aDepartment);
 		this.aEmployee = this.allEmployees.iterator().next();
 		this.aFreelance = this.allFreelances.iterator().next();

 	}

 	/**
 	 * This method fetches some meta object form the model which are used to create ModelChangeEvents later on
 	 */
 	private void buildModel() {
 		this.comp = company.CompanyPackage.eINSTANCE;
 		this.companyClass = this.comp.getCompany();
 		this.division = this.comp.getDivision();
 		this.divisionBudget = (EAttribute) this.division.getEStructuralFeature("budget");
 		this.department = this.comp.getDepartment();
 		this.departmentName = (EAttribute) this.department.getEStructuralFeature("name");
 		this.departmentMaxJuniors = (EAttribute) this.department.getEStructuralFeature("maxJuniors");
 		this.departmentBudget = (EAttribute) this.department.getEStructuralFeature("budget");
 		this.bossRef = (EReference) this.department.getEStructuralFeature("boss");
 		this.employeeRef = (EReference) this.department.getEStructuralFeature("employee");
 		this.departmentRef = (EReference) this.division.getEStructuralFeature("department");
 		this.departmentEmployeeOfTheMonth = (EReference) this.department.getEStructuralFeature("employeeOfTheMonth");
 		this.divisionEmployeesOfTheMonth = (EReference) this.division.getEStructuralFeature("employeesOfTheMonth");
 		this.numberEmployeesOfTheMonth = (EAttribute) this.division.getEStructuralFeature("numberEmployeesOfTheMonth");
 		this.eotmDelta = (EAttribute) this.companyClass.getEStructuralFeature("eotmDelta");
 		this.employee = this.comp.getEmployee();
 		this.employeeName = (EAttribute) this.employee.getEStructuralFeature("name");
 		this.employeeAge = (EAttribute) this.employee.getEStructuralFeature("age");
 		this.employeeSalary = (EAttribute) this.employee.getEStructuralFeature("salary");
 		this.employerRef = (EReference) this.employee.getEStructuralFeature("employer");
 		this.employeeSecretary = (EReference) this.employee.getEStructuralFeature("secretary");
 		this.directedRef = (EReference) this.employee.getEStructuralFeature("directed");
 		this.managedRef = (EReference) this.employee.getEStructuralFeature("managed");
 		this.internRef = (EReference) this.employee.getEStructuralFeature("intern");
 		this.freelance = this.comp.getFreelance();
 		this.student = this.comp.getStudent();
 		this.freelanceAssignment = (EAttribute) this.freelance.getEStructuralFeature("assignment");
 		this.rs = new ResourceSetImpl();
 		this.rs.eAdapters().add(new ECrossReferenceAdapter());
 		this.rs.getResources().add(this.comp.eResource());
 	}

 	/**
 	 * Creates a Department instance
 	 *
 	 * @param employees
 	 *            number of employees which are not freelances in the department
 	 * @param freelances
 	 *            the number of freelances in the department
 	 * @param maxJuniors
 	 *            the value for the maxJunior attribute
 	 * @param budget
 	 *            the value for the budget attribute
 	 */
 	private DepartmentImpl createDepartment(int employees, int freelances, int maxNumJuniors, int budget) {

 		DepartmentImpl dep = (DepartmentImpl) CompanyFactory.eINSTANCE.createDepartment();
 		dep.setName("Dep" + this.curDepartmentID);
 		dep.setMaxJuniors(maxNumJuniors);
 		dep.setBudget(budget);
 		this.curDepartmentID++;
 		EmployeeImpl e = null;
 		FreelanceImpl f = null;
 		for (int i = 0; i < employees; i++) {
 			e = createEmployee();
 			dep.getEmployee().add(e);
 			e.setEmployer(dep);
 		}
 		for (int i = 0; i < freelances; i++) {
 			f = createFreelance();
 			dep.getEmployee().add(f);
 			f.setEmployer(dep);
 		}
 		this.allDepartments.add(dep);
 		if (this.comp.eResource() != null) {
 			this.comp.eResource().getContents().add(dep);
 		}
 		return dep;
 	}

 	/**
 	 * @return an instance of {@link Employee}
 	 */
 	private EmployeeImpl createEmployee() {

 		EmployeeImpl e = (EmployeeImpl) CompanyFactory.eINSTANCE.createEmployee();
 		e.setName("empl" + this.curImployeeID);
 		e.setAge(42);
 		e.setSalary(2345);
 		this.curImployeeID++;
 		this.allEmployees.add(e);
 		if (this.comp.eResource() != null) {
 			this.comp.eResource().getContents().add(e);
 		}
 		return e;
 	}

 	/**
 	 * @return a instances of {@link Freelance}
 	 */
 	private FreelanceImpl createFreelance() {

 		FreelanceImpl f = (FreelanceImpl) CompanyFactory.eINSTANCE.createFreelance();
 		f.setName("empl" + this.curImployeeID);
 		f.setAge(42);
 		f.setAssignment(7);
 		f.setSalary(2345);
 		this.curImployeeID++;
 		this.allFreelances.add(f);
 		if (this.comp.eResource() != null) {
 			this.comp.eResource().getContents().add(f);
 		}
 		return f;
 	}

 	private void resetInstances() {
 		this.rs = null;
 		this.companyPackage = null;
 		this.comp = null;
 		this.allDepartments = null;
 		this.allFreelances = null;
 		this.allEmployees = null;
 		this.aEmployee = null;
 		this.aDepartment = null;
 		this.aDivision = null;
 		this.aFreelance = null;
 		this.companyClass = null;
 		this.division = null;
 		this.divisionBudget = null;
 		this.department = null;
 		this.departmentName = null;
 		this.departmentMaxJuniors = null;
 		this.departmentBudget = null;
 		this.employee = null;
 		this.employeeName = null;
 		this.employeeAge = null;
 		this.employeeSalary = null;
 		this.employeeSecretary = null;
 		this.freelance = null;
 		this.student = null;
 		this.freelanceAssignment = null;
 		this.departmentRef = null;
 		this.departmentEmployeeOfTheMonth = null;
 		this.divisionEmployeesOfTheMonth = null;
 		this.numberEmployeesOfTheMonth = null;
 		this.eotmDelta = null;
 		this.bossRef = null;
 		this.managedRef = null;
 		this.employerRef = null;
 		this.employeeRef = null;
 		this.directedRef = null;
 	}
 }
	/*******************************************************************************
	* 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.testutils;

	import java.util.HashSet;
	import java.util.Set;

	import org.eclipse.emf.ecore.EAttribute;
	import org.eclipse.emf.ecore.EClass;
	import org.eclipse.emf.ecore.EPackage;
	import org.eclipse.emf.ecore.EReference;
	import org.eclipse.emf.ecore.resource.ResourceSet;
	import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
	import org.eclipse.emf.ecore.util.ECrossReferenceAdapter;
	import org.junit.After;
	import org.junit.Before;
	import org.junit.Ignore;

	import company.CompanyFactory;
	import company.CompanyPackage;
	import company.Employee;
	import company.Freelance;
	import company.impl.DepartmentImpl;
	import company.impl.DivisionImpl;
	import company.impl.EmployeeImpl;
	import company.impl.FreelanceImpl;


	/**
	* This is the super class for all tests based on the Department model.
	*/
	@Ignore
	public class BaseDepartmentTest extends BaseTest {

	protected ResourceSet rs = null;
	/**
	*
	*/
	protected EPackage companyPackage = null;

	/**
	* the package containing the Company/Department meta model
	*/
	protected CompanyPackage comp = null;

	/**
	* the set for all instances of {@link DepartmentImpl}
	*/
	protected Set<DepartmentImpl> allDepartments = new HashSet<DepartmentImpl>();

	/**
	* the set of all instances of {@link FreelanceImpl}
	*/
	protected Set<FreelanceImpl> allFreelances = new HashSet<FreelanceImpl>();

	/**
	* the set of all instances of {@link EmployeeImpl}
	*/
	protected Set<EmployeeImpl> allEmployees = new HashSet<EmployeeImpl>();

	/**
	* a ID used to create unique names for employees and freelances
	*/
	protected int curImployeeID = 0;

	/**
	* a ID used to create unique names for departments
	*/
	protected int curDepartmentID = 0;

	/**
	* a instances of {@link EmployeeImpl}
	*/
	protected EmployeeImpl aEmployee = null;

	/**
	* a instance of {@link DepartmentImpl}
	*/
	protected DepartmentImpl aDepartment = null;

	/**
	* an instance of {@link DivisionImpl}
	*/
	protected DivisionImpl aDivision = null;

	/**
	* a instance of {@link FreelanceImpl}
	*/
	protected FreelanceImpl aFreelance = null;

	/*
	* for easy access to the model
	*/
	protected EClass companyClass = null;

	protected EClass division = null;

	protected EAttribute divisionBudget = null;

	protected EClass department = null;

	protected EAttribute departmentName = null;

	protected EAttribute departmentMaxJuniors = null;

	protected EAttribute departmentBudget = null;

	protected EClass employee = null;

	protected EAttribute employeeName = null;

	protected EAttribute employeeAge = null;

	protected EAttribute employeeSalary = null;

	protected EReference employeeSecretary = null;

	protected EAttribute employeeIsSecretary = null;

	protected EClass freelance = null;

	protected EClass student = null;

	protected EAttribute freelanceAssignment = null;

	protected EReference departmentRef = null;

	protected EReference departmentEmployeeOfTheMonth = null;

	protected EReference divisionEmployeesOfTheMonth = null;

	protected EAttribute numberEmployeesOfTheMonth = null;

	protected EAttribute eotmDelta = null;

	protected EReference bossRef = null;

	protected EReference managedRef = null;

	protected EReference employerRef = null;

	protected EReference employeeRef = null;

	protected EReference directedRef = null;

	protected EReference internRef = null;

	@Override
	@Before
	public void setUp() {
	beforeTestMethod();
	}

	@Override
	@After
	public void tearDown() {
	this.resetInstances();
	}

	protected void beforeTestMethod() {
	// build up the test model
	buildModel();
	}

	/**
	* creates a whole bunch of instances
	*
	* @param numDepartments
	* the number of departments
	* @param numEmployees
	* the number of employees (not freelance) per department
	* @param numFreelance
	* the number of freelance per department
	*/
	protected void createInstances(double numDepartments, int numEmployees, int numFreelance) {

	int maxNumJuniors = 3;
	int budget = 50000;

	this.aDivision = (DivisionImpl) CompanyFactory.eINSTANCE.createDivision();
	this.aDivision.setName("The super Division");
	this.aDivision.setBudget(2000000);
	for (double i = 0; i < numDepartments; i++) {
	createDepartment(numEmployees, numFreelance, maxNumJuniors, budget);
	}
	// pick some instances to which the events will be related
	this.aDepartment = this.allDepartments.iterator().next();
	this.aDivision.getDepartment().add(this.aDepartment);
	this.aEmployee = this.allEmployees.iterator().next();
	this.aFreelance = this.allFreelances.iterator().next();

	}

	/**
	* This method fetches some meta object form the model which are used to create ModelChangeEvents later on
	*/
	private void buildModel() {
	this.comp = company.CompanyPackage.eINSTANCE;
	this.companyClass = this.comp.getCompany();
	this.division = this.comp.getDivision();
	this.divisionBudget = (EAttribute) this.division.getEStructuralFeature("budget");
	this.department = this.comp.getDepartment();
	this.departmentName = (EAttribute) this.department.getEStructuralFeature("name");
	this.departmentMaxJuniors = (EAttribute) this.department.getEStructuralFeature("maxJuniors");
	this.departmentBudget = (EAttribute) this.department.getEStructuralFeature("budget");
	this.bossRef = (EReference) this.department.getEStructuralFeature("boss");
	this.employeeRef = (EReference) this.department.getEStructuralFeature("employee");
	this.departmentRef = (EReference) this.division.getEStructuralFeature("department");
	this.departmentEmployeeOfTheMonth = (EReference) this.department.getEStructuralFeature("employeeOfTheMonth");
	this.divisionEmployeesOfTheMonth = (EReference) this.division.getEStructuralFeature("employeesOfTheMonth");
	this.numberEmployeesOfTheMonth = (EAttribute) this.division.getEStructuralFeature("numberEmployeesOfTheMonth");
	this.eotmDelta = (EAttribute) this.companyClass.getEStructuralFeature("eotmDelta");
	this.employee = this.comp.getEmployee();
	this.employeeName = (EAttribute) this.employee.getEStructuralFeature("name");
	this.employeeAge = (EAttribute) this.employee.getEStructuralFeature("age");
	this.employeeSalary = (EAttribute) this.employee.getEStructuralFeature("salary");
	this.employerRef = (EReference) this.employee.getEStructuralFeature("employer");
	this.employeeSecretary = (EReference) this.employee.getEStructuralFeature("secretary");
	this.directedRef = (EReference) this.employee.getEStructuralFeature("directed");
	this.managedRef = (EReference) this.employee.getEStructuralFeature("managed");
	this.internRef = (EReference) this.employee.getEStructuralFeature("intern");
	this.freelance = this.comp.getFreelance();
	this.student = this.comp.getStudent();
	this.freelanceAssignment = (EAttribute) this.freelance.getEStructuralFeature("assignment");
	this.rs = new ResourceSetImpl();
	this.rs.eAdapters().add(new ECrossReferenceAdapter());
	this.rs.getResources().add(this.comp.eResource());
	}

	/**
	* Creates a Department instance
	*
	* @param employees
	* number of employees which are not freelances in the department
	* @param freelances
	* the number of freelances in the department
	* @param maxJuniors
	* the value for the maxJunior attribute
	* @param budget
	* the value for the budget attribute
	*/
	private DepartmentImpl createDepartment(int employees, int freelances, int maxNumJuniors, int budget) {

	DepartmentImpl dep = (DepartmentImpl) CompanyFactory.eINSTANCE.createDepartment();
	dep.setName("Dep" + this.curDepartmentID);
	dep.setMaxJuniors(maxNumJuniors);
	dep.setBudget(budget);
	this.curDepartmentID++;
	EmployeeImpl e = null;
	FreelanceImpl f = null;
	for (int i = 0; i < employees; i++) {
	e = createEmployee();
	dep.getEmployee().add(e);
	e.setEmployer(dep);
	}
	for (int i = 0; i < freelances; i++) {
	f = createFreelance();
	dep.getEmployee().add(f);
	f.setEmployer(dep);
	}
	this.allDepartments.add(dep);
	if (this.comp.eResource() != null) {
	this.comp.eResource().getContents().add(dep);
	}
	return dep;
	}

	/**
	* @return an instance of {@link Employee}
	*/
	private EmployeeImpl createEmployee() {

	EmployeeImpl e = (EmployeeImpl) CompanyFactory.eINSTANCE.createEmployee();
	e.setName("empl" + this.curImployeeID);
	e.setAge(42);
	e.setSalary(2345);
	this.curImployeeID++;
	this.allEmployees.add(e);
	if (this.comp.eResource() != null) {
	this.comp.eResource().getContents().add(e);
	}
	return e;
	}

	/**
	* @return a instances of {@link Freelance}
	*/
	private FreelanceImpl createFreelance() {

	FreelanceImpl f = (FreelanceImpl) CompanyFactory.eINSTANCE.createFreelance();
	f.setName("empl" + this.curImployeeID);
	f.setAge(42);
	f.setAssignment(7);
	f.setSalary(2345);
	this.curImployeeID++;
	this.allFreelances.add(f);
	if (this.comp.eResource() != null) {
	this.comp.eResource().getContents().add(f);
	}
	return f;
	}

	private void resetInstances() {
	this.rs = null;
	this.companyPackage = null;
	this.comp = null;
	this.allDepartments = null;
	this.allFreelances = null;
	this.allEmployees = null;
	this.aEmployee = null;
	this.aDepartment = null;
	this.aDivision = null;
	this.aFreelance = null;
	this.companyClass = null;
	this.division = null;
	this.divisionBudget = null;
	this.department = null;
	this.departmentName = null;
	this.departmentMaxJuniors = null;
	this.departmentBudget = null;
	this.employee = null;
	this.employeeName = null;
	this.employeeAge = null;
	this.employeeSalary = null;
	this.employeeSecretary = null;
	this.freelance = null;
	this.student = null;
	this.freelanceAssignment = null;
	this.departmentRef = null;
	this.departmentEmployeeOfTheMonth = null;
	this.divisionEmployeesOfTheMonth = null;
	this.numberEmployeesOfTheMonth = null;
	this.eotmDelta = null;
	this.bossRef = null;
	this.managedRef = null;
	this.employerRef = null;
	this.employeeRef = null;
	this.directedRef = null;
	}
	}