MultiFilesCompilationTestHelper.java

/**
 * Copyright (c) 2011 - 2014, Lunifera GmbH (Gross Enzersdorf), Loetz KG (Heidelberg)
 * 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
 * 
 * Sources based on {@link org.eclipse.xtext.xbase.compiler.CompilationTestHelper}
 * 
 * Contributors: 
 * 		Florian Pirchner - Initial implementation
 */

package org.lunifera.dsl.entity.xtext.tests;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import org.eclipse.xtext.resource.FileExtensionProvider;
import org.eclipse.xtext.resource.XtextResourceSet;
import org.eclipse.xtext.util.IAcceptor;
import org.eclipse.xtext.xbase.compiler.CompilationTestHelper;
import org.eclipse.xtext.xbase.compiler.OnTheFlyJavaCompiler;
import org.eclipse.xtext.xbase.lib.Pair;
import org.junit.Assert;

import com.google.inject.Inject;
import com.google.inject.Provider;

@SuppressWarnings("restriction")
public class MultiFilesCompilationTestHelper extends CompilationTestHelper {

	@Inject
	private OnTheFlyJavaCompiler javaCompiler;

	@Inject
	private Provider<XtextResourceSet> resourceSetProvider;

	@Inject
	private FileExtensionProvider extensionProvider;

	/**
	 * Asserts that the expected code is generated for the given source. Assumes
	 * that
	 * 
	 * @param source
	 *            some valid source code written in the language under test
	 * @param expected
	 *            the expected Java source code.
	 * @throws IOException
	 */
	@SuppressWarnings("unchecked")
	public void assertCompilesFrom(final Map<String, CharSequence> expected,
			CharSequence... input) throws IOException {
		final boolean[] called = { false };

		List<Pair<String, CharSequence>> resources = new ArrayList<Pair<String, CharSequence>>();
		for (int i = 0; i < input.length; i++) {
			String fileName = "MyFile" + i + "."
					+ extensionProvider.getPrimaryFileExtension();
			resources.add(new Pair<String, CharSequence>(fileName, input[i]));
		}

		compile(resourceSet(resources.toArray(new Pair[resources.size()])),
				new IAcceptor<CompilationTestHelper.Result>() {
					public void accept(Result r) {
						for (Entry<String, CharSequence> entry : expected
								.entrySet()) {
							Assert.assertEquals(
									entry.getValue().toString(),
									r.getGeneratedCode(entry.getKey().replace(
											'/', '.')));
							called[0] = true;
						}
					}
				});
		Assert.assertTrue("Nothing was generated but the expectation was :\n"
				+ expected, called[0]);
	}

}