libcore/tools/runner/java/vogar/target/TestRunner.java

/*
 * Copyright (C) 2009 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package vogar.target;

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintStream;
import java.util.Properties;
import vogar.TestProperties;

/**
 * Runs an action, in process on the target.
 */
public class TestRunner {

    protected final Properties properties;

    protected final String qualifiedName;
    protected final Class<?> testClass;
    protected final Class<?> runnerClass;
    protected final int monitorPort;

    protected TestRunner() {
        properties = loadProperties();
        qualifiedName = properties.getProperty(TestProperties.QUALIFIED_NAME);
        testClass = classProperty(TestProperties.TEST_CLASS, Object.class);
        runnerClass = classProperty(TestProperties.RUNNER_CLASS, Runner.class);
        monitorPort = Integer.parseInt(properties.getProperty(TestProperties.MONITOR_PORT));
    }

    protected static Properties loadProperties() {
        Properties properties = new Properties();
        try {
            InputStream propertiesStream = TestRunner.class.getResourceAsStream(
                    "/" + TestProperties.FILE);
            if (propertiesStream == null) {
                throw new RuntimeException(TestProperties.FILE + " missing!");
            }
            properties.load(propertiesStream);
            return properties;
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    private Class<?> classProperty(String propertyName, Class<?> superClass) {
        String className = properties.getProperty(propertyName);
        if (className == null) {
            throw new IllegalArgumentException("Could not find property for " +
                                               propertyName);
        }
        try {
            Class<?> klass = Class.forName(className);
            if (!superClass.isAssignableFrom(klass)) {
                throw new IllegalArgumentException(
                        className + " can not be assigned to " + Runner.class);
            }
            return klass;
        } catch (ClassNotFoundException e) {
            throw new RuntimeException(e);
        }
    }

    public void run() {
        final TargetMonitor monitor = new TargetMonitor();
        monitor.await(monitorPort);

        PrintStream monitorPrintStream = new PrintStream(System.out) {
            @Override public void print(String str) {
                super.print(str);
                monitor.output(str);
            }
        };
        System.setOut(monitorPrintStream);
        System.setErr(monitorPrintStream);

        Runner runner;
        try {
            runner = (Runner) runnerClass.newInstance();
        } catch (InstantiationException e) {
            throw new RuntimeException(e);
        } catch (IllegalAccessException e) {
            throw new RuntimeException(e);
        }
        runner.init(monitor, qualifiedName, testClass);
        runner.run(qualifiedName, testClass);

        monitor.close();
    }



    public static void main(String[] args) {
        if (args.length != 0) {
            throw new RuntimeException("TestRunner doesn't take arguments");
        }
        new TestRunner().run();
    }
}