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