tests/org.eclipse.epsilon.hutn.xmi.test.integration/src/org/eclipse/epsilon/hutn/xmi/test/integration/HutnXmiBridgeIntegrationTest.java

/*******************************************************************************
 * Copyright (c) 2008 The University of York.
 * This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 * 
 * Contributors:
 *     Louis Rose - initial API and implementation
 ******************************************************************************
 *
 * $Id$
 */
package org.eclipse.epsilon.hutn.xmi.test.integration;

import static org.junit.Assert.fail;

import org.eclipse.epsilon.hutn.model.hutn.ClassObject;
import org.eclipse.epsilon.hutn.model.hutn.ContainmentSlot;
import org.eclipse.epsilon.hutn.model.hutn.PackageObject;
import org.eclipse.epsilon.hutn.model.hutn.Slot;
import org.eclipse.epsilon.hutn.model.hutn.Spec;
import org.eclipse.epsilon.hutn.test.model.HutnTestWithFamiliesMetaModel;
import org.eclipse.epsilon.hutn.xmi.HutnXmiBridgeException;
import org.eclipse.epsilon.hutn.xmi.Xmi2Hutn;

public abstract class HutnXmiBridgeIntegrationTest extends HutnTestWithFamiliesMetaModel {
	
	protected static Spec spec;
	
	protected static void integrationTestWithExceptions(String xmi) throws HutnXmiBridgeException {
		spec = new Xmi2Hutn(xmi).getSpec();
	}
	
	protected static void integrationTest(String xmi) {
		try {
			integrationTestWithExceptions(xmi);
		
		} catch (HutnXmiBridgeException e) {
			fail("Caught exception: " + e.getMessage());
			e.printStackTrace();
		}
	}
	
	protected static void integrationTestWithModelAsRoot() {
		integrationTestWithModelAsRoot("");
	}
	
	protected static void integrationTestWithModelAsRoot(String nestedXmi) {
		integrationTestWithModelAsRoot("", nestedXmi);
	}
	
	protected static void integrationTestWithModelAsRoot(String xmiAttributesForModel, String nestedXmi) {
		integrationTest(constructXmiForModel(xmiAttributesForModel, nestedXmi));
	}
	
	private static String constructXmiForModel(String attributes, String contents) {
		final StringBuilder builder = new StringBuilder();
		
		builder.append("<?xml version=\"1.0\" encoding=\"ASCII\"?>");
		builder.append('\n');
		
		builder.append("<families:Model xmi:version=\"2.0\" xmlns:xmi=\"http://www.omg.org/XMI\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xmlns:families=\"families\" xmi:id=\"_I6yJURhKEd6d_-caKAfnUw\" ");
		builder.append(attributes);
		builder.append(">");
		builder.append('\n');
		
		builder.append(contents);
		builder.append('\n');
		builder.append("</families:Model>");
		
		return builder.toString();
	}
	
	
	
	protected static PackageObject getPackageObject() {
		return spec.getObjects().get(0);
	}
	
	protected static ClassObject getModel() {
		return getPackageObject().getClassObjects().get(0);
	}
	
	protected static ContainmentSlot getFirstSlotOfModel() {
		return (ContainmentSlot)getModel().getSlots().get(0);
	}
	
	protected static ClassObject getFamily() {
		return getFirstSlotOfModel().getValues().get(0);
	}
	
	protected static Slot<?> getFirstSlotOfFamily() {
		return getFamily().getSlots().get(0);
	}
}