plugins/org.eclipse.ease.lang.unittest.ui/src/org/eclipse/ease/lang/unittest/ui/handlers/RunAllTests.java

/*******************************************************************************
 * Copyright (c) 2017 Christian Pontesegger 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:
 *     Christian Pontesegger - initial API and implementation
 *******************************************************************************/

package org.eclipse.ease.lang.unittest.ui.handlers;

import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.ease.lang.unittest.TestSuiteScriptEngine;
import org.eclipse.ease.lang.unittest.runtime.ITestContainer;
import org.eclipse.ease.lang.unittest.runtime.ITestSuite;
import org.eclipse.ease.lang.unittest.ui.views.UnitTestView;
import org.eclipse.ease.ui.console.ScriptConsole;

public class RunAllTests extends AbstractViewToolbarHandler {

	@Override
	public Object execute(ExecutionEvent event) throws ExecutionException {
		final UnitTestView view = getView(event, UnitTestView.class);
		if (view != null) {

			final Object input = view.getFileTreeViewer().getInput();
			if (input instanceof ITestContainer) {
				final ITestSuite testRoot = (ITestSuite) ((ITestContainer) input).getChildren().iterator().next();
				final TestSuiteScriptEngine engine = view.getCurrentEngine();

				view.notifyEngineCreation(engine);

				// prepare fresh console
				final ScriptConsole console = ScriptConsole.create(engine.getName() + ": " + testRoot.getResource(), engine);
				engine.setOutputStream(console.getOutputStream());
				engine.setErrorStream(console.getErrorStream());
				engine.setInputStream(console.getInputStream());

				engine.executeAsync(getTestRoot(testRoot, event));
				engine.schedule();
			}
		}

		return null;
	}

	/**
	 * Get the root object to be executed by the engine.
	 *
	 * @param testSuite
	 *            test suite
	 * @param event
	 *            event that triggered this handler
	 * @return object to be executed
	 */
	protected Object getTestRoot(ITestSuite testSuite, ExecutionEvent event) {
		return testSuite;
	}
}