MetricRulesTest.java example

Explorer
gmf-tooling-master
/*
 * Copyright (c) 2005, 2010 Borland Software Corporation and others
 * 
 * All rights reserved. This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License v1.0 which
 * accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors: Radek Dvorak (Borland) - initial API and implementation
 */
package org.eclipse.gmf.tests.rt;

import java.lang.reflect.Method;

import org.eclipse.emf.codegen.util.CodeGenUtil;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EClassifier;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.gmf.codegen.gmfgen.GenMetricContainer;
import org.eclipse.gmf.codegen.gmfgen.GenMetricRule;
import org.eclipse.gmf.mappings.MetricContainer;
import org.eclipse.gmf.tests.setup.RuntimeBasedGeneratorConfiguration;

/**
 * TODO: need to validate that not only calcXXX methods are present and work,
 * but also rest of the calculate[Domain|Diagram|Notation]ElementMetrics. 
 */
public class MetricRulesTest extends GeneratedCanvasTest {
	private MetricContainer metricContainer;
	private GenMetricContainer genMetricContainer;
	
	public MetricRulesTest(String name) {
		this(name, new RuntimeBasedGeneratorConfiguration());
	}
	
	public MetricRulesTest(String name, RuntimeBasedGeneratorConfiguration genConfig) {
		super(name, genConfig);
	}
	
	protected void setUp() throws Exception {
		super.setUp();
		this.metricContainer = getSetup().getMapModel().getMapping().getMetrics();
		this.genMetricContainer = getGenModel().getGenDiagram().getEditorGen().getMetrics();		
		assertTrue(
			"Setup with at least one metric definition required", //$NON-NLS-1$
			metricContainer != null && !metricContainer.getMetrics().isEmpty());
	}

	public void testMetrics() throws Exception {
		assertTrue(
				"Problem in gen-model transformation detected", //$NON-NLS-1$
				genMetricContainer != null && genMetricContainer.getMetrics().size() == metricContainer.getMetrics().size());
		Class<?> providerClass = null;
//		Class<?> viewClass = null;		
		try {
			providerClass = getSetup().loadGeneratedClass(getGenModel().getGenDiagram().getMetricProviderQualifiedClassName());		
			getSetup().loadGeneratedClass(getGenModel().getGenDiagram().getMetricProviderQualifiedClassName() + "$ResultView"); //$NON-NLS-1$
		} catch (ClassNotFoundException e) {
			fail("Could not find generated metric provider or view"); //$NON-NLS-1$
		}
			
//		Method metricAccessor = null;
//		try {
//			metricAccessor = providerClass.getMethod("getMetricKeys", (Class[]) null); //$NON-NLS-1$
//		} catch (Exception e) {			
//			fail("Could not find generated accessor for metric keys in " + providerClass.getName()); //$NON-NLS-1$			
//		}
//		
//		String[] metricKeys = (String[])metricAccessor.invoke(viewClass WTF??? - method was static!, (Object[]) null); 
//		assertEquals("All metrics from gmfmap must be registered", //$NON-NLS-1$
//			metricKeys.length, metricContainer.getMetrics().size());	
		assertMetricEvaluation(providerClass);
	}
	
	private void assertMetricEvaluation(Class<?> metricProviderClass) {
		Method calcMetricMethod = null;
		for (GenMetricRule nextRule : genMetricContainer.getMetrics()) {
			EClassifier eClassifier = nextRule.getTarget().getContext().getEcoreClassifier();
			EPackage ePackage = EPackage.Registry.INSTANCE.getEPackage(eClassifier.getEPackage().getNsURI());
			assertNotNull("EPackage must be already generated and initialized", ePackage); //$NON-NLS-1$
			EClassifier generatedClassifier = ePackage.getEClassifier(eClassifier.getName());
			assertNotNull("Could not find metric target classifier", generatedClassifier); //$NON-NLS-1$
			assertNotNull("PREREQ", generatedClassifier.getInstanceClass());

			try {
				calcMetricMethod = metricProviderClass.getMethod("calc" + CodeGenUtil.capName(CodeGenUtil.validJavaIdentifier(nextRule.getKey())), new Class[] { generatedClassifier.getInstanceClass() }); //$NON-NLS-1$
			} catch (Exception e) {
				e.printStackTrace();
				fail("Could not find calculateMetric method in the provider. " + e.toString()); //$NON-NLS-1$
			} 

			EObject contextInstance = ePackage.getEFactoryInstance().create((EClass)generatedClassifier); 
			try {
				Object metricValue = calcMetricMethod.invoke(null, new Object[] { contextInstance });
				assertTrue(metricValue instanceof Number);
			} catch (Exception e) {
				fail("Metric rule evaluation failed. " + e.toString()); //$NON-NLS-1$
			}
		}		
	}
}