InstrumentationTestRunner.java revision 08d13c3cbb0b250b84ca861edb462060556aead9
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.PerformanceCollector.PerformanceResultsWriter; 319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.TestMethod; 329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.TestPredicates; 339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.test.suitebuilder.TestSuiteBuilder; 349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport android.util.Log; 359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 36ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.io.ByteArrayOutputStream; 37ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.io.File; 38ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.io.PrintStream; 39ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.lang.reflect.InvocationTargetException; 40ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.lang.reflect.Method; 41ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.util.ArrayList; 42ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wangimport java.util.List; 439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.AssertionFailedError; 459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.Test; 469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestCase; 479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestListener; 489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestResult; 499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.framework.TestSuite; 509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.runner.BaseTestRunner; 519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectimport junit.textui.ResultPrinter; 529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/** 549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * An {@link Instrumentation} that runs various types of {@link junit.framework.TestCase}s against 559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * an Android package (application). Typical usage: 569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <ol> 579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Write {@link junit.framework.TestCase}s that perform unit, functional, or performance tests 589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * against the classes in your package. Typically these are subclassed from: 59ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * <ul><li>{@link android.test.ActivityInstrumentationTestCase2}</li> 609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>{@link android.test.ActivityUnitTestCase}</li> 619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>{@link android.test.AndroidTestCase}</li> 629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>{@link android.test.ApplicationTestCase}</li> 639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>{@link android.test.InstrumentationTestCase}</li> 649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>{@link android.test.ProviderTestCase}</li> 659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>{@link android.test.ServiceTestCase}</li> 669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>{@link android.test.SingleLaunchActivityTestCase}</li></ul> 679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>In an appropriate AndroidManifest.xml, define the this instrumentation with 689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * the appropriate android:targetPackage set. 699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Run the instrumentation using "adb shell am instrument -w", 709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * with no optional arguments, to run all tests (except performance tests). 719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Run the instrumentation using "adb shell am instrument -w", 729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * with the argument '-e func true' to run all functional tests. These are tests that derive from 739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * {@link android.test.InstrumentationTestCase}. 749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Run the instrumentation using "adb shell am instrument -w", 759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * with the argument '-e unit true' to run all unit tests. These are tests that <i>do not</i>derive 769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * from {@link android.test.InstrumentationTestCase} (and are not performance tests). 779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <li>Run the instrumentation using "adb shell am instrument -w", 789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * with the argument '-e class' set to run an individual {@link junit.framework.TestCase}. 799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * </ol> 809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running all tests:</b> adb shell am instrument -w 829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner 839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running all small tests:</b> adb shell am instrument -w 859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e size small 869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner 879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running all medium tests:</b> adb shell am instrument -w 899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e size medium 909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner 919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running all large tests:</b> adb shell am instrument -w 939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e size large 949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner 959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running a single testcase:</b> adb shell am instrument -w 979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e class com.android.foo.FooTest 989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner 999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 1009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running a single test:</b> adb shell am instrument -w 1019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e class com.android.foo.FooTest#testFoo 1029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner 1039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 1049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Running multiple tests:</b> adb shell am instrument -w 1059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e class com.android.foo.FooTest,com.android.foo.TooTest 1069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner 1079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 1089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>Including performance tests:</b> adb shell am instrument -w 1099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e perf true 1109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * com.android.foo/android.test.InstrumentationTestRunner 1119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 1129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>To debug your tests, set a break point in your code and pass:</b> 1139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e debug true 1149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 1159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>To run in 'log only' mode</b> 1169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e log true 117ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * This option will load and iterate through all test classes and methods, but will bypass actual 118ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * test execution. Useful for quickly obtaining info on the tests to be executed by an 1199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * instrumentation command. 1209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 1219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b>To generate EMMA code coverage:</b> 1229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e coverage true 123ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * Note: this requires an emma instrumented build. By default, the code coverage results file 12451e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot * will be saved in a /data/<app>/coverage.ec file, unless overridden by coverageFile flag (see 12551e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot * below) 1269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 1279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <b> To specify EMMA code coverage results file path:</b> 1289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * -e coverageFile /sdcard/myFile.ec 1299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <br/> 1309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * in addition to the other arguments. 1319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 1329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project/* (not JavaDoc) 1349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Although not necessary in most case, another way to use this class is to extend it and have the 135ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * derived class return the desired test suite from the {@link #getTestSuite()} method. The test 136ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * suite returned from this method will be used if no target class is defined in the meta-data or 137ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * command line argument parameters. If a derived class is used it needs to be added as an 138ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * instrumentation to the AndroidManifest.xml and the command to run it would look like: 1399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 1409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * adb shell am instrument -w com.android.foo/<i>com.android.FooInstrumentationTestRunner</i> 1419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * <p/> 1429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Where <i>com.android.FooInstrumentationTestRunner</i> is the derived class. 1439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * 1449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * This model is used by many existing app tests, but can probably be deprecated. 1459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 1469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Projectpublic class InstrumentationTestRunner extends Instrumentation implements TestSuiteProvider { 1479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** @hide */ 1499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String ARGUMENT_TEST_CLASS = "class"; 1509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** @hide */ 1519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String ARGUMENT_TEST_PACKAGE = "package"; 1529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** @hide */ 1539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String ARGUMENT_TEST_SIZE_PREDICATE = "size"; 1549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** @hide */ 1559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String ARGUMENT_INCLUDE_PERF = "perf"; 1569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** @hide */ 1579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String ARGUMENT_DELAY_MSEC = "delay_msec"; 1589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private static final String SMALL_SUITE = "small"; 160ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang private static final String MEDIUM_SUITE = "medium"; 1619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private static final String LARGE_SUITE = "large"; 162ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 1639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private static final String ARGUMENT_LOG_ONLY = "log"; 1649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 166ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * This constant defines the maximum allowed runtime (in ms) for a test included in the "small" 167ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * suite. It is used to make an educated guess at what suite an unlabeled test belongs. 1689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 1699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private static final float SMALL_SUITE_MAX_RUNTIME = 100; 170ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 1719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 172ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * This constant defines the maximum allowed runtime (in ms) for a test included in the 173ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * "medium" suite. It is used to make an educated guess at what suite an unlabeled test belongs. 1749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 1759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private static final float MEDIUM_SUITE_MAX_RUNTIME = 1000; 176ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 1779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 178ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * The following keys are used in the status bundle to provide structured reports to 179ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * an IInstrumentationWatcher. 1809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 1819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 1829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 183ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * This value, if stored with key {@link android.app.Instrumentation#REPORT_KEY_IDENTIFIER}, 1849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * identifies InstrumentationTestRunner as the source of the report. This is sent with all 1859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * status messages. 1869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 1879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String REPORT_VALUE_ID = "InstrumentationTestRunner"; 1889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 189ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * If included in the status or final bundle sent to an IInstrumentationWatcher, this key 1909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * identifies the total number of tests that are being run. This is sent with all status 1919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * messages. 1929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 1939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String REPORT_KEY_NUM_TOTAL = "numtests"; 1949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 195ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * If included in the status or final bundle sent to an IInstrumentationWatcher, this key 1969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * identifies the sequence number of the current test. This is sent with any status message 1979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * describing a specific test being started or completed. 1989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 1999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String REPORT_KEY_NUM_CURRENT = "current"; 2009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 201ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * If included in the status or final bundle sent to an IInstrumentationWatcher, this key 2029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * identifies the name of the current test class. This is sent with any status message 2039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * describing a specific test being started or completed. 2049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String REPORT_KEY_NAME_CLASS = "class"; 2069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 207ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * If included in the status or final bundle sent to an IInstrumentationWatcher, this key 2089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * identifies the name of the current test. This is sent with any status message 2099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * describing a specific test being started or completed. 2109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String REPORT_KEY_NAME_TEST = "test"; 2129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 213ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * If included in the status or final bundle sent to an IInstrumentationWatcher, this key 2149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * reports the run time in seconds of the current test. 2159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private static final String REPORT_KEY_RUN_TIME = "runtime"; 2179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 218ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * If included in the status or final bundle sent to an IInstrumentationWatcher, this key 2199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * reports the guessed suite assignment for the current test. 2209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private static final String REPORT_KEY_SUITE_ASSIGNMENT = "suiteassignment"; 2229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 22351e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot * If included in the status or final bundle sent to an IInstrumentationWatcher, this key 22451e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot * identifies the path to the generated code coverage file. 22551e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot */ 22651e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot private static final String REPORT_KEY_COVERAGE_PATH = "coverageFilePath"; 227ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 228ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang /** 2299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * The test is starting. 2309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final int REPORT_VALUE_RESULT_START = 1; 2329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 2339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * The test completed successfully. 2349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final int REPORT_VALUE_RESULT_OK = 0; 2369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 2379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * The test completed with an error. 2389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final int REPORT_VALUE_RESULT_ERROR = -1; 2409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 2419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * The test completed with a failure. 2429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final int REPORT_VALUE_RESULT_FAILURE = -2; 2449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 245ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * If included in the status bundle sent to an IInstrumentationWatcher, this key 246ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * identifies a stack trace describing an error or failure. This is sent with any status 2479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * message describing a specific test being completed. 2489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 2499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public static final String REPORT_KEY_STACK = "stack"; 2509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 25151e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot // Default file name for code coverage 25251e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot private static final String DEFAULT_COVERAGE_FILE_NAME = "coverage.ec"; 253ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 2549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private static final String LOG_TAG = "InstrumentationTestRunner"; 2559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private final Bundle mResults = new Bundle(); 2579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private AndroidTestRunner mTestRunner; 2589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private boolean mDebug; 2599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private boolean mJustCount; 2609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private boolean mSuiteAssignmentMode; 2619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private int mTestCount; 2629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private String mPackageOfTests; 2639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private boolean mCoverage; 2649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private String mCoverageFilePath; 2659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private int mDelayMsec; 2669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project @Override 2689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void onCreate(Bundle arguments) { 2699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project super.onCreate(arguments); 2709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // Apk paths used to search for test classes when using TestSuiteBuilders. 2729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String[] apkPaths = 2739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project {getTargetContext().getPackageCodePath(), getContext().getPackageCodePath()}; 2749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ClassPathPackageInfoSource.setApkPaths(apkPaths); 2759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Predicate<TestMethod> testSizePredicate = null; 2779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project boolean includePerformance = false; 2789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String testClassesArg = null; 2799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project boolean logOnly = false; 2809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (arguments != null) { 2829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // Test class name passed as an argument should override any meta-data declaration. 2839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project testClassesArg = arguments.getString(ARGUMENT_TEST_CLASS); 2849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mDebug = getBooleanArgument(arguments, "debug"); 2859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mJustCount = getBooleanArgument(arguments, "count"); 2869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mSuiteAssignmentMode = getBooleanArgument(arguments, "suiteAssignment"); 2879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mPackageOfTests = arguments.getString(ARGUMENT_TEST_PACKAGE); 2889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project testSizePredicate = getSizePredicateFromArg( 2899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project arguments.getString(ARGUMENT_TEST_SIZE_PREDICATE)); 2909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project includePerformance = getBooleanArgument(arguments, ARGUMENT_INCLUDE_PERF); 2919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project logOnly = getBooleanArgument(arguments, ARGUMENT_LOG_ONLY); 2929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mCoverage = getBooleanArgument(arguments, "coverage"); 2939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mCoverageFilePath = arguments.getString("coverageFile"); 2949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 2959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project try { 2969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Object delay = arguments.get(ARGUMENT_DELAY_MSEC); // Accept either string or int 2979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (delay != null) mDelayMsec = Integer.parseInt(delay.toString()); 2989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } catch (NumberFormatException e) { 2999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Log.e(LOG_TAG, "Invalid delay_msec parameter", e); 3009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project TestSuiteBuilder testSuiteBuilder = new TestSuiteBuilder(getClass().getName(), 3049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project getTargetContext().getClassLoader()); 3059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (testSizePredicate != null) { 3079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project testSuiteBuilder.addRequirements(testSizePredicate); 3089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (!includePerformance) { 3109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project testSuiteBuilder.addRequirements(REJECT_PERFORMANCE); 3119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (testClassesArg == null) { 3149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (mPackageOfTests != null) { 3159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project testSuiteBuilder.includePackages(mPackageOfTests); 3169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 31761b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot TestSuite testSuite = getTestSuite(); 31861b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot if (testSuite != null) { 31961b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot testSuiteBuilder.addTestSuite(testSuite); 32061b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot } else { 321ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang // no package or class bundle arguments were supplied, and no test suite 32261b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot // provided so add all tests in application 32361b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot testSuiteBuilder.includePackages(""); 32461b10ac95de29c3ff826250cf12b04b932e0cfd7Brett Chabot } 3259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 3279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project parseTestClasses(testClassesArg, testSuiteBuilder); 3289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 329ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 330da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob testSuiteBuilder.addRequirements(getBuilderRequirements()); 3319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestRunner = getAndroidTestRunner(); 3339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestRunner.setContext(getTargetContext()); 3347aba54b2edcb94f43c991c555b23fc1de364835eJack Wang mTestRunner.setInstrumentation(this); 3359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestRunner.setSkipExecution(logOnly); 3369066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestRunner.setTest(testSuiteBuilder.build()); 3379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestCount = mTestRunner.getTestCases().size(); 3389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (mSuiteAssignmentMode) { 3399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestRunner.addTestListener(new SuiteAssignmentPrinter()); 3409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 341ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang WatcherResultPrinter resultPrinter = new WatcherResultPrinter(mTestCount); 3429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestRunner.addTestListener(new TestPrinter("TestRunner", false)); 343ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestRunner.addTestListener(resultPrinter); 344ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestRunner.setPerformanceResultsWriter(resultPrinter); 3459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project start(); 3479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 349da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob List<Predicate<TestMethod>> getBuilderRequirements() { 350da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob return new ArrayList<Predicate<TestMethod>>(); 351da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob } 352da13ef5dca5cac0e89b82a6d7360a575610b7bccUrs Grob 3539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 354ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * Parses and loads the specified set of test classes 355ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * 3569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * @param testClassArg - comma-separated list of test classes and methods 3579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * @param testSuiteBuilder - builder to add tests to 3589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 3599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private void parseTestClasses(String testClassArg, TestSuiteBuilder testSuiteBuilder) { 3609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String[] testClasses = testClassArg.split(","); 3619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project for (String testClass : testClasses) { 3629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project parseTestClass(testClass, testSuiteBuilder); 3639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 3679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Parse and load the given test class and, optionally, method 368ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * 369ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * @param testClassName - full package name of test class and optionally method to add. 370ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * Expected format: com.android.TestClass#testMethod 3719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * @param testSuiteBuilder - builder to add tests to 3729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 3739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private void parseTestClass(String testClassName, TestSuiteBuilder testSuiteBuilder) { 3749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int methodSeparatorIndex = testClassName.indexOf('#'); 3759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String testMethodName = null; 3769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (methodSeparatorIndex > 0) { 3789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project testMethodName = testClassName.substring(methodSeparatorIndex + 1); 3799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project testClassName = testClassName.substring(0, methodSeparatorIndex); 3809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 381ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang testSuiteBuilder.addTestClassByName(testClassName, testMethodName, getTargetContext()); 3829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project protected AndroidTestRunner getAndroidTestRunner() { 3859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return new AndroidTestRunner(); 3869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 3879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 3889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private boolean getBooleanArgument(Bundle arguments, String tag) { 3899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String tagString = arguments.getString(tag); 3909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return tagString != null && Boolean.parseBoolean(tagString); 3919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 392ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 3939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /* 3949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Returns the size predicate object, corresponding to the "size" argument value. 3959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 3969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private Predicate<TestMethod> getSizePredicateFromArg(String sizeArg) { 397ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 3989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (SMALL_SUITE.equals(sizeArg)) { 3999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return TestPredicates.SELECT_SMALL; 4009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else if (MEDIUM_SUITE.equals(sizeArg)) { 4019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return TestPredicates.SELECT_MEDIUM; 4029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else if (LARGE_SUITE.equals(sizeArg)) { 4039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return TestPredicates.SELECT_LARGE; 4049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 4059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return null; 4069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4079066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 408ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 4099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project @Override 4109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void onStart() { 4119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Looper.prepare(); 412ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 4139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (mJustCount) { 4149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mResults.putString(Instrumentation.REPORT_KEY_IDENTIFIER, REPORT_VALUE_ID); 4159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mResults.putInt(REPORT_KEY_NUM_TOTAL, mTestCount); 4169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project finish(Activity.RESULT_OK, mResults); 4179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 4189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (mDebug) { 4199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Debug.waitForDebugger(); 4209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 421ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 4229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); 4239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project PrintStream writer = new PrintStream(byteArrayOutputStream); 4249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project try { 4259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project StringResultPrinter resultPrinter = new StringResultPrinter(writer); 426ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 4279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestRunner.addTestListener(resultPrinter); 428ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 4299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project long startTime = System.currentTimeMillis(); 4309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestRunner.runTest(); 4319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project long runTime = System.currentTimeMillis() - startTime; 432ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 4339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project resultPrinter.print(mTestRunner.getTestResult(), runTime); 4349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } finally { 435ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mResults.putString(Instrumentation.REPORT_KEY_STREAMRESULT, 436ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang String.format("\nTest results for %s=%s", 437ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestRunner.getTestClassName(), 4389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project byteArrayOutputStream.toString())); 4399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4409066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (mCoverage) { 4419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project generateCoverageReport(); 4429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project writer.close(); 444ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 4459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project finish(Activity.RESULT_OK, mResults); 4469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4489066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public TestSuite getTestSuite() { 4519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return getAllTests(); 4529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 4559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Override this to define all of the tests to run in your package. 4569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 4579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public TestSuite getAllTests() { 4589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return null; 4599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 4609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 4619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 4629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * Override this to provide access to the class loader of your package. 4639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 4649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public ClassLoader getLoader() { 4659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return null; 4669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 467ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 4689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private void generateCoverageReport() { 4699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // use reflection to call emma dump coverage method, to avoid 4709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // always statically compiling against emma jar 47151e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot String coverageFilePath = getCoverageFilePath(); 47251e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot java.io.File coverageFile = new java.io.File(coverageFilePath); 4739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project try { 4749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Class emmaRTClass = Class.forName("com.vladium.emma.rt.RT"); 475ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang Method dumpCoverageMethod = emmaRTClass.getMethod("dumpCoverageData", 4769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project coverageFile.getClass(), boolean.class, boolean.class); 477ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 4789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project dumpCoverageMethod.invoke(null, coverageFile, false, false); 47951e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot // output path to generated coverage file so it can be parsed by a test harness if 48051e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot // needed 48151e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot mResults.putString(REPORT_KEY_COVERAGE_PATH, coverageFilePath); 48251e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot // also output a more user friendly msg 48308d13c3cbb0b250b84ca861edb462060556aead9Brett Chabot final String currentStream = mResults.getString( 48408d13c3cbb0b250b84ca861edb462060556aead9Brett Chabot Instrumentation.REPORT_KEY_STREAMRESULT); 48551e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot mResults.putString(Instrumentation.REPORT_KEY_STREAMRESULT, 48608d13c3cbb0b250b84ca861edb462060556aead9Brett Chabot String.format("%s\nGenerated code coverage data to %s", currentStream, 48708d13c3cbb0b250b84ca861edb462060556aead9Brett Chabot coverageFilePath)); 4889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } catch (ClassNotFoundException e) { 4899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project reportEmmaError("Is emma jar on classpath?", e); 4909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } catch (SecurityException e) { 4919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project reportEmmaError(e); 4929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } catch (NoSuchMethodException e) { 4939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project reportEmmaError(e); 4949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } catch (IllegalArgumentException e) { 4959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project reportEmmaError(e); 4969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } catch (IllegalAccessException e) { 4979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project reportEmmaError(e); 4989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } catch (InvocationTargetException e) { 4999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project reportEmmaError(e); 5009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private String getCoverageFilePath() { 5049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (mCoverageFilePath == null) { 50551e03642cee5a695d6ff6f5b318925efcab05100Brett Chabot return getTargetContext().getFilesDir().getAbsolutePath() + File.separator + 506ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang DEFAULT_COVERAGE_FILE_NAME; 507ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } else { 5089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project return mCoverageFilePath; 5099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private void reportEmmaError(Exception e) { 513ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang reportEmmaError("", e); 5149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private void reportEmmaError(String hint, Exception e) { 5179066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String msg = "Failed to generate emma coverage. " + hint; 5189066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Log.e(LOG_TAG, msg, e); 5199066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mResults.putString(Instrumentation.REPORT_KEY_STREAMRESULT, "\nError: " + msg); 5209066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // TODO kill this, use status() and prettyprint model for better output 5239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private class StringResultPrinter extends ResultPrinter { 5249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public StringResultPrinter(PrintStream writer) { 5269066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project super(writer); 5279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project synchronized void print(TestResult result, long runTime) { 5309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printHeader(runTime); 5319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project printFooter(result); 5329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5339066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 534ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 5359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 536ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * This class sends status reports back to the IInstrumentationWatcher about 5379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * which suite each test belongs. 5389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 539ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang private class SuiteAssignmentPrinter implements TestListener { 540ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 5419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private Bundle mTestResult; 5429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private long mStartTime; 5439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private long mEndTime; 5449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private boolean mTimingValid; 545ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 5469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public SuiteAssignmentPrinter() { 5479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 548ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 5499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 5509066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * send a status for the start of a each test, so long tests can be seen as "running" 5519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 5529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void startTest(Test test) { 5539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTimingValid = true; 554ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mStartTime = System.currentTimeMillis(); 5559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 556ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 5579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 5589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * @see junit.framework.TestListener#addError(Test, Throwable) 5599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 5609066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void addError(Test test, Throwable t) { 5619066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTimingValid = false; 5629066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5639066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5649066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 5659066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * @see junit.framework.TestListener#addFailure(Test, AssertionFailedError) 5669066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 5679066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void addFailure(Test test, AssertionFailedError t) { 5689066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTimingValid = false; 5699066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5709066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5719066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 5729066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * @see junit.framework.TestListener#endTest(Test) 5739066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 5749066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void endTest(Test test) { 5759066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project float runTime; 5769066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String assignmentSuite; 5779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mEndTime = System.currentTimeMillis(); 5789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult = new Bundle(); 5799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 5809066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (!mTimingValid || mStartTime < 0) { 5819066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project assignmentSuite = "NA"; 5829066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project runTime = -1; 5839066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 5849066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project runTime = mEndTime - mStartTime; 585ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang if (runTime < SMALL_SUITE_MAX_RUNTIME 5869066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project && !InstrumentationTestCase.class.isAssignableFrom(test.getClass())) { 5879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project assignmentSuite = SMALL_SUITE; 5889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else if (runTime < MEDIUM_SUITE_MAX_RUNTIME) { 5899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project assignmentSuite = MEDIUM_SUITE; 5909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 5919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project assignmentSuite = LARGE_SUITE; 5929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 5949066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // Clear mStartTime so that we can verify that it gets set next time. 5959066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mStartTime = -1; 5969066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 597ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, 598ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang test.getClass().getName() + "#" + ((TestCase) test).getName() 5999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project + "\nin " + assignmentSuite + " suite\nrunTime: " 6009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project + String.valueOf(runTime) + "\n"); 6019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult.putFloat(REPORT_KEY_RUN_TIME, runTime); 6029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult.putString(REPORT_KEY_SUITE_ASSIGNMENT, assignmentSuite); 6039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 6049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project sendStatus(0, mTestResult); 6059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 6069066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 607ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 6089066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 6099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * This class sends status reports back to the IInstrumentationWatcher 6109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 611ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang private class WatcherResultPrinter implements TestListener, PerformanceResultsWriter { 6129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project private final Bundle mResultTemplate; 6139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Bundle mTestResult; 6149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int mTestNum = 0; 6159066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project int mTestResultCode = 0; 6169066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String mTestClass = null; 6174f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang PerformanceCollector mPerfCollector = new PerformanceCollector(); 618ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang boolean mIsTimedTest = false; 6194f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang boolean mIncludeDetailedStats = false; 620ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 6219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public WatcherResultPrinter(int numTests) { 6229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mResultTemplate = new Bundle(); 6239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mResultTemplate.putString(Instrumentation.REPORT_KEY_IDENTIFIER, REPORT_VALUE_ID); 6249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mResultTemplate.putInt(REPORT_KEY_NUM_TOTAL, numTests); 6259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 626ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 6279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 628ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * send a status for the start of a each test, so long tests can be seen 629ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang * as "running" 6309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 6319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void startTest(Test test) { 6329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String testClass = test.getClass().getName(); 633ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang String testName = ((TestCase)test).getName(); 6349066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult = new Bundle(mResultTemplate); 6359066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult.putString(REPORT_KEY_NAME_CLASS, testClass); 636ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestResult.putString(REPORT_KEY_NAME_TEST, testName); 6379066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult.putInt(REPORT_KEY_NUM_CURRENT, ++mTestNum); 6389066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // pretty printing 6399066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (testClass != null && !testClass.equals(mTestClass)) { 640ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, 6419066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project String.format("\n%s:", testClass)); 6429066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestClass = testClass; 6439066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } else { 6449066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, ""); 6459066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 6469066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 6479066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // The delay_msec parameter is normally used to provide buffers of idle time 648ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang // for power measurement purposes. To make sure there is a delay before and after 6499066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // every test in a suite, we delay *after* every test (see endTest below) and also 650ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang // delay *before* the first test. So, delay test1 delay test2 delay. 6519066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 6529066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project try { 6539066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (mTestNum == 1) Thread.sleep(mDelayMsec); 6549066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } catch (InterruptedException e) { 6559066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project throw new IllegalStateException(e); 6569066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 6579066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 6589066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project sendStatus(REPORT_VALUE_RESULT_START, mTestResult); 6599066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResultCode = 0; 660ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 661ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mIsTimedTest = false; 6624f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mIncludeDetailedStats = false; 663ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang try { 6644f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang // Look for TimedTest annotation on both test class and test method 6654f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang if (test.getClass().getMethod(testName).isAnnotationPresent(TimedTest.class)) { 6664f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mIsTimedTest = true; 6674f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mIncludeDetailedStats = test.getClass().getMethod(testName).getAnnotation( 6684f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang TimedTest.class).includeDetailedStats(); 6694f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang } else if (test.getClass().isAnnotationPresent(TimedTest.class)) { 6704f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mIsTimedTest = true; 6714f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mIncludeDetailedStats = test.getClass().getAnnotation( 6724f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang TimedTest.class).includeDetailedStats(); 6734f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang } 674ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } catch (SecurityException e) { 675ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang throw new IllegalStateException(e); 676ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } catch (NoSuchMethodException e) { 677ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang throw new IllegalStateException(e); 678ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } 679ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 6804f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang if (mIsTimedTest && mIncludeDetailedStats) { 6814f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mPerfCollector.beginSnapshot(""); 6824f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang } else if (mIsTimedTest) { 6834f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mPerfCollector.startTiming(""); 684ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } 6859066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 686ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 6879066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 6889066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * @see junit.framework.TestListener#addError(Test, Throwable) 6899066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 6909066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void addError(Test test, Throwable t) { 6919066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult.putString(REPORT_KEY_STACK, BaseTestRunner.getFilteredTrace(t)); 6929066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResultCode = REPORT_VALUE_RESULT_ERROR; 6939066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // pretty printing 694ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, 695ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang String.format("\nError in %s:\n%s", 696ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang ((TestCase)test).getName(), BaseTestRunner.getFilteredTrace(t))); 6979066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 6989066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 6999066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 7009066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * @see junit.framework.TestListener#addFailure(Test, AssertionFailedError) 7019066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 7029066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void addFailure(Test test, AssertionFailedError t) { 7039066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult.putString(REPORT_KEY_STACK, BaseTestRunner.getFilteredTrace(t)); 7049066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResultCode = REPORT_VALUE_RESULT_FAILURE; 7059066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // pretty printing 706ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, 707ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang String.format("\nFailure in %s:\n%s", 708ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang ((TestCase)test).getName(), BaseTestRunner.getFilteredTrace(t))); 7099066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 7109066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 7119066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project /** 7129066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project * @see junit.framework.TestListener#endTest(Test) 7139066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project */ 7149066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project public void endTest(Test test) { 7154f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang if (mIsTimedTest && mIncludeDetailedStats) { 7164f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mTestResult.putAll(mPerfCollector.endSnapshot()); 7174f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang } else if (mIsTimedTest) { 7184f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang writeStopTiming(mPerfCollector.stopTiming("")); 719ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } 720ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 7219066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project if (mTestResultCode == 0) { 7229066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project mTestResult.putString(Instrumentation.REPORT_KEY_STREAMRESULT, "."); 7239066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 7249066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project sendStatus(mTestResultCode, mTestResult); 7259066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 726ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang try { // Sleep after every test, if specified 7279066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project Thread.sleep(mDelayMsec); 7289066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } catch (InterruptedException e) { 7299066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project throw new IllegalStateException(e); 7309066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 7319066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 7329066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project 733ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang public void writeBeginSnapshot(String label) { 734ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang // Do nothing 735ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } 736ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 737ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang public void writeEndSnapshot(Bundle results) { 738075997f12e2fb2b646172a92926be0f26f739099Jack Wang // Copy all snapshot data fields into mResults, which is outputted 739075997f12e2fb2b646172a92926be0f26f739099Jack Wang // via Instrumentation.finish 740075997f12e2fb2b646172a92926be0f26f739099Jack Wang mResults.putAll(results); 741ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } 742ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 743ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang public void writeStartTiming(String label) { 744ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang // Do nothing 745ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } 746ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 747ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang public void writeStopTiming(Bundle results) { 748ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang // Copy results into mTestResult by flattening list of iterations, 749ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang // which is outputted via WatcherResultPrinter.endTest 750ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang int i = 0; 751ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang for (Parcelable p : 752ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang results.getParcelableArrayList(PerformanceCollector.METRIC_KEY_ITERATIONS)) { 753ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang Bundle iteration = (Bundle)p; 7544f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang String index = "iteration" + i + "."; 755ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestResult.putString(index + PerformanceCollector.METRIC_KEY_LABEL, 756ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang iteration.getString(PerformanceCollector.METRIC_KEY_LABEL)); 757ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestResult.putLong(index + PerformanceCollector.METRIC_KEY_CPU_TIME, 758ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang iteration.getLong(PerformanceCollector.METRIC_KEY_CPU_TIME)); 759ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang mTestResult.putLong(index + PerformanceCollector.METRIC_KEY_EXECUTION_TIME, 760ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang iteration.getLong(PerformanceCollector.METRIC_KEY_EXECUTION_TIME)); 761ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang i++; 762ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } 763ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang } 764ff1df69dd4835c177c724e1b5f1ba02d1f674047Jack Wang 765075997f12e2fb2b646172a92926be0f26f739099Jack Wang public void writeMeasurement(String label, long value) { 7664f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mTestResult.putLong(label, value); 767075997f12e2fb2b646172a92926be0f26f739099Jack Wang } 768075997f12e2fb2b646172a92926be0f26f739099Jack Wang 769075997f12e2fb2b646172a92926be0f26f739099Jack Wang public void writeMeasurement(String label, float value) { 7704f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mTestResult.putFloat(label, value); 771075997f12e2fb2b646172a92926be0f26f739099Jack Wang } 772075997f12e2fb2b646172a92926be0f26f739099Jack Wang 773075997f12e2fb2b646172a92926be0f26f739099Jack Wang public void writeMeasurement(String label, String value) { 7744f414bd7a76fc7ff063db70d1cdd09303c28cbe5Jack Wang mTestResult.putString(label, value); 775075997f12e2fb2b646172a92926be0f26f739099Jack Wang } 776075997f12e2fb2b646172a92926be0f26f739099Jack Wang 7779066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project // TODO report the end of the cycle 7789066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project } 7799066cfe9886ac131c34d59ed0e2d287b0e3c0087The Android Open Source Project} 780