InstrumentationTestRunner.java revision ff1df69dd4835c177c724e1b5f1ba02d1f674047
19066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/*
29066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Copyright (C) 2007 The Android Open Source Project
39066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *
49066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Licensed under the Apache License, Version 2.0 (the "License");
59066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * you may not use this file except in compliance with the License.
69066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * You may obtain a copy of the License at
79066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *
89066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *      http://www.apache.org/licenses/LICENSE-2.0
99066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *
109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Unless required by applicable law or agreed to in writing, software
119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * distributed under the License is distributed on an "AS IS" BASIS,
129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * See the License for the specific language governing permissions and
149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * limitations under the License.
159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */
169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectpackage android.test;
189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport static android.test.suitebuilder.TestPredicates.REJECT_PERFORMANCE;
20ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
21ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport com.android.internal.util.Predicate;
22ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.app.Activity;
249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.app.Instrumentation;
259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.os.Bundle;
269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.os.Debug;
279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.os.Looper;
28ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport android.os.Parcelable;
29ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport android.os.PerformanceCollector;
30ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport android.os.Process;
31ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport android.os.SystemClock;
32ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport android.os.PerformanceCollector.PerformanceResultsWriter;
339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.TestMethod;
349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.TestPredicates;
359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.TestSuiteBuilder;
369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.util.Log;
379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
38ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.io.ByteArrayOutputStream;
39ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.io.File;
40ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.io.PrintStream;
41ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.lang.reflect.InvocationTargetException;
42ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.lang.reflect.Method;
43ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.util.ArrayList;
44ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.util.List;
459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.AssertionFailedError;
479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.Test;
489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestCase;
499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestListener;
509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestResult;
519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestSuite;
529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.runner.BaseTestRunner;
539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.textui.ResultPrinter;
549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/**
569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * An {@link Instrumentation} that runs various types of {@link junit.framework.TestCase}s against
579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * an Android package (application). Typical usage:
589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <ol>
599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Write {@link junit.framework.TestCase}s that perform unit, functional, or performance tests
609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * against the classes in your package.  Typically these are subclassed from:
61ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang *   <ul><li>{@link android.test.ActivityInstrumentationTestCase2}</li>
629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *   <li>{@link android.test.ActivityUnitTestCase}</li>
639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *   <li>{@link android.test.AndroidTestCase}</li>
649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *   <li>{@link android.test.ApplicationTestCase}</li>
659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *   <li>{@link android.test.InstrumentationTestCase}</li>
669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *   <li>{@link android.test.ProviderTestCase}</li>
679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *   <li>{@link android.test.ServiceTestCase}</li>
689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *   <li>{@link android.test.SingleLaunchActivityTestCase}</li></ul>
699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *   <li>In an appropriate AndroidManifest.xml, define the this instrumentation with
709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * the appropriate android:targetPackage set.
719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Run the instrumentation using "adb shell am instrument -w",
729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * with no optional arguments, to run all tests (except performance tests).
739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Run the instrumentation using "adb shell am instrument -w",
749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * with the argument '-e func true' to run all functional tests. These are tests that derive from
759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * {@link android.test.InstrumentationTestCase}.
769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Run the instrumentation using "adb shell am instrument -w",
779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * with the argument '-e unit true' to run all unit tests. These are tests that <i>do not</i>derive
789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * from {@link android.test.InstrumentationTestCase} (and are not performance tests).
799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Run the instrumentation using "adb shell am instrument -w",
809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * with the argument '-e class' set to run an individual {@link junit.framework.TestCase}.
819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * </ol>
829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running all tests:</b> adb shell am instrument -w
849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running all small tests:</b> adb shell am instrument -w
879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e size small
889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running all medium tests:</b> adb shell am instrument -w
919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e size medium
929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running all large tests:</b> adb shell am instrument -w
959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e size large
969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running a single testcase:</b> adb shell am instrument -w
999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e class com.android.foo.FooTest
1009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
1019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running a single test:</b> adb shell am instrument -w
1039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e class com.android.foo.FooTest#testFoo
1049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
1059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running multiple tests:</b> adb shell am instrument -w
1079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e class com.android.foo.FooTest,com.android.foo.TooTest
1089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
1099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Including performance tests:</b> adb shell am instrument -w
1119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e perf true
1129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
1139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>To debug your tests, set a break point in your code and pass:</b>
1159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e debug true
1169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>To run in 'log only' mode</b>
1189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e log true
119ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * This option will load and iterate through all test classes and methods, but will bypass actual
120ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * test execution. Useful for quickly obtaining info on the tests to be executed by an
1219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * instrumentation command.
1229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>To generate EMMA code coverage:</b>
1249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e coverage true
125ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * Note: this requires an emma instrumented build. By default, the code coverage results file
12651e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot * will be saved in a /data/<app>/coverage.ec file, unless overridden by coverageFile flag (see
12751e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot * below)
1289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b> To specify EMMA code coverage results file path:</b>
1309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e coverageFile /sdcard/myFile.ec
1319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <br/>
1329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * in addition to the other arguments.
1339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */
1349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
1359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/* (not JavaDoc)
1369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Although not necessary in most case, another way to use this class is to extend it and have the
137ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * derived class return the desired test suite from the {@link #getTestSuite()} method. The test
138ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * suite returned from this method will be used if no target class is defined in the meta-data or
139ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * command line argument parameters. If a derived class is used it needs to be added as an
140ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * instrumentation to the AndroidManifest.xml and the command to run it would look like:
1419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * adb shell am instrument -w com.android.foo/<i>com.android.FooInstrumentationTestRunner</i>
1439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Where <i>com.android.FooInstrumentationTestRunner</i> is the derived class.
1459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *
1469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * This model is used by many existing app tests, but can probably be deprecated.
1479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */
1489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectpublic class InstrumentationTestRunner extends Instrumentation implements TestSuiteProvider {
1499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
1509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /** @hide */
1519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String ARGUMENT_TEST_CLASS = "class";
1529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /** @hide */
1539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String ARGUMENT_TEST_PACKAGE = "package";
1549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /** @hide */
1559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String ARGUMENT_TEST_SIZE_PREDICATE = "size";
1569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /** @hide */
1579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String ARGUMENT_INCLUDE_PERF = "perf";
1589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /** @hide */
1599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String ARGUMENT_DELAY_MSEC = "delay_msec";
1609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
1619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String SMALL_SUITE = "small";
162ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    private static final String MEDIUM_SUITE = "medium";
1639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String LARGE_SUITE = "large";
164ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
1659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String ARGUMENT_LOG_ONLY = "log";
1669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
1679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
168ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * This constant defines the maximum allowed runtime (in ms) for a test included in the "small"
169ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * suite. It is used to make an educated guess at what suite an unlabeled test belongs.
1709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
1719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final float SMALL_SUITE_MAX_RUNTIME = 100;
172ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
1739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
174ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * This constant defines the maximum allowed runtime (in ms) for a test included in the
175ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * "medium" suite. It is used to make an educated guess at what suite an unlabeled test belongs.
1769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
1779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final float MEDIUM_SUITE_MAX_RUNTIME = 1000;
178ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
1799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
180ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * The following keys are used in the status bundle to provide structured reports to
181ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * an IInstrumentationWatcher.
1829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
1839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
1849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
185ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * This value, if stored with key {@link android.app.Instrumentation#REPORT_KEY_IDENTIFIER},
1869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * identifies InstrumentationTestRunner as the source of the report.  This is sent with all
1879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * status messages.
1889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
1899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_VALUE_ID = "InstrumentationTestRunner";
1909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
191ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
1929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * identifies the total number of tests that are being run.  This is sent with all status
1939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * messages.
1949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
1959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_NUM_TOTAL = "numtests";
1969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
197ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
1989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * identifies the sequence number of the current test.  This is sent with any status message
1999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * describing a specific test being started or completed.
2009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_NUM_CURRENT = "current";
2029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
203ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
2049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * identifies the name of the current test class.  This is sent with any status message
2059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * describing a specific test being started or completed.
2069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_NAME_CLASS = "class";
2089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
209ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
2109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * identifies the name of the current test.  This is sent with any status message
2119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * describing a specific test being started or completed.
2129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_NAME_TEST = "test";
2149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
215ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
2169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * reports the run time in seconds of the current test.
2179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String REPORT_KEY_RUN_TIME = "runtime";
2199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
220ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
2219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * reports the guessed suite assignment for the current test.
2229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String REPORT_KEY_SUITE_ASSIGNMENT = "suiteassignment";
2249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
22551e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
22651e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot     * identifies the path to the generated code coverage file.
22751e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot     */
22851e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot    private static final String REPORT_KEY_COVERAGE_PATH = "coverageFilePath";
22951e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot    /**
230ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
231ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * reports the cpu time in milliseconds of the current test.
232ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     */
233ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    private static final String REPORT_KEY_PERF_CPU_TIME =
234ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        "performance." + PerformanceCollector.METRIC_KEY_CPU_TIME;
235ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    /**
236ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
237ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * reports the run time in milliseconds of the current test.
238ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     */
239ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    private static final String REPORT_KEY_PERF_EXECUTION_TIME =
240ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        "performance." + PerformanceCollector.METRIC_KEY_EXECUTION_TIME;
241ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
242ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    /**
2439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * The test is starting.
2449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final int REPORT_VALUE_RESULT_START = 1;
2469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
2479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * The test completed successfully.
2489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final int REPORT_VALUE_RESULT_OK = 0;
2509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
2519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * The test completed with an error.
2529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final int REPORT_VALUE_RESULT_ERROR = -1;
2549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
2559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * The test completed with a failure.
2569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final int REPORT_VALUE_RESULT_FAILURE = -2;
2589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
259ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status bundle sent to an IInstrumentationWatcher, this key
260ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * identifies a stack trace describing an error or failure.  This is sent with any status
2619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * message describing a specific test being completed.
2629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_STACK = "stack";
2649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
26551e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot    // Default file name for code coverage
26651e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot    private static final String DEFAULT_COVERAGE_FILE_NAME = "coverage.ec";
267ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
2689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String LOG_TAG = "InstrumentationTestRunner";
2699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
2709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private final Bundle mResults = new Bundle();
2719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private AndroidTestRunner mTestRunner;
2729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean mDebug;
2739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean mJustCount;
2749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean mSuiteAssignmentMode;
2759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private int mTestCount;
2769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private String mPackageOfTests;
2779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean mCoverage;
2789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private String mCoverageFilePath;
2799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private int mDelayMsec;
2809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
2819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    @Override
2829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public void onCreate(Bundle arguments) {
2839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        super.onCreate(arguments);
2849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
2859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        // Apk paths used to search for test classes when using TestSuiteBuilders.
2869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String[] apkPaths =
2879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                {getTargetContext().getPackageCodePath(), getContext().getPackageCodePath()};
2889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        ClassPathPackageInfoSource.setApkPaths(apkPaths);
2899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
2909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        Predicate<TestMethod> testSizePredicate = null;
2919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        boolean includePerformance = false;
2929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String testClassesArg = null;
2939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        boolean logOnly = false;
2949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
2959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (arguments != null) {
2969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // Test class name passed as an argument should override any meta-data declaration.
2979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testClassesArg = arguments.getString(ARGUMENT_TEST_CLASS);
2989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mDebug = getBooleanArgument(arguments, "debug");
2999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mJustCount = getBooleanArgument(arguments, "count");
3009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mSuiteAssignmentMode = getBooleanArgument(arguments, "suiteAssignment");
3019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mPackageOfTests = arguments.getString(ARGUMENT_TEST_PACKAGE);
3029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testSizePredicate = getSizePredicateFromArg(
3039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    arguments.getString(ARGUMENT_TEST_SIZE_PREDICATE));
3049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            includePerformance = getBooleanArgument(arguments, ARGUMENT_INCLUDE_PERF);
3059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            logOnly = getBooleanArgument(arguments, ARGUMENT_LOG_ONLY);
3069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mCoverage = getBooleanArgument(arguments, "coverage");
3079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mCoverageFilePath = arguments.getString("coverageFile");
3089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            try {
3109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                Object delay = arguments.get(ARGUMENT_DELAY_MSEC);  // Accept either string or int
3119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                if (delay != null) mDelayMsec = Integer.parseInt(delay.toString());
3129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } catch (NumberFormatException e) {
3139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                Log.e(LOG_TAG, "Invalid delay_msec parameter", e);
3149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
3159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
3169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        TestSuiteBuilder testSuiteBuilder = new TestSuiteBuilder(getClass().getName(),
3189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                getTargetContext().getClassLoader());
3199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (testSizePredicate != null) {
3219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testSuiteBuilder.addRequirements(testSizePredicate);
3229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
3239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (!includePerformance) {
3249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testSuiteBuilder.addRequirements(REJECT_PERFORMANCE);
3259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
3269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (testClassesArg == null) {
3289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (mPackageOfTests != null) {
3299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                testSuiteBuilder.includePackages(mPackageOfTests);
3309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } else {
33161b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                TestSuite testSuite = getTestSuite();
33261b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                if (testSuite != null) {
33361b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                    testSuiteBuilder.addTestSuite(testSuite);
33461b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                } else {
335ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    // no package or class bundle arguments were supplied, and no test suite
33661b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                    // provided so add all tests in application
33761b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                    testSuiteBuilder.includePackages("");
33861b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                }
3399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
3409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else {
3419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            parseTestClasses(testClassesArg, testSuiteBuilder);
3429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
343ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
344da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob        testSuiteBuilder.addRequirements(getBuilderRequirements());
3459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestRunner = getAndroidTestRunner();
3479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestRunner.setContext(getTargetContext());
3487aba54b2edcb94f43c991c555b23fc1de364835eJack Wang        mTestRunner.setInstrumentation(this);
3499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestRunner.setSkipExecution(logOnly);
3509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestRunner.setTest(testSuiteBuilder.build());
3519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestCount = mTestRunner.getTestCases().size();
3529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (mSuiteAssignmentMode) {
3539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestRunner.addTestListener(new SuiteAssignmentPrinter());
3549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else {
355ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            WatcherResultPrinter resultPrinter = new WatcherResultPrinter(mTestCount);
3569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestRunner.addTestListener(new TestPrinter("TestRunner", false));
357ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestRunner.addTestListener(resultPrinter);
358ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestRunner.setPerformanceResultsWriter(resultPrinter);
3599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
3609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        start();
3619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
3629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
363da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob    List<Predicate<TestMethod>> getBuilderRequirements() {
364da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob        return new ArrayList<Predicate<TestMethod>>();
365da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob    }
366da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob
3679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
368ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * Parses and loads the specified set of test classes
369ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     *
3709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * @param testClassArg - comma-separated list of test classes and methods
3719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * @param testSuiteBuilder - builder to add tests to
3729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
3739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void parseTestClasses(String testClassArg, TestSuiteBuilder testSuiteBuilder) {
3749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String[] testClasses = testClassArg.split(",");
3759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        for (String testClass : testClasses) {
3769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            parseTestClass(testClass, testSuiteBuilder);
3779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
3789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
3799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
3819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * Parse and load the given test class and, optionally, method
382ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     *
383ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * @param testClassName - full package name of test class and optionally method to add.
384ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     *        Expected format: com.android.TestClass#testMethod
3859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * @param testSuiteBuilder - builder to add tests to
3869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
3879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void parseTestClass(String testClassName, TestSuiteBuilder testSuiteBuilder) {
3889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        int methodSeparatorIndex = testClassName.indexOf('#');
3899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String testMethodName = null;
3909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (methodSeparatorIndex > 0) {
3929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testMethodName = testClassName.substring(methodSeparatorIndex + 1);
3939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testClassName = testClassName.substring(0, methodSeparatorIndex);
3949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
395ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        testSuiteBuilder.addTestClassByName(testClassName, testMethodName, getTargetContext());
3969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
3979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    protected AndroidTestRunner getAndroidTestRunner() {
3999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return new AndroidTestRunner();
4009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
4019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
4029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean getBooleanArgument(Bundle arguments, String tag) {
4039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String tagString = arguments.getString(tag);
4049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return tagString != null && Boolean.parseBoolean(tagString);
4059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
406ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /*
4089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * Returns the size predicate object, corresponding to the "size" argument value.
4099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
4109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private Predicate<TestMethod> getSizePredicateFromArg(String sizeArg) {
411ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (SMALL_SUITE.equals(sizeArg)) {
4139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return TestPredicates.SELECT_SMALL;
4149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else if (MEDIUM_SUITE.equals(sizeArg)) {
4159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return TestPredicates.SELECT_MEDIUM;
4169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else if (LARGE_SUITE.equals(sizeArg)) {
4179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return TestPredicates.SELECT_LARGE;
4189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else {
4199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return null;
4209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
4219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
422ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    @Override
4249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public void onStart() {
4259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        Looper.prepare();
426ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (mJustCount) {
4289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResults.putString(Instrumentation.REPORT_KEY_IDENTIFIER, REPORT_VALUE_ID);
4299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResults.putInt(REPORT_KEY_NUM_TOTAL, mTestCount);
4309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            finish(Activity.RESULT_OK, mResults);
4319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else {
4329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (mDebug) {
4339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                Debug.waitForDebugger();
4349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
435ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
4379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            PrintStream writer = new PrintStream(byteArrayOutputStream);
4389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            try {
4399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                StringResultPrinter resultPrinter = new StringResultPrinter(writer);
440ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestRunner.addTestListener(resultPrinter);
442ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                long startTime = System.currentTimeMillis();
4449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestRunner.runTest();
4459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                long runTime = System.currentTimeMillis() - startTime;
446ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                resultPrinter.print(mTestRunner.getTestResult(), runTime);
4489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } finally {
449ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mResults.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
450ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        String.format("\nTest results for %s=%s",
451ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        mTestRunner.getTestClassName(),
4529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                        byteArrayOutputStream.toString()));
4539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
4549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                if (mCoverage) {
4559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    generateCoverageReport();
4569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                }
4579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                writer.close();
458ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                finish(Activity.RESULT_OK, mResults);
4609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
4619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
4629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
4639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
4649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public TestSuite getTestSuite() {
4659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return getAllTests();
4669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
4679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
4689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
4699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * Override this to define all of the tests to run in your package.
4709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
4719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public TestSuite getAllTests() {
4729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return null;
4739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
4749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
4759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
4769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * Override this to provide access to the class loader of your package.
4779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
4789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public ClassLoader getLoader() {
4799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return null;
4809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
481ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void generateCoverageReport() {
4839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        // use reflection to call emma dump coverage method, to avoid
4849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        // always statically compiling against emma jar
48551e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot        String coverageFilePath = getCoverageFilePath();
48651e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot        java.io.File coverageFile = new java.io.File(coverageFilePath);
4879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        try {
4889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            Class emmaRTClass = Class.forName("com.vladium.emma.rt.RT");
489ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            Method dumpCoverageMethod = emmaRTClass.getMethod("dumpCoverageData",
4909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    coverageFile.getClass(), boolean.class, boolean.class);
491ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            dumpCoverageMethod.invoke(null, coverageFile, false, false);
49351e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            // output path to generated coverage file so it can be parsed by a test harness if
49451e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            // needed
49551e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            mResults.putString(REPORT_KEY_COVERAGE_PATH, coverageFilePath);
49651e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            // also output a more user friendly msg
49751e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            mResults.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
49851e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot                String.format("Generated code coverage data to %s", coverageFilePath));
4999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (ClassNotFoundException e) {
5009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError("Is emma jar on classpath?", e);
5019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (SecurityException e) {
5029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
5039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (NoSuchMethodException e) {
5049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
5059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (IllegalArgumentException e) {
5069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
5079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (IllegalAccessException e) {
5089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
5099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (InvocationTargetException e) {
5109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
5119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
5129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
5139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private String getCoverageFilePath() {
5159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (mCoverageFilePath == null) {
51651e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            return getTargetContext().getFilesDir().getAbsolutePath() + File.separator +
517ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                   DEFAULT_COVERAGE_FILE_NAME;
518ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        } else {
5199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return mCoverageFilePath;
5209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
5219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
5229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void reportEmmaError(Exception e) {
524ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        reportEmmaError("", e);
5259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
5269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void reportEmmaError(String hint, Exception e) {
5289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String msg = "Failed to generate emma coverage. " + hint;
5299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        Log.e(LOG_TAG, msg, e);
5309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mResults.putString(Instrumentation.REPORT_KEY_STREAMRESULT, "\nError: " + msg);
5319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
5329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    // TODO kill this, use status() and prettyprint model for better output
5349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private class StringResultPrinter extends ResultPrinter {
5359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public StringResultPrinter(PrintStream writer) {
5379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            super(writer);
5389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
5399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        synchronized void print(TestResult result, long runTime) {
5419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            printHeader(runTime);
5429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            printFooter(result);
5439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
5449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
545ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
547ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * This class sends status reports back to the IInstrumentationWatcher about
5489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * which suite each test belongs.
5499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
550ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    private class SuiteAssignmentPrinter implements TestListener {
551ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private Bundle mTestResult;
5539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private long mStartTime;
5549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private long mEndTime;
5559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private boolean mTimingValid;
556ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public SuiteAssignmentPrinter() {
5589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
559ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
5619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * send a status for the start of a each test, so long tests can be seen as "running"
5629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
5639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void startTest(Test test) {
5649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTimingValid = true;
565ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mStartTime = System.currentTimeMillis();
5669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
567ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
5699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#addError(Test, Throwable)
5709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
5719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void addError(Test test, Throwable t) {
5729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTimingValid = false;
5739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
5749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
5769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#addFailure(Test, AssertionFailedError)
5779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
5789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void addFailure(Test test, AssertionFailedError t) {
5799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTimingValid = false;
5809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
5819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
5839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#endTest(Test)
5849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
5859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void endTest(Test test) {
5869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            float runTime;
5879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            String assignmentSuite;
5889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mEndTime = System.currentTimeMillis();
5899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult = new Bundle();
5909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (!mTimingValid || mStartTime < 0) {
5929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                assignmentSuite = "NA";
5939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                runTime = -1;
5949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } else {
5959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                runTime = mEndTime - mStartTime;
596ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                if (runTime < SMALL_SUITE_MAX_RUNTIME
5979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                        && !InstrumentationTestCase.class.isAssignableFrom(test.getClass())) {
5989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    assignmentSuite = SMALL_SUITE;
5999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                } else if (runTime < MEDIUM_SUITE_MAX_RUNTIME) {
6009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    assignmentSuite = MEDIUM_SUITE;
6019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                } else {
6029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    assignmentSuite = LARGE_SUITE;
6039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                }
6049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
6059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // Clear mStartTime so that we can verify that it gets set next time.
6069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mStartTime = -1;
6079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
608ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
609ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    test.getClass().getName() + "#" + ((TestCase) test).getName()
6109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    + "\nin " + assignmentSuite + " suite\nrunTime: "
6119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    + String.valueOf(runTime) + "\n");
6129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putFloat(REPORT_KEY_RUN_TIME, runTime);
6139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putString(REPORT_KEY_SUITE_ASSIGNMENT, assignmentSuite);
6149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            sendStatus(0, mTestResult);
6169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
6179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
618ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
6209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * This class sends status reports back to the IInstrumentationWatcher
6219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
622ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    private class WatcherResultPrinter implements TestListener, PerformanceResultsWriter {
6239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private final Bundle mResultTemplate;
6249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        Bundle mTestResult;
6259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        int mTestNum = 0;
6269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        int mTestResultCode = 0;
6279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String mTestClass = null;
628ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        boolean mIsTimedTest = false;
629ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        long mCpuTime = 0;
630ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        long mExecTime = 0;
631ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public WatcherResultPrinter(int numTests) {
6339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResultTemplate = new Bundle();
6349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResultTemplate.putString(Instrumentation.REPORT_KEY_IDENTIFIER, REPORT_VALUE_ID);
6359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResultTemplate.putInt(REPORT_KEY_NUM_TOTAL, numTests);
6369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
637ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
639ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang         * send a status for the start of a each test, so long tests can be seen
640ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang         * as "running"
6419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
6429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void startTest(Test test) {
6439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            String testClass = test.getClass().getName();
644ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            String testName = ((TestCase)test).getName();
6459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult = new Bundle(mResultTemplate);
6469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putString(REPORT_KEY_NAME_CLASS, testClass);
647ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestResult.putString(REPORT_KEY_NAME_TEST, testName);
6489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putInt(REPORT_KEY_NUM_CURRENT, ++mTestNum);
6499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // pretty printing
6509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (testClass != null && !testClass.equals(mTestClass)) {
651ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
6529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                        String.format("\n%s:", testClass));
6539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestClass = testClass;
6549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } else {
6559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, "");
6569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
6579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // The delay_msec parameter is normally used to provide buffers of idle time
659ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // for power measurement purposes. To make sure there is a delay before and after
6609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // every test in a suite, we delay *after* every test (see endTest below) and also
661ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // delay *before* the first test. So, delay test1 delay test2 delay.
6629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            try {
6649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                if (mTestNum == 1) Thread.sleep(mDelayMsec);
6659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } catch (InterruptedException e) {
6669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                throw new IllegalStateException(e);
6679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
6689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            sendStatus(REPORT_VALUE_RESULT_START, mTestResult);
6709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResultCode = 0;
671ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
672ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mIsTimedTest = false;
673ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            try {
674ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                // Look for TimedTest annotation on both test class and test
675ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                // method
676ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mIsTimedTest = test.getClass().isAnnotationPresent(TimedTest.class) ||
677ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    test.getClass().getMethod(testName).isAnnotationPresent(TimedTest.class);
678ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            } catch (SecurityException e) {
679ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                throw new IllegalStateException(e);
680ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            } catch (NoSuchMethodException e) {
681ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                throw new IllegalStateException(e);
682ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            }
683ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
684ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            if (mIsTimedTest) {
685ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mExecTime = SystemClock.uptimeMillis();
686ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mCpuTime = Process.getElapsedCpuTime();
687ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            }
6889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
689ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
6919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#addError(Test, Throwable)
6929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
6939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void addError(Test test, Throwable t) {
6949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putString(REPORT_KEY_STACK, BaseTestRunner.getFilteredTrace(t));
6959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResultCode = REPORT_VALUE_RESULT_ERROR;
6969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // pretty printing
697ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
698ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                String.format("\nError in %s:\n%s",
699ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    ((TestCase)test).getName(), BaseTestRunner.getFilteredTrace(t)));
7009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
7019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
7029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
7039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#addFailure(Test, AssertionFailedError)
7049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
7059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void addFailure(Test test, AssertionFailedError t) {
7069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putString(REPORT_KEY_STACK, BaseTestRunner.getFilteredTrace(t));
7079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResultCode = REPORT_VALUE_RESULT_FAILURE;
7089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // pretty printing
709ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
710ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                String.format("\nFailure in %s:\n%s",
711ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    ((TestCase)test).getName(), BaseTestRunner.getFilteredTrace(t)));
7129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
7139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
7149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
7159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#endTest(Test)
7169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
7179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void endTest(Test test) {
718ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            if (mIsTimedTest) {
719ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mCpuTime = Process.getElapsedCpuTime() - mCpuTime;
720ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mExecTime = SystemClock.uptimeMillis() - mExecTime;
721ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putLong(REPORT_KEY_PERF_CPU_TIME, mCpuTime);
722ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putLong(REPORT_KEY_PERF_EXECUTION_TIME, mExecTime);
723ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            }
724ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
7259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (mTestResultCode == 0) {
7269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, ".");
7279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
7289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            sendStatus(mTestResultCode, mTestResult);
7299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
730ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            try { // Sleep after every test, if specified
7319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                Thread.sleep(mDelayMsec);
7329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } catch (InterruptedException e) {
7339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                throw new IllegalStateException(e);
7349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
7359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
7369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
737ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        public void writeBeginSnapshot(String label) {
738ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // Do nothing
739ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        }
740ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
741ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        public void writeEndSnapshot(Bundle results) {
742ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // Copy all snapshot data fields as type long into mResults, which
743ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // is outputted via Instrumentation.finish
744ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            for (String key : results.keySet()) {
745ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mResults.putLong(key, results.getLong(key));
746ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            }
747ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        }
748ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
749ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        public void writeStartTiming(String label) {
750ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // Do nothing
751ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        }
752ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
753ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        public void writeStopTiming(Bundle results) {
754ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // Copy results into mTestResult by flattening list of iterations,
755ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // which is outputted via WatcherResultPrinter.endTest
756ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            int i = 0;
757ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            for (Parcelable p :
758ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    results.getParcelableArrayList(PerformanceCollector.METRIC_KEY_ITERATIONS)) {
759ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                Bundle iteration = (Bundle)p;
760ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                String index = "performance.iteration" + i + ".";
761ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putString(index + PerformanceCollector.METRIC_KEY_LABEL,
762ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        iteration.getString(PerformanceCollector.METRIC_KEY_LABEL));
763ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putLong(index + PerformanceCollector.METRIC_KEY_CPU_TIME,
764ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        iteration.getLong(PerformanceCollector.METRIC_KEY_CPU_TIME));
765ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putLong(index + PerformanceCollector.METRIC_KEY_EXECUTION_TIME,
766ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        iteration.getLong(PerformanceCollector.METRIC_KEY_EXECUTION_TIME));
767ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                i++;
768ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            }
769ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        }
770ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
7719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        // TODO report the end of the cycle
7729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
7739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project}
774