InstrumentationTestRunner.java revision 8cc9a8e132ebbf852e055ed17ab91f35b02fd974
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
19ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport com.android.internal.util.Predicate;
2088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabotimport com.android.internal.util.Predicates;
21ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.app.Activity;
239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.app.Instrumentation;
249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.os.Bundle;
259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.os.Debug;
269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.os.Looper;
27ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport android.os.Parcelable;
28ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport android.os.PerformanceCollector;
29ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport android.os.PerformanceCollector.PerformanceResultsWriter;
309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.TestMethod;
319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.TestPredicates;
329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.TestSuiteBuilder;
3388e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabotimport android.test.suitebuilder.annotation.HasAnnotation;
340375fa12c7f94bfc9c840721dcc5215b6c7575d6Doug Zongkerimport android.test.suitebuilder.annotation.LargeTest;
359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.util.Log;
369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
37ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.io.ByteArrayOutputStream;
38ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.io.File;
39ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.io.PrintStream;
4088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabotimport java.lang.annotation.Annotation;
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/>
9888e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * <b>Filter test run to tests with given annotation:</b> adb shell am instrument -w
9988e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * -e annotation com.android.foo.MyAnnotation
10088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * com.android.foo/android.test.InstrumentationTestRunner
10188e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * <p/>
10288e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * If used with other options, the resulting test run will contain the union of the two options.
10388e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * e.g. "-e size large -e annotation com.android.foo.MyAnnotation" will run only tests with both
10488e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * the {@link LargeTest} and "com.android.foo.MyAnnotation" annotations.
10588e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * <p/>
10688e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * <b>Filter test run to tests <i>without</i> given annotation:</b> adb shell am instrument -w
10788e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * -e notAnnotation com.android.foo.MyAnnotation
10888e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * com.android.foo/android.test.InstrumentationTestRunner
10988e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot * <p/>
1109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running a single testcase:</b> adb shell am instrument -w
1119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e class com.android.foo.FooTest
1129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
1139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running a single test:</b> adb shell am instrument -w
1159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e class com.android.foo.FooTest#testFoo
1169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
1179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running multiple tests:</b> adb shell am instrument -w
1199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e class com.android.foo.FooTest,com.android.foo.TooTest
1209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
1219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
12289c0ef4aba57172bbc9288f2d22c943f220a3dd1Brett Chabot * <b>Running all tests in a java package:</b> adb shell am instrument -w
12389c0ef4aba57172bbc9288f2d22c943f220a3dd1Brett Chabot * -e package com.android.foo.subpkg
12489c0ef4aba57172bbc9288f2d22c943f220a3dd1Brett Chabot *  com.android.foo/android.test.InstrumentationTestRunner
12589c0ef4aba57172bbc9288f2d22c943f220a3dd1Brett Chabot * <p/>
1269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Including performance tests:</b> adb shell am instrument -w
1279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e perf true
1289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner
1299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>To debug your tests, set a break point in your code and pass:</b>
1319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e debug true
1329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>To run in 'log only' mode</b>
1349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e log true
135ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * This option will load and iterate through all test classes and methods, but will bypass actual
136ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * test execution. Useful for quickly obtaining info on the tests to be executed by an
1379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * instrumentation command.
1389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>To generate EMMA code coverage:</b>
1409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e coverage true
141ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * Note: this requires an emma instrumented build. By default, the code coverage results file
14251e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot * will be saved in a /data/<app>/coverage.ec file, unless overridden by coverageFile flag (see
14351e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot * below)
1449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b> To specify EMMA code coverage results file path:</b>
1469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e coverageFile /sdcard/myFile.ec
1479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <br/>
1489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * in addition to the other arguments.
1499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */
1509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
1519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/* (not JavaDoc)
1529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Although not necessary in most case, another way to use this class is to extend it and have the
153ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * derived class return the desired test suite from the {@link #getTestSuite()} method. The test
154ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * suite returned from this method will be used if no target class is defined in the meta-data or
155ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * command line argument parameters. If a derived class is used it needs to be added as an
156ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * instrumentation to the AndroidManifest.xml and the command to run it would look like:
1579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * adb shell am instrument -w com.android.foo/<i>com.android.FooInstrumentationTestRunner</i>
1599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/>
1609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Where <i>com.android.FooInstrumentationTestRunner</i> is the derived class.
1619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project *
1629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * This model is used by many existing app tests, but can probably be deprecated.
1639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */
1649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectpublic class InstrumentationTestRunner extends Instrumentation implements TestSuiteProvider {
1659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
1669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /** @hide */
1679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String ARGUMENT_TEST_CLASS = "class";
1689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /** @hide */
1699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String ARGUMENT_TEST_PACKAGE = "package";
1709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /** @hide */
1719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String ARGUMENT_TEST_SIZE_PREDICATE = "size";
1729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /** @hide */
1739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String ARGUMENT_DELAY_MSEC = "delay_msec";
1749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
1759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String SMALL_SUITE = "small";
176ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    private static final String MEDIUM_SUITE = "medium";
1779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String LARGE_SUITE = "large";
178ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
1799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String ARGUMENT_LOG_ONLY = "log";
18088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    /** @hide */
18188e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    static final String ARGUMENT_ANNOTATION = "annotation";
18288e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    /** @hide */
18388e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    static final String ARGUMENT_NOT_ANNOTATION = "notAnnotation";
1849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
1859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
186ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * This constant defines the maximum allowed runtime (in ms) for a test included in the "small"
187ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * suite. It is used to make an educated guess at what suite an unlabeled test belongs.
1889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
1899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final float SMALL_SUITE_MAX_RUNTIME = 100;
190ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
1919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
192ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * This constant defines the maximum allowed runtime (in ms) for a test included in the
193ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * "medium" suite. It is used to make an educated guess at what suite an unlabeled test belongs.
1949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
1959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final float MEDIUM_SUITE_MAX_RUNTIME = 1000;
196ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
1979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
198ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * The following keys are used in the status bundle to provide structured reports to
199ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * an IInstrumentationWatcher.
2009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
2029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
203ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * This value, if stored with key {@link android.app.Instrumentation#REPORT_KEY_IDENTIFIER},
2049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * identifies InstrumentationTestRunner as the source of the report.  This is sent with all
2059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * status messages.
2069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_VALUE_ID = "InstrumentationTestRunner";
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 total number of tests that are being run.  This is sent with all status
2119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * messages.
2129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_NUM_TOTAL = "numtests";
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     * identifies the sequence number of the current test.  This is sent with any status message
2179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * describing a specific test being started or completed.
2189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_NUM_CURRENT = "current";
2209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
221ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
2229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * identifies the name of the current test class.  This is sent with any status message
2239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * describing a specific test being started or completed.
2249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_NAME_CLASS = "class";
2269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
227ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
2289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * identifies the name of the current test.  This is sent with any status message
2299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * describing a specific test being started or completed.
2309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_NAME_TEST = "test";
2329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
233ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
2349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * reports the run time in seconds of the current test.
2359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String REPORT_KEY_RUN_TIME = "runtime";
2379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
238ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
2393fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang     * reports the number of total iterations of the current test.
2403fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang     */
2413fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang    private static final String REPORT_KEY_NUM_ITERATIONS = "numiterations";
2423fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang    /**
2433fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
2449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * reports the guessed suite assignment for the current test.
2459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String REPORT_KEY_SUITE_ASSIGNMENT = "suiteassignment";
2479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
24851e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot     * If included in the status or final bundle sent to an IInstrumentationWatcher, this key
24951e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot     * identifies the path to the generated code coverage file.
25051e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot     */
25151e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot    private static final String REPORT_KEY_COVERAGE_PATH = "coverageFilePath";
252ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
253ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    /**
2549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * The test is starting.
2559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final int REPORT_VALUE_RESULT_START = 1;
2579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
2589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * The test completed successfully.
2599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final int REPORT_VALUE_RESULT_OK = 0;
2619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
2629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * The test completed with an error.
2639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final int REPORT_VALUE_RESULT_ERROR = -1;
2659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
2669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * The test completed with a failure.
2679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final int REPORT_VALUE_RESULT_FAILURE = -2;
2699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
270ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * If included in the status bundle sent to an IInstrumentationWatcher, this key
271ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * identifies a stack trace describing an error or failure.  This is sent with any status
2729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * message describing a specific test being completed.
2739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
2749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public static final String REPORT_KEY_STACK = "stack";
2759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
27651e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot    // Default file name for code coverage
27751e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot    private static final String DEFAULT_COVERAGE_FILE_NAME = "coverage.ec";
278ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
2799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private static final String LOG_TAG = "InstrumentationTestRunner";
2809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
2819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private final Bundle mResults = new Bundle();
282cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu    private Bundle mArguments;
2839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private AndroidTestRunner mTestRunner;
2849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean mDebug;
2859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean mJustCount;
2869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean mSuiteAssignmentMode;
2879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private int mTestCount;
2889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private String mPackageOfTests;
2899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean mCoverage;
2909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private String mCoverageFilePath;
2919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private int mDelayMsec;
2929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
2939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    @Override
2949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public void onCreate(Bundle arguments) {
2959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        super.onCreate(arguments);
296cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu        mArguments = arguments;
2979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
2989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        // Apk paths used to search for test classes when using TestSuiteBuilders.
2999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String[] apkPaths =
3009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                {getTargetContext().getPackageCodePath(), getContext().getPackageCodePath()};
3019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        ClassPathPackageInfoSource.setApkPaths(apkPaths);
3029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        Predicate<TestMethod> testSizePredicate = null;
30488e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        Predicate<TestMethod> testAnnotationPredicate = null;
30588e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        Predicate<TestMethod> testNotAnnotationPredicate = null;
3069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String testClassesArg = null;
3079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        boolean logOnly = false;
3089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (arguments != null) {
3109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // Test class name passed as an argument should override any meta-data declaration.
3119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testClassesArg = arguments.getString(ARGUMENT_TEST_CLASS);
3129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mDebug = getBooleanArgument(arguments, "debug");
3139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mJustCount = getBooleanArgument(arguments, "count");
3149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mSuiteAssignmentMode = getBooleanArgument(arguments, "suiteAssignment");
3159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mPackageOfTests = arguments.getString(ARGUMENT_TEST_PACKAGE);
3169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testSizePredicate = getSizePredicateFromArg(
3179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    arguments.getString(ARGUMENT_TEST_SIZE_PREDICATE));
31888e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot            testAnnotationPredicate = getAnnotationPredicate(
31988e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot                    arguments.getString(ARGUMENT_ANNOTATION));
32088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot            testNotAnnotationPredicate = getNotAnnotationPredicate(
32188e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot                    arguments.getString(ARGUMENT_NOT_ANNOTATION));
32288e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot
3239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            logOnly = getBooleanArgument(arguments, ARGUMENT_LOG_ONLY);
3249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mCoverage = getBooleanArgument(arguments, "coverage");
3259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mCoverageFilePath = arguments.getString("coverageFile");
3269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            try {
3289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                Object delay = arguments.get(ARGUMENT_DELAY_MSEC);  // Accept either string or int
3299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                if (delay != null) mDelayMsec = Integer.parseInt(delay.toString());
3309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } catch (NumberFormatException e) {
3319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                Log.e(LOG_TAG, "Invalid delay_msec parameter", e);
3329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
3339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
3349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        TestSuiteBuilder testSuiteBuilder = new TestSuiteBuilder(getClass().getName(),
3369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                getTargetContext().getClassLoader());
3379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (testSizePredicate != null) {
3399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testSuiteBuilder.addRequirements(testSizePredicate);
3409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
34188e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        if (testAnnotationPredicate != null) {
34288e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot            testSuiteBuilder.addRequirements(testAnnotationPredicate);
34388e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        }
34488e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        if (testNotAnnotationPredicate != null) {
34588e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot            testSuiteBuilder.addRequirements(testNotAnnotationPredicate);
34688e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        }
3479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (testClassesArg == null) {
3499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (mPackageOfTests != null) {
3509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                testSuiteBuilder.includePackages(mPackageOfTests);
3519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } else {
35261b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                TestSuite testSuite = getTestSuite();
35361b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                if (testSuite != null) {
35461b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                    testSuiteBuilder.addTestSuite(testSuite);
35561b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                } else {
356ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    // no package or class bundle arguments were supplied, and no test suite
35761b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                    // provided so add all tests in application
35861b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                    testSuiteBuilder.includePackages("");
35961b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot                }
3609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
3619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else {
3629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            parseTestClasses(testClassesArg, testSuiteBuilder);
3639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
364ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
365da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob        testSuiteBuilder.addRequirements(getBuilderRequirements());
3669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
3679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestRunner = getAndroidTestRunner();
3689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestRunner.setContext(getTargetContext());
3697aba54b2edcb94f43c991c555b23fc1de364835eJack Wang        mTestRunner.setInstrumentation(this);
3709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestRunner.setSkipExecution(logOnly);
3719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestRunner.setTest(testSuiteBuilder.build());
3729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mTestCount = mTestRunner.getTestCases().size();
3739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (mSuiteAssignmentMode) {
3749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestRunner.addTestListener(new SuiteAssignmentPrinter());
3759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else {
376ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            WatcherResultPrinter resultPrinter = new WatcherResultPrinter(mTestCount);
3779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestRunner.addTestListener(new TestPrinter("TestRunner", false));
378ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestRunner.addTestListener(resultPrinter);
379ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestRunner.setPerformanceResultsWriter(resultPrinter);
3809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
3819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        start();
3829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
3839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
384cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu    /**
385cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu     * Get the Bundle object that contains the arguments
386cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu     *
387cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu     * @return the Bundle object
388cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu     * @hide
389cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu     */
390cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu    public Bundle getBundle(){
391cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu        return mArguments;
392cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu    }
393cbc584d7ebe332f78bf8012d21265ccb482c7cfcDavid Hu
3948cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu    /**
3958cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu     * Add a {@link TestListener}
3968cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu     * @hide
3978cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu     */
3988cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu    protected void addTestListener(TestListener listener){
3998cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu        if(mTestRunner!=null && listener!=null){
4008cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu            mTestRunner.addTestListener(listener);
4018cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu        }
4028cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu    }
4038cc9a8e132ebbf852e055ed17ab91f35b02fd974David Hu
404da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob    List<Predicate<TestMethod>> getBuilderRequirements() {
405da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob        return new ArrayList<Predicate<TestMethod>>();
406da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob    }
407da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob
4089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
409ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * Parses and loads the specified set of test classes
410ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     *
4119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * @param testClassArg - comma-separated list of test classes and methods
4129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * @param testSuiteBuilder - builder to add tests to
4139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
4149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void parseTestClasses(String testClassArg, TestSuiteBuilder testSuiteBuilder) {
4159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String[] testClasses = testClassArg.split(",");
4169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        for (String testClass : testClasses) {
4179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            parseTestClass(testClass, testSuiteBuilder);
4189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
4199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
4209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
4219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
4229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * Parse and load the given test class and, optionally, method
423ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     *
424ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * @param testClassName - full package name of test class and optionally method to add.
425ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     *        Expected format: com.android.TestClass#testMethod
4269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * @param testSuiteBuilder - builder to add tests to
4279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
4289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void parseTestClass(String testClassName, TestSuiteBuilder testSuiteBuilder) {
4299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        int methodSeparatorIndex = testClassName.indexOf('#');
4309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String testMethodName = null;
4319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
4329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (methodSeparatorIndex > 0) {
4339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testMethodName = testClassName.substring(methodSeparatorIndex + 1);
4349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            testClassName = testClassName.substring(0, methodSeparatorIndex);
4359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
436ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        testSuiteBuilder.addTestClassByName(testClassName, testMethodName, getTargetContext());
4379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
4389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
4399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    protected AndroidTestRunner getAndroidTestRunner() {
4409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return new AndroidTestRunner();
4419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
4429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
4439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private boolean getBooleanArgument(Bundle arguments, String tag) {
4449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String tagString = arguments.getString(tag);
4459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return tagString != null && Boolean.parseBoolean(tagString);
4469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
447ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /*
4499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * Returns the size predicate object, corresponding to the "size" argument value.
4509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
4519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private Predicate<TestMethod> getSizePredicateFromArg(String sizeArg) {
452ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
4539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (SMALL_SUITE.equals(sizeArg)) {
4549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return TestPredicates.SELECT_SMALL;
4559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else if (MEDIUM_SUITE.equals(sizeArg)) {
4569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return TestPredicates.SELECT_MEDIUM;
4579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else if (LARGE_SUITE.equals(sizeArg)) {
4589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return TestPredicates.SELECT_LARGE;
4599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else {
4609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return null;
4619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
4629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
463ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
46488e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot   /**
46588e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    * Returns the test predicate object, corresponding to the annotation class value provided via
46688e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    * the {@link ARGUMENT_ANNOTATION} argument.
46788e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    *
46888e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    * @return the predicate or <code>null</code>
46988e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    */
47088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    private Predicate<TestMethod> getAnnotationPredicate(String annotationClassName) {
47188e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        Class<? extends Annotation> annotationClass = getAnnotationClass(annotationClassName);
47288e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        if (annotationClass != null) {
47388e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot            return new HasAnnotation(annotationClass);
47488e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        }
47588e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        return null;
47688e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    }
47788e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot
47888e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    /**
47988e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     * Returns the negative test predicate object, corresponding to the annotation class value
48088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     * provided via the {@link ARGUMENT_NOT_ANNOTATION} argument.
48188e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     *
48288e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     * @return the predicate or <code>null</code>
48388e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     */
48488e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     private Predicate<TestMethod> getNotAnnotationPredicate(String annotationClassName) {
48588e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot         Class<? extends Annotation> annotationClass = getAnnotationClass(annotationClassName);
48688e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot         if (annotationClass != null) {
48788e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot             return Predicates.not(new HasAnnotation(annotationClass));
48888e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot         }
48988e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot         return null;
49088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     }
49188e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot
49288e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    /**
49388e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     * Helper method to return the annotation class with specified name
49488e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     *
49588e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     * @param annotationClassName the fully qualified name of the class
49688e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     * @return the annotation class or <code>null</code>
49788e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot     */
49888e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    private Class<? extends Annotation> getAnnotationClass(String annotationClassName) {
49988e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        if (annotationClassName == null) {
50088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot            return null;
50188e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        }
50288e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        try {
50388e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot           Class<?> annotationClass = Class.forName(annotationClassName);
50488e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot           if (annotationClass.isAnnotation()) {
50588e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot               return (Class<? extends Annotation>)annotationClass;
50688e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot           } else {
50788e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot               Log.e(LOG_TAG, String.format("Provided annotation value %s is not an Annotation",
50888e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot                       annotationClassName));
50988e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot           }
51088e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        } catch (ClassNotFoundException e) {
51188e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot            Log.e(LOG_TAG, String.format("Could not find class for specified annotation %s",
51288e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot                    annotationClassName));
51388e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        }
51488e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot        return null;
51588e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot    }
51688e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot
51731e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot    /**
51831e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot     * Initialize the current thread as a looper.
51931e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot     * <p/>
52031e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot     * Exposed for unit testing.
52131e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot     */
52231e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot    void prepareLooper() {
52331e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot        Looper.prepare();
52431e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot    }
52531e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot
5269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    @Override
5279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public void onStart() {
52831e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot        prepareLooper();
529ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (mJustCount) {
5319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResults.putString(Instrumentation.REPORT_KEY_IDENTIFIER, REPORT_VALUE_ID);
5329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResults.putInt(REPORT_KEY_NUM_TOTAL, mTestCount);
5339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            finish(Activity.RESULT_OK, mResults);
5349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } else {
5359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (mDebug) {
5369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                Debug.waitForDebugger();
5379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
538ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
5409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            PrintStream writer = new PrintStream(byteArrayOutputStream);
5419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            try {
5429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                StringResultPrinter resultPrinter = new StringResultPrinter(writer);
543ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestRunner.addTestListener(resultPrinter);
545ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                long startTime = System.currentTimeMillis();
5479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestRunner.runTest();
5489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                long runTime = System.currentTimeMillis() - startTime;
549ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                resultPrinter.print(mTestRunner.getTestResult(), runTime);
55131e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot            } catch (Throwable t) {
55231e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot                // catch all exceptions so a more verbose error message can be outputted
55331e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot                writer.println(String.format("Test run aborted due to unexpected exception: %s",
55431e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot                                t.getMessage()));
55531e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot                t.printStackTrace(writer);
5569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } finally {
557ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mResults.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
558ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        String.format("\nTest results for %s=%s",
559ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        mTestRunner.getTestClassName(),
5609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                        byteArrayOutputStream.toString()));
5619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                if (mCoverage) {
5639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    generateCoverageReport();
5649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                }
5659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                writer.close();
566ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                finish(Activity.RESULT_OK, mResults);
5689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
5699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
5709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
5719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public TestSuite getTestSuite() {
5739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return getAllTests();
5749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
5759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
5779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * Override this to define all of the tests to run in your package.
5789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
5799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public TestSuite getAllTests() {
5809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return null;
5819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
5829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
5839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
5849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * Override this to provide access to the class loader of your package.
5859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
5869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    public ClassLoader getLoader() {
5879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        return null;
5889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
589ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
5909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void generateCoverageReport() {
5919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        // use reflection to call emma dump coverage method, to avoid
5929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        // always statically compiling against emma jar
59351e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot        String coverageFilePath = getCoverageFilePath();
59451e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot        java.io.File coverageFile = new java.io.File(coverageFilePath);
5959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        try {
59688e03a97366d08fc69d852cf2219f9d0b1916af4Brett Chabot            Class<?> emmaRTClass = Class.forName("com.vladium.emma.rt.RT");
597ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            Method dumpCoverageMethod = emmaRTClass.getMethod("dumpCoverageData",
5989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    coverageFile.getClass(), boolean.class, boolean.class);
599ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            dumpCoverageMethod.invoke(null, coverageFile, false, false);
60151e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            // output path to generated coverage file so it can be parsed by a test harness if
60251e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            // needed
60351e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            mResults.putString(REPORT_KEY_COVERAGE_PATH, coverageFilePath);
60451e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            // also output a more user friendly msg
60508d13c3cbb0b250b84ca861edb462060556aead9Brett Chabot            final String currentStream = mResults.getString(
60608d13c3cbb0b250b84ca861edb462060556aead9Brett Chabot                    Instrumentation.REPORT_KEY_STREAMRESULT);
60751e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            mResults.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
60808d13c3cbb0b250b84ca861edb462060556aead9Brett Chabot                String.format("%s\nGenerated code coverage data to %s", currentStream,
60908d13c3cbb0b250b84ca861edb462060556aead9Brett Chabot                coverageFilePath));
6109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (ClassNotFoundException e) {
6119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError("Is emma jar on classpath?", e);
6129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (SecurityException e) {
6139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
6149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (NoSuchMethodException e) {
6159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
6169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (IllegalArgumentException e) {
6179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
6189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (IllegalAccessException e) {
6199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
6209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        } catch (InvocationTargetException e) {
6219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            reportEmmaError(e);
6229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
6239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
6249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private String getCoverageFilePath() {
6269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        if (mCoverageFilePath == null) {
62751e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot            return getTargetContext().getFilesDir().getAbsolutePath() + File.separator +
628ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                   DEFAULT_COVERAGE_FILE_NAME;
629ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        } else {
6309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            return mCoverageFilePath;
6319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
6329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
6339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void reportEmmaError(Exception e) {
635ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        reportEmmaError("", e);
6369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
6379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private void reportEmmaError(String hint, Exception e) {
6399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String msg = "Failed to generate emma coverage. " + hint;
6409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        Log.e(LOG_TAG, msg, e);
6419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        mResults.putString(Instrumentation.REPORT_KEY_STREAMRESULT, "\nError: " + msg);
6429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
6439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    // TODO kill this, use status() and prettyprint model for better output
6459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    private class StringResultPrinter extends ResultPrinter {
6469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public StringResultPrinter(PrintStream writer) {
6489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            super(writer);
6499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
6509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        synchronized void print(TestResult result, long runTime) {
6529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            printHeader(runTime);
6539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            printFooter(result);
6549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
6559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
656ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
658ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang     * This class sends status reports back to the IInstrumentationWatcher about
6599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * which suite each test belongs.
6609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
661ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    private class SuiteAssignmentPrinter implements TestListener {
662ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private Bundle mTestResult;
6649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private long mStartTime;
6659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private long mEndTime;
6669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private boolean mTimingValid;
667ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public SuiteAssignmentPrinter() {
6699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
670ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
6729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * send a status for the start of a each test, so long tests can be seen as "running"
6739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
6749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void startTest(Test test) {
6759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTimingValid = true;
676ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mStartTime = System.currentTimeMillis();
6779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
678ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
6799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
6809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#addError(Test, Throwable)
6819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
6829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void addError(Test test, Throwable t) {
6839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTimingValid = false;
6849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
6859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
6879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#addFailure(Test, AssertionFailedError)
6889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
6899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void addFailure(Test test, AssertionFailedError t) {
6909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTimingValid = false;
6919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
6929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
6939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
6949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#endTest(Test)
6959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
6969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void endTest(Test test) {
6979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            float runTime;
6989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            String assignmentSuite;
6999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mEndTime = System.currentTimeMillis();
7009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult = new Bundle();
7019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
7029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (!mTimingValid || mStartTime < 0) {
7039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                assignmentSuite = "NA";
7049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                runTime = -1;
7059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } else {
7069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                runTime = mEndTime - mStartTime;
707ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                if (runTime < SMALL_SUITE_MAX_RUNTIME
7089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                        && !InstrumentationTestCase.class.isAssignableFrom(test.getClass())) {
7099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    assignmentSuite = SMALL_SUITE;
7109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                } else if (runTime < MEDIUM_SUITE_MAX_RUNTIME) {
7119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    assignmentSuite = MEDIUM_SUITE;
7129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                } else {
7139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    assignmentSuite = LARGE_SUITE;
7149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                }
7159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
7169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // Clear mStartTime so that we can verify that it gets set next time.
7179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mStartTime = -1;
7189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
719ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
720ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    test.getClass().getName() + "#" + ((TestCase) test).getName()
7219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    + "\nin " + assignmentSuite + " suite\nrunTime: "
7229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                    + String.valueOf(runTime) + "\n");
7239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putFloat(REPORT_KEY_RUN_TIME, runTime);
7249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putString(REPORT_KEY_SUITE_ASSIGNMENT, assignmentSuite);
7259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
7269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            sendStatus(0, mTestResult);
7279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
7289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
729ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
7309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    /**
7319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     * This class sends status reports back to the IInstrumentationWatcher
7329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project     */
733ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang    private class WatcherResultPrinter implements TestListener, PerformanceResultsWriter {
7349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        private final Bundle mResultTemplate;
7359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        Bundle mTestResult;
7369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        int mTestNum = 0;
7379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        int mTestResultCode = 0;
7389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        String mTestClass = null;
7394f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang        PerformanceCollector mPerfCollector = new PerformanceCollector();
740ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        boolean mIsTimedTest = false;
7414f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang        boolean mIncludeDetailedStats = false;
742ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
7439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public WatcherResultPrinter(int numTests) {
7449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResultTemplate = new Bundle();
7459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResultTemplate.putString(Instrumentation.REPORT_KEY_IDENTIFIER, REPORT_VALUE_ID);
7469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mResultTemplate.putInt(REPORT_KEY_NUM_TOTAL, numTests);
7479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
748ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
7499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
750ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang         * send a status for the start of a each test, so long tests can be seen
751ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang         * as "running"
7529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
7539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void startTest(Test test) {
7549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            String testClass = test.getClass().getName();
755ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            String testName = ((TestCase)test).getName();
7569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult = new Bundle(mResultTemplate);
7579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putString(REPORT_KEY_NAME_CLASS, testClass);
758ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestResult.putString(REPORT_KEY_NAME_TEST, testName);
7599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putInt(REPORT_KEY_NUM_CURRENT, ++mTestNum);
7609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // pretty printing
7619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (testClass != null && !testClass.equals(mTestClass)) {
762ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
7639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                        String.format("\n%s:", testClass));
7649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestClass = testClass;
7659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } else {
7669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, "");
7679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
7689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
7693fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang            Method testMethod = null;
7703fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang            try {
7713fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                testMethod = test.getClass().getMethod(testName);
7723fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                // Report total number of iterations, if test is repetitive
7733fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                if (testMethod.isAnnotationPresent(RepetitiveTest.class)) {
7743fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                    int numIterations = testMethod.getAnnotation(
7753fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                        RepetitiveTest.class).numIterations();
7763fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                    mTestResult.putInt(REPORT_KEY_NUM_ITERATIONS, numIterations);
7773fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                }
7783fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang            } catch (NoSuchMethodException e) {
7793fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                // ignore- the test with given name does not exist. Will be handled during test
7803fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                // execution
7813fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang            }
7823fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang
7839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // The delay_msec parameter is normally used to provide buffers of idle time
784ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // for power measurement purposes. To make sure there is a delay before and after
7859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // every test in a suite, we delay *after* every test (see endTest below) and also
786ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // delay *before* the first test. So, delay test1 delay test2 delay.
7879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
7889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            try {
7899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                if (mTestNum == 1) Thread.sleep(mDelayMsec);
7909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } catch (InterruptedException e) {
7919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                throw new IllegalStateException(e);
7929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
7939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
7949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            sendStatus(REPORT_VALUE_RESULT_START, mTestResult);
7959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResultCode = 0;
796ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
797ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mIsTimedTest = false;
7984f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang            mIncludeDetailedStats = false;
799ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            try {
8004f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                // Look for TimedTest annotation on both test class and test method
80148983959a34f67024bd0411c8353c196e6a87717Brian Muramatsu                if (testMethod != null && testMethod.isAnnotationPresent(TimedTest.class)) {
8024f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                    mIsTimedTest = true;
8033fc03e619fb01678549b80e7a89af2c8e3f19968Jack Wang                    mIncludeDetailedStats = testMethod.getAnnotation(
8044f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                            TimedTest.class).includeDetailedStats();
8054f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                } else if (test.getClass().isAnnotationPresent(TimedTest.class)) {
8064f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                    mIsTimedTest = true;
8074f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                    mIncludeDetailedStats = test.getClass().getAnnotation(
8084f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                            TimedTest.class).includeDetailedStats();
8094f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                }
810ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            } catch (SecurityException e) {
81131e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot                // ignore - the test with given name cannot be accessed. Will be handled during
81231e7ce762ccbbb747fabb4581e42a0a2fe56e780Brett Chabot                // test execution
813ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            }
814ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
8154f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang            if (mIsTimedTest && mIncludeDetailedStats) {
8164f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                mPerfCollector.beginSnapshot("");
8174f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang            } else if (mIsTimedTest) {
8184f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                mPerfCollector.startTiming("");
819ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            }
8209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
821ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
8229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
8239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#addError(Test, Throwable)
8249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
8259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void addError(Test test, Throwable t) {
8269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putString(REPORT_KEY_STACK, BaseTestRunner.getFilteredTrace(t));
8279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResultCode = REPORT_VALUE_RESULT_ERROR;
8289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // pretty printing
829ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
830ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                String.format("\nError in %s:\n%s",
831ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    ((TestCase)test).getName(), BaseTestRunner.getFilteredTrace(t)));
8329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
8339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
8349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
8359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#addFailure(Test, AssertionFailedError)
8369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
8379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void addFailure(Test test, AssertionFailedError t) {
8389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResult.putString(REPORT_KEY_STACK, BaseTestRunner.getFilteredTrace(t));
8399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            mTestResultCode = REPORT_VALUE_RESULT_FAILURE;
8409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            // pretty printing
841ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT,
842ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                String.format("\nFailure in %s:\n%s",
843ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    ((TestCase)test).getName(), BaseTestRunner.getFilteredTrace(t)));
8449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
8459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
8469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        /**
8479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         * @see junit.framework.TestListener#endTest(Test)
8489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project         */
8499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        public void endTest(Test test) {
8504f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang            if (mIsTimedTest && mIncludeDetailedStats) {
8514f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                mTestResult.putAll(mPerfCollector.endSnapshot());
8524f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang            } else if (mIsTimedTest) {
8534f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                writeStopTiming(mPerfCollector.stopTiming(""));
854ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            }
855ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
8569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            if (mTestResultCode == 0) {
8579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, ".");
8589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
8599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            sendStatus(mTestResultCode, mTestResult);
8609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
861ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            try { // Sleep after every test, if specified
8629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                Thread.sleep(mDelayMsec);
8639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            } catch (InterruptedException e) {
8649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project                throw new IllegalStateException(e);
8659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project            }
8669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        }
8679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project
868ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        public void writeBeginSnapshot(String label) {
869ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // Do nothing
870ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        }
871ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
872ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        public void writeEndSnapshot(Bundle results) {
873075997f12e2fb2b646172a92926be0f26f739099Jack Wang            // Copy all snapshot data fields into mResults, which is outputted
874075997f12e2fb2b646172a92926be0f26f739099Jack Wang            // via Instrumentation.finish
875075997f12e2fb2b646172a92926be0f26f739099Jack Wang            mResults.putAll(results);
876ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        }
877ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
878ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        public void writeStartTiming(String label) {
879ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // Do nothing
880ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        }
881ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
882ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        public void writeStopTiming(Bundle results) {
883ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // Copy results into mTestResult by flattening list of iterations,
884ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            // which is outputted via WatcherResultPrinter.endTest
885ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            int i = 0;
886ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            for (Parcelable p :
887ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                    results.getParcelableArrayList(PerformanceCollector.METRIC_KEY_ITERATIONS)) {
888ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                Bundle iteration = (Bundle)p;
8894f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang                String index = "iteration" + i + ".";
890ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putString(index + PerformanceCollector.METRIC_KEY_LABEL,
891ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        iteration.getString(PerformanceCollector.METRIC_KEY_LABEL));
892ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putLong(index + PerformanceCollector.METRIC_KEY_CPU_TIME,
893ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        iteration.getLong(PerformanceCollector.METRIC_KEY_CPU_TIME));
894ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                mTestResult.putLong(index + PerformanceCollector.METRIC_KEY_EXECUTION_TIME,
895ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                        iteration.getLong(PerformanceCollector.METRIC_KEY_EXECUTION_TIME));
896ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang                i++;
897ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang            }
898ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang        }
899ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang
900075997f12e2fb2b646172a92926be0f26f739099Jack Wang        public void writeMeasurement(String label, long value) {
9014f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang            mTestResult.putLong(label, value);
902075997f12e2fb2b646172a92926be0f26f739099Jack Wang        }
903075997f12e2fb2b646172a92926be0f26f739099Jack Wang
904075997f12e2fb2b646172a92926be0f26f739099Jack Wang        public void writeMeasurement(String label, float value) {
9054f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang            mTestResult.putFloat(label, value);
906075997f12e2fb2b646172a92926be0f26f739099Jack Wang        }
907075997f12e2fb2b646172a92926be0f26f739099Jack Wang
908075997f12e2fb2b646172a92926be0f26f739099Jack Wang        public void writeMeasurement(String label, String value) {
9094f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang            mTestResult.putString(label, value);
910075997f12e2fb2b646172a92926be0f26f739099Jack Wang        }
911075997f12e2fb2b646172a92926be0f26f739099Jack Wang
9129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project        // TODO report the end of the cycle
9139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project    }
9149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project}
915