tests/org.eclipse.dltk.ruby.formatter.tests/src/org/eclipse/dltk/ruby/formatter/tests/ScriptedTest.java

/*******************************************************************************
 * Copyright (c) 2008 xored software, Inc.
 *
 * 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:
 *     xored software, Inc. - initial API and Implementation (Alex Panchenko)
 *******************************************************************************/
package org.eclipse.dltk.ruby.formatter.tests;

import java.io.IOException;
import java.net.URL;

import junit.framework.TestCase;
import junit.framework.TestSuite;

import org.eclipse.dltk.compiler.util.Util;
import org.eclipse.dltk.utils.TextUtils;
import org.osgi.framework.Bundle;

public class ScriptedTest extends AbstractFormatterTest {

	private String input;
	private String expected;

	protected void runTest() throws Throwable {
		final String output = format(input);
		assertEquals(expected, output);
	}

	protected static Bundle getResourceBundle() {
		return RubyFormatterTestsPlugin.getDefault().getBundle();
	}

	protected static char[] readResource(String resourceName)
			throws IOException {
		final URL resource = getResourceBundle().getResource(resourceName);
		assertNotNull(resourceName + " is not found", resource); //$NON-NLS-1$
		return Util.getInputStreamAsCharArray(resource.openStream(), -1,
				AllTests.CHARSET);
	}

	private static final String TEST_MARKER = "====";
	private static final String RESPONSE_MARKER = "==";

	public TestSuite createScriptedSuite(String resourceName) {
		return createScriptedSuite(getClass().getName(), resourceName, 0);
	}

	/**
	 * @param resourceName
	 * @return
	 */
	public TestSuite createScriptedSuite(String suiteName, String resourceName,
			int beginTestIndex) {
		final TestSuite suite = new TestSuite(suiteName);
		try {
			final String content = new String(readResource(resourceName));
			final String[] lines = TextUtils.splitLines(content);
			String testName = "START";
			int testIndex = 0;
			int testBegin = 0;
			int responseBegin = -1;
			int i = 0;
			while (i < lines.length) {
				final String line = lines[i++];
				if (line.startsWith(TEST_MARKER)) {
					final int testEnd = i - 1;
					if (testEnd > testBegin) {
						if (responseBegin < 0) {
							throw new IllegalArgumentException(
									"No response marker - next test started on line "
											+ testEnd);
						}
						if (testIndex >= beginTestIndex) {
							suite.addTest(createTest(testName, lines,
									testBegin, responseBegin, testEnd));
						}
						++testIndex;
					}
					testBegin = i;
					responseBegin = -1;
					testName = line.substring(TEST_MARKER.length()).trim();
				} else if (line.startsWith(RESPONSE_MARKER)) {
					if (responseBegin >= 0) {
						throw new IllegalArgumentException(
								"Multiple response markers: line " + (i - 1)
										+ ", previous on line " + responseBegin);
					}
					responseBegin = i;
				}
			}
			if (lines.length > testBegin) {
				if (responseBegin < 0) {
					throw new IllegalArgumentException(
							"No response marker in last test");
				}
				if (testIndex >= beginTestIndex) {
					suite.addTest(createTest(testName, lines, testBegin,
							responseBegin, lines.length));
				}
			}
		} catch (final Throwable e) {
			suite.addTest(new TestCase(e.getClass().getName()) { //$NON-NLS-1$
						protected void runTest() throws Throwable {
							throw e;
						}
					});
		}
		return suite;
	}

	private ScriptedTest createTest(String testName, String[] lines,
			int testBegin, int responseBegin, final int testEnd)
			throws Exception {
		final String input = joinLines(lines, testBegin, responseBegin - 1);
		final String expected = joinLines(lines, responseBegin, testEnd);
		ScriptedTest test = (ScriptedTest) getClass().newInstance();
		test.setName(testName);
		test.input = input;
		test.expected = expected;
		return test;
	}

}