[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | # |
| 3 | # Copyright 2013 The Chromium Authors. All rights reserved. |
| 4 | # Use of this source code is governed by a BSD-style license that can be |
| 5 | # found in the LICENSE file. |
| 6 | |
[email protected] | 181a5c9 | 2013-09-06 17:11:46 | [diff] [blame] | 7 | """Runs all types of tests from one unified interface.""" |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 8 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 9 | import argparse |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 10 | import collections |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 11 | import itertools |
[email protected] | f7148dd4 | 2013-08-20 14:24:57 | [diff] [blame] | 12 | import logging |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 13 | import os |
[email protected] | 83bb815 | 2013-11-19 15:02:21 | [diff] [blame] | 14 | import signal |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 15 | import sys |
[email protected] | 83bb815 | 2013-11-19 15:02:21 | [diff] [blame] | 16 | import threading |
jbudorick | 256fd53 | 2014-10-24 01:50:13 | [diff] [blame] | 17 | import unittest |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 18 | |
jbudorick | 06162944 | 2015-09-03 18:00:57 | [diff] [blame] | 19 | from devil import base_error |
| 20 | from devil.android import apk_helper |
| 21 | from devil.android import device_blacklist |
| 22 | from devil.android import device_errors |
| 23 | from devil.android import device_utils |
| 24 | from devil.android import ports |
| 25 | from devil.utils import reraiser_thread |
| 26 | from devil.utils import run_tests_helper |
| 27 | |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 28 | from pylib import constants |
[email protected] | c0662e09 | 2013-11-12 11:51:25 | [diff] [blame] | 29 | from pylib import forwarder |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 30 | from pylib.base import base_test_result |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 31 | from pylib.base import environment_factory |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 32 | from pylib.base import test_dispatcher |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 33 | from pylib.base import test_instance_factory |
| 34 | from pylib.base import test_run_factory |
[email protected] | 6b6abac6d | 2013-10-03 11:56:38 | [diff] [blame] | 35 | from pylib.linker import setup as linker_setup |
[email protected] | 37ee0c79 | 2013-08-06 19:10:13 | [diff] [blame] | 36 | from pylib.host_driven import setup as host_driven_setup |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 37 | from pylib.instrumentation import setup as instrumentation_setup |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 38 | from pylib.instrumentation import test_options as instrumentation_test_options |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 39 | from pylib.junit import setup as junit_setup |
| 40 | from pylib.junit import test_dispatcher as junit_dispatcher |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 41 | from pylib.monkey import setup as monkey_setup |
| 42 | from pylib.monkey import test_options as monkey_test_options |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 43 | from pylib.perf import setup as perf_setup |
| 44 | from pylib.perf import test_options as perf_test_options |
| 45 | from pylib.perf import test_runner as perf_test_runner |
jbudorick | b8c4207 | 2014-12-01 18:07:54 | [diff] [blame] | 46 | from pylib.results import json_results |
| 47 | from pylib.results import report_results |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 48 | from pylib.uiautomator import setup as uiautomator_setup |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 49 | from pylib.uiautomator import test_options as uiautomator_test_options |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 50 | |
| 51 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 52 | def AddCommonOptions(parser): |
| 53 | """Adds all common options to |parser|.""" |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 54 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 55 | group = parser.add_argument_group('Common Options') |
| 56 | |
[email protected] | dfffbcbc | 2013-09-17 22:06:01 | [diff] [blame] | 57 | default_build_type = os.environ.get('BUILDTYPE', 'Debug') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 58 | |
| 59 | debug_or_release_group = group.add_mutually_exclusive_group() |
| 60 | debug_or_release_group.add_argument( |
| 61 | '--debug', action='store_const', const='Debug', dest='build_type', |
| 62 | default=default_build_type, |
| 63 | help=('If set, run test suites under out/Debug. ' |
| 64 | 'Default is env var BUILDTYPE or Debug.')) |
| 65 | debug_or_release_group.add_argument( |
| 66 | '--release', action='store_const', const='Release', dest='build_type', |
| 67 | help=('If set, run test suites under out/Release. ' |
| 68 | 'Default is env var BUILDTYPE or Debug.')) |
| 69 | |
| 70 | group.add_argument('--build-directory', dest='build_directory', |
| 71 | help=('Path to the directory in which build files are' |
| 72 | ' located (should not include build type)')) |
| 73 | group.add_argument('--output-directory', dest='output_directory', |
| 74 | help=('Path to the directory in which build files are' |
| 75 | ' located (must include build type). This will take' |
| 76 | ' precedence over --debug, --release and' |
| 77 | ' --build-directory')) |
| 78 | group.add_argument('--num_retries', dest='num_retries', type=int, default=2, |
| 79 | help=('Number of retries for a test before ' |
| 80 | 'giving up (default: %(default)s).')) |
| 81 | group.add_argument('-v', |
| 82 | '--verbose', |
| 83 | dest='verbose_count', |
| 84 | default=0, |
| 85 | action='count', |
| 86 | help='Verbose level (multiple times for more)') |
| 87 | group.add_argument('--flakiness-dashboard-server', |
| 88 | dest='flakiness_dashboard_server', |
| 89 | help=('Address of the server that is hosting the ' |
| 90 | 'Chrome for Android flakiness dashboard.')) |
| 91 | group.add_argument('--enable-platform-mode', action='store_true', |
| 92 | help=('Run the test scripts in platform mode, which ' |
| 93 | 'conceptually separates the test runner from the ' |
| 94 | '"device" (local or remote, real or emulated) on ' |
| 95 | 'which the tests are running. [experimental]')) |
| 96 | group.add_argument('-e', '--environment', default='local', |
| 97 | choices=constants.VALID_ENVIRONMENTS, |
| 98 | help='Test environment to run in (default: %(default)s).') |
| 99 | group.add_argument('--adb-path', |
| 100 | help=('Specify the absolute path of the adb binary that ' |
| 101 | 'should be used.')) |
| 102 | group.add_argument('--json-results-file', dest='json_results_file', |
| 103 | help='If set, will dump results in JSON form ' |
| 104 | 'to specified file.') |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 105 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 106 | def ProcessCommonOptions(args): |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 107 | """Processes and handles all common options.""" |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 108 | run_tests_helper.SetLogLevel(args.verbose_count) |
| 109 | constants.SetBuildType(args.build_type) |
| 110 | if args.build_directory: |
| 111 | constants.SetBuildDirectory(args.build_directory) |
| 112 | if args.output_directory: |
mikecase | 0aea9c5 | 2015-04-30 00:12:33 | [diff] [blame] | 113 | constants.SetOutputDirectory(args.output_directory) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 114 | if args.adb_path: |
| 115 | constants.SetAdbPath(args.adb_path) |
mikecase | 48e16bf | 2014-11-19 22:46:45 | [diff] [blame] | 116 | # Some things such as Forwarder require ADB to be in the environment path. |
| 117 | adb_dir = os.path.dirname(constants.GetAdbPath()) |
| 118 | if adb_dir and adb_dir not in os.environ['PATH'].split(os.pathsep): |
| 119 | os.environ['PATH'] = adb_dir + os.pathsep + os.environ['PATH'] |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 120 | |
| 121 | |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 122 | def AddRemoteDeviceOptions(parser): |
| 123 | group = parser.add_argument_group('Remote Device Options') |
| 124 | |
rnephew | efe44b4 | 2015-02-04 04:45:15 | [diff] [blame] | 125 | group.add_argument('--trigger', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 126 | help=('Only triggers the test if set. Stores test_run_id ' |
| 127 | 'in given file path. ')) |
rnephew | efe44b4 | 2015-02-04 04:45:15 | [diff] [blame] | 128 | group.add_argument('--collect', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 129 | help=('Only collects the test results if set. ' |
| 130 | 'Gets test_run_id from given file path.')) |
rnephew | efe44b4 | 2015-02-04 04:45:15 | [diff] [blame] | 131 | group.add_argument('--remote-device', action='append', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 132 | help='Device type to run test on.') |
rnephew | efe44b4 | 2015-02-04 04:45:15 | [diff] [blame] | 133 | group.add_argument('--results-path', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 134 | help='File path to download results to.') |
rnephew | 7f1e205 | 2014-12-12 23:00:11 | [diff] [blame] | 135 | group.add_argument('--api-protocol', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 136 | help='HTTP protocol to use. (http or https)') |
rnephew | efe44b4 | 2015-02-04 04:45:15 | [diff] [blame] | 137 | group.add_argument('--api-address', |
| 138 | help='Address to send HTTP requests.') |
| 139 | group.add_argument('--api-port', |
| 140 | help='Port to send HTTP requests to.') |
| 141 | group.add_argument('--runner-type', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 142 | help='Type of test to run as.') |
rnephew | efe44b4 | 2015-02-04 04:45:15 | [diff] [blame] | 143 | group.add_argument('--runner-package', |
| 144 | help='Package name of test.') |
| 145 | group.add_argument('--device-type', |
rnephew | a46fc56 | 2015-01-23 16:00:14 | [diff] [blame] | 146 | choices=constants.VALID_DEVICE_TYPES, |
| 147 | help=('Type of device to run on. iOS or android')) |
rnephew | efe44b4 | 2015-02-04 04:45:15 | [diff] [blame] | 148 | group.add_argument('--device-oem', action='append', |
| 149 | help='Device OEM to run on.') |
| 150 | group.add_argument('--remote-device-file', |
| 151 | help=('File with JSON to select remote device. ' |
| 152 | 'Overrides all other flags.')) |
rnephew | c9ae8f5 | 2015-02-13 03:02:55 | [diff] [blame] | 153 | group.add_argument('--remote-device-timeout', type=int, |
| 154 | help='Times to retry finding remote device') |
mikecase | 520cbbb5 | 2015-04-21 18:51:18 | [diff] [blame] | 155 | group.add_argument('--network-config', type=int, |
| 156 | help='Integer that specifies the network environment ' |
| 157 | 'that the tests will be run in.') |
rnephew | efe44b4 | 2015-02-04 04:45:15 | [diff] [blame] | 158 | |
| 159 | device_os_group = group.add_mutually_exclusive_group() |
| 160 | device_os_group.add_argument('--remote-device-minimum-os', |
| 161 | help='Minimum OS on device.') |
| 162 | device_os_group.add_argument('--remote-device-os', action='append', |
| 163 | help='OS to have on the device.') |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 164 | |
| 165 | api_secret_group = group.add_mutually_exclusive_group() |
| 166 | api_secret_group.add_argument('--api-secret', default='', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 167 | help='API secret for remote devices.') |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 168 | api_secret_group.add_argument('--api-secret-file', default='', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 169 | help='Path to file that contains API secret.') |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 170 | |
| 171 | api_key_group = group.add_mutually_exclusive_group() |
| 172 | api_key_group.add_argument('--api-key', default='', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 173 | help='API key for remote devices.') |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 174 | api_key_group.add_argument('--api-key-file', default='', |
jbudorick | e6c56015 | 2015-01-13 23:49:28 | [diff] [blame] | 175 | help='Path to file that contains API key.') |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 176 | |
| 177 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 178 | def AddDeviceOptions(parser): |
| 179 | """Adds device options to |parser|.""" |
| 180 | group = parser.add_argument_group(title='Device Options') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 181 | group.add_argument('--tool', |
| 182 | dest='tool', |
| 183 | help=('Run the test under a tool ' |
| 184 | '(use --tool help to list them)')) |
| 185 | group.add_argument('-d', '--device', dest='test_device', |
| 186 | help=('Target device for the test suite ' |
| 187 | 'to run on.')) |
jbudorick | dde688fb | 2015-08-27 03:00:17 | [diff] [blame] | 188 | group.add_argument('--blacklist-file', help='Device blacklist file.') |
jbudorick | 256fd53 | 2014-10-24 01:50:13 | [diff] [blame] | 189 | |
| 190 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 191 | def AddGTestOptions(parser): |
| 192 | """Adds gtest options to |parser|.""" |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 193 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 194 | group = parser.add_argument_group('GTest Options') |
jbudorick | 15cdcd5 | 2014-12-03 19:58:49 | [diff] [blame] | 195 | group.add_argument('-s', '--suite', dest='suite_name', |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 196 | nargs='+', metavar='SUITE_NAME', required=True, |
jbudorick | 277f231 | 2015-09-24 16:37:43 | [diff] [blame] | 197 | help='Executable name of the test suite to run.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 198 | group.add_argument('--gtest_also_run_disabled_tests', |
| 199 | '--gtest-also-run-disabled-tests', |
| 200 | dest='run_disabled', action='store_true', |
| 201 | help='Also run disabled tests if applicable.') |
| 202 | group.add_argument('-a', '--test-arguments', dest='test_arguments', |
| 203 | default='', |
| 204 | help='Additional arguments to pass to the test.') |
| 205 | group.add_argument('-t', dest='timeout', type=int, default=60, |
| 206 | help='Timeout to wait for each test ' |
| 207 | '(default: %(default)s).') |
| 208 | group.add_argument('--isolate_file_path', |
| 209 | '--isolate-file-path', |
| 210 | dest='isolate_file_path', |
| 211 | help='.isolate file path to override the default ' |
| 212 | 'path') |
jbudorick | 5ee4589 | 2015-06-10 18:46:22 | [diff] [blame] | 213 | group.add_argument('--app-data-file', action='append', dest='app_data_files', |
| 214 | help='A file path relative to the app data directory ' |
| 215 | 'that should be saved to the host.') |
| 216 | group.add_argument('--app-data-file-dir', |
| 217 | help='Host directory to which app data files will be' |
| 218 | ' saved. Used with --app-data-file.') |
mlliu | d7f9fe9 | 2015-06-15 19:36:56 | [diff] [blame] | 219 | group.add_argument('--delete-stale-data', dest='delete_stale_data', |
| 220 | action='store_true', |
| 221 | help='Delete stale test data on the device.') |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 222 | group.add_argument('--repeat', '--gtest_repeat', '--gtest-repeat', |
| 223 | dest='repeat', type=int, default=0, |
| 224 | help='Number of times to repeat the specified set of ' |
| 225 | 'tests.') |
jbudorick | 442a693 | 2015-02-03 03:01:15 | [diff] [blame] | 226 | |
| 227 | filter_group = group.add_mutually_exclusive_group() |
| 228 | filter_group.add_argument('-f', '--gtest_filter', '--gtest-filter', |
| 229 | dest='test_filter', |
| 230 | help='googletest-style filter string.') |
| 231 | filter_group.add_argument('--gtest-filter-file', dest='test_filter_file', |
| 232 | help='Path to file that contains googletest-style ' |
| 233 | 'filter strings. (Lines will be joined with ' |
| 234 | '":" to create a single filter string.)') |
| 235 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 236 | AddDeviceOptions(parser) |
| 237 | AddCommonOptions(parser) |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 238 | AddRemoteDeviceOptions(parser) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 239 | |
| 240 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 241 | def AddLinkerTestOptions(parser): |
| 242 | group = parser.add_argument_group('Linker Test Options') |
| 243 | group.add_argument('-f', '--gtest-filter', dest='test_filter', |
| 244 | help='googletest-style filter string.') |
| 245 | AddCommonOptions(parser) |
| 246 | AddDeviceOptions(parser) |
[email protected] | 6b6abac6d | 2013-10-03 11:56:38 | [diff] [blame] | 247 | |
| 248 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 249 | def AddJavaTestOptions(argument_group): |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 250 | """Adds the Java test options to |option_parser|.""" |
| 251 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 252 | argument_group.add_argument( |
| 253 | '-f', '--test-filter', dest='test_filter', |
| 254 | help=('Test filter (if not fully qualified, will run all matches).')) |
| 255 | argument_group.add_argument( |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 256 | '--repeat', dest='repeat', type=int, default=0, |
| 257 | help='Number of times to repeat the specified set of tests.') |
| 258 | argument_group.add_argument( |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 259 | '-A', '--annotation', dest='annotation_str', |
| 260 | help=('Comma-separated list of annotations. Run only tests with any of ' |
| 261 | 'the given annotations. An annotation can be either a key or a ' |
| 262 | 'key-values pair. A test that has no annotation is considered ' |
| 263 | '"SmallTest".')) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 264 | argument_group.add_argument( |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 265 | '-E', '--exclude-annotation', dest='exclude_annotation_str', |
| 266 | help=('Comma-separated list of annotations. Exclude tests with these ' |
| 267 | 'annotations.')) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 268 | argument_group.add_argument( |
jbudorick | cbcc115d | 2014-09-18 17:50:59 | [diff] [blame] | 269 | '--screenshot', dest='screenshot_failures', action='store_true', |
| 270 | help='Capture screenshots of test failures') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 271 | argument_group.add_argument( |
jbudorick | cbcc115d | 2014-09-18 17:50:59 | [diff] [blame] | 272 | '--save-perf-json', action='store_true', |
| 273 | help='Saves the JSON file for each UI Perf test.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 274 | argument_group.add_argument( |
jbudorick | cbcc115d | 2014-09-18 17:50:59 | [diff] [blame] | 275 | '--official-build', action='store_true', help='Run official build tests.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 276 | argument_group.add_argument( |
jbudorick | cbcc115d | 2014-09-18 17:50:59 | [diff] [blame] | 277 | '--test_data', '--test-data', action='append', default=[], |
| 278 | help=('Each instance defines a directory of test data that should be ' |
| 279 | 'copied to the target(s) before running the tests. The argument ' |
| 280 | 'should be of the form <target>:<source>, <target> is relative to ' |
| 281 | 'the device data directory, and <source> is relative to the ' |
| 282 | 'chromium build directory.')) |
davileen | 98efad1 | 2015-01-05 19:48:21 | [diff] [blame] | 283 | argument_group.add_argument( |
| 284 | '--disable-dalvik-asserts', dest='set_asserts', action='store_false', |
| 285 | default=True, help='Removes the dalvik.vm.enableassertions property') |
| 286 | |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 287 | |
| 288 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 289 | def ProcessJavaTestOptions(args): |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 290 | """Processes options/arguments and populates |options| with defaults.""" |
| 291 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 292 | # TODO(jbudorick): Handle most of this function in argparse. |
| 293 | if args.annotation_str: |
| 294 | args.annotations = args.annotation_str.split(',') |
| 295 | elif args.test_filter: |
| 296 | args.annotations = [] |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 297 | else: |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 298 | args.annotations = ['Smoke', 'SmallTest', 'MediumTest', 'LargeTest', |
| 299 | 'EnormousTest', 'IntegrationTest'] |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 300 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 301 | if args.exclude_annotation_str: |
| 302 | args.exclude_annotations = args.exclude_annotation_str.split(',') |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 303 | else: |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 304 | args.exclude_annotations = [] |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 305 | |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 306 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 307 | def AddInstrumentationTestOptions(parser): |
| 308 | """Adds Instrumentation test options to |parser|.""" |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 309 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 310 | parser.usage = '%(prog)s [options]' |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 311 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 312 | group = parser.add_argument_group('Instrumentation Test Options') |
| 313 | AddJavaTestOptions(group) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 314 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 315 | java_or_python_group = group.add_mutually_exclusive_group() |
| 316 | java_or_python_group.add_argument( |
| 317 | '-j', '--java-only', action='store_false', |
| 318 | dest='run_python_tests', default=True, help='Run only the Java tests.') |
| 319 | java_or_python_group.add_argument( |
| 320 | '-p', '--python-only', action='store_false', |
| 321 | dest='run_java_tests', default=True, |
| 322 | help='Run only the host-driven tests.') |
| 323 | |
| 324 | group.add_argument('--host-driven-root', |
| 325 | help='Root of the host-driven tests.') |
| 326 | group.add_argument('-w', '--wait_debugger', dest='wait_for_debugger', |
| 327 | action='store_true', |
| 328 | help='Wait for debugger.') |
jbudorick | 911be58d | 2015-01-13 02:51:06 | [diff] [blame] | 329 | group.add_argument('--apk-under-test', dest='apk_under_test', |
| 330 | help=('the name of the apk under test.')) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 331 | group.add_argument('--test-apk', dest='test_apk', required=True, |
| 332 | help=('The name of the apk containing the tests ' |
| 333 | '(without the .apk extension; ' |
| 334 | 'e.g. "ContentShellTest").')) |
mikecase | e725862 | 2015-09-29 13:47:35 | [diff] [blame^] | 335 | group.add_argument('--additional-apk', action='append', |
| 336 | dest='additional_apks', |
| 337 | help='Additional apk that must be installed on ' |
| 338 | 'the device when the tests are run') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 339 | group.add_argument('--coverage-dir', |
| 340 | help=('Directory in which to place all generated ' |
| 341 | 'EMMA coverage files.')) |
| 342 | group.add_argument('--device-flags', dest='device_flags', default='', |
| 343 | help='The relative filepath to a file containing ' |
| 344 | 'command-line flags to set on the device') |
jbudorick | 911be58d | 2015-01-13 02:51:06 | [diff] [blame] | 345 | group.add_argument('--device-flags-file', default='', |
| 346 | help='The relative filepath to a file containing ' |
| 347 | 'command-line flags to set on the device') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 348 | group.add_argument('--isolate_file_path', |
| 349 | '--isolate-file-path', |
| 350 | dest='isolate_file_path', |
| 351 | help='.isolate file path to override the default ' |
| 352 | 'path') |
mlliu | d7f9fe9 | 2015-06-15 19:36:56 | [diff] [blame] | 353 | group.add_argument('--delete-stale-data', dest='delete_stale_data', |
| 354 | action='store_true', |
| 355 | help='Delete stale test data on the device.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 356 | |
| 357 | AddCommonOptions(parser) |
| 358 | AddDeviceOptions(parser) |
rnephew | e416dff | 2015-01-21 21:26:37 | [diff] [blame] | 359 | AddRemoteDeviceOptions(parser) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 360 | |
| 361 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 362 | def ProcessInstrumentationOptions(args): |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 363 | """Processes options/arguments and populate |options| with defaults. |
| 364 | |
| 365 | Args: |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 366 | args: argparse.Namespace object. |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 367 | |
| 368 | Returns: |
| 369 | An InstrumentationOptions named tuple which contains all options relevant to |
| 370 | instrumentation tests. |
| 371 | """ |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 372 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 373 | ProcessJavaTestOptions(args) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 374 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 375 | if not args.host_driven_root: |
| 376 | args.run_python_tests = False |
[email protected] | 37ee0c79 | 2013-08-06 19:10:13 | [diff] [blame] | 377 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 378 | args.test_apk_path = os.path.join( |
[email protected] | 2eea487 | 2014-07-28 23:06:17 | [diff] [blame] | 379 | constants.GetOutDirectory(), |
| 380 | constants.SDK_BUILD_APKS_DIR, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 381 | '%s.apk' % args.test_apk) |
| 382 | args.test_apk_jar_path = os.path.join( |
[email protected] | ae68d4a | 2013-09-24 21:57:15 | [diff] [blame] | 383 | constants.GetOutDirectory(), |
| 384 | constants.SDK_BUILD_TEST_JAVALIB_DIR, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 385 | '%s.jar' % args.test_apk) |
yusufo | 72c598c0 | 2015-07-16 23:40:20 | [diff] [blame] | 386 | args.test_support_apk_path = '%sSupport%s' % ( |
| 387 | os.path.splitext(args.test_apk_path)) |
[email protected] | 5e2f3f6 | 2014-06-23 12:31:46 | [diff] [blame] | 388 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 389 | args.test_runner = apk_helper.GetInstrumentationName(args.test_apk_path) |
[email protected] | 5e2f3f6 | 2014-06-23 12:31:46 | [diff] [blame] | 390 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 391 | # TODO(jbudorick): Get rid of InstrumentationOptions. |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 392 | return instrumentation_test_options.InstrumentationOptions( |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 393 | args.tool, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 394 | args.annotations, |
| 395 | args.exclude_annotations, |
| 396 | args.test_filter, |
| 397 | args.test_data, |
| 398 | args.save_perf_json, |
| 399 | args.screenshot_failures, |
| 400 | args.wait_for_debugger, |
| 401 | args.coverage_dir, |
| 402 | args.test_apk, |
| 403 | args.test_apk_path, |
| 404 | args.test_apk_jar_path, |
| 405 | args.test_runner, |
| 406 | args.test_support_apk_path, |
| 407 | args.device_flags, |
davileen | 98efad1 | 2015-01-05 19:48:21 | [diff] [blame] | 408 | args.isolate_file_path, |
mlliu | d7f9fe9 | 2015-06-15 19:36:56 | [diff] [blame] | 409 | args.set_asserts, |
| 410 | args.delete_stale_data |
[email protected] | 5e2f3f6 | 2014-06-23 12:31:46 | [diff] [blame] | 411 | ) |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 412 | |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 413 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 414 | def AddUIAutomatorTestOptions(parser): |
| 415 | """Adds UI Automator test options to |parser|.""" |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 416 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 417 | group = parser.add_argument_group('UIAutomator Test Options') |
| 418 | AddJavaTestOptions(group) |
| 419 | group.add_argument( |
| 420 | '--package', required=True, choices=constants.PACKAGE_INFO.keys(), |
| 421 | metavar='PACKAGE', help='Package under test.') |
| 422 | group.add_argument( |
| 423 | '--test-jar', dest='test_jar', required=True, |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 424 | help=('The name of the dexed jar containing the tests (without the ' |
| 425 | '.dex.jar extension). Alternatively, this can be a full path ' |
| 426 | 'to the jar.')) |
| 427 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 428 | AddCommonOptions(parser) |
| 429 | AddDeviceOptions(parser) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 430 | |
| 431 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 432 | def ProcessUIAutomatorOptions(args): |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 433 | """Processes UIAutomator options/arguments. |
| 434 | |
| 435 | Args: |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 436 | args: argparse.Namespace object. |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 437 | |
| 438 | Returns: |
| 439 | A UIAutomatorOptions named tuple which contains all options relevant to |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 440 | uiautomator tests. |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 441 | """ |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 442 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 443 | ProcessJavaTestOptions(args) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 444 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 445 | if os.path.exists(args.test_jar): |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 446 | # The dexed JAR is fully qualified, assume the info JAR lives along side. |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 447 | args.uiautomator_jar = args.test_jar |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 448 | else: |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 449 | args.uiautomator_jar = os.path.join( |
[email protected] | ae68d4a | 2013-09-24 21:57:15 | [diff] [blame] | 450 | constants.GetOutDirectory(), |
| 451 | constants.SDK_BUILD_JAVALIB_DIR, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 452 | '%s.dex.jar' % args.test_jar) |
| 453 | args.uiautomator_info_jar = ( |
| 454 | args.uiautomator_jar[:args.uiautomator_jar.find('.dex.jar')] + |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 455 | '_java.jar') |
| 456 | |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 457 | return uiautomator_test_options.UIAutomatorOptions( |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 458 | args.tool, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 459 | args.annotations, |
| 460 | args.exclude_annotations, |
| 461 | args.test_filter, |
| 462 | args.test_data, |
| 463 | args.save_perf_json, |
| 464 | args.screenshot_failures, |
| 465 | args.uiautomator_jar, |
| 466 | args.uiautomator_info_jar, |
davileen | 98efad1 | 2015-01-05 19:48:21 | [diff] [blame] | 467 | args.package, |
| 468 | args.set_asserts) |
[email protected] | 2a68422 | 2013-08-01 16:59:22 | [diff] [blame] | 469 | |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 470 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 471 | def AddJUnitTestOptions(parser): |
| 472 | """Adds junit test options to |parser|.""" |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 473 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 474 | group = parser.add_argument_group('JUnit Test Options') |
| 475 | group.add_argument( |
| 476 | '-s', '--test-suite', dest='test_suite', required=True, |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 477 | help=('JUnit test suite to run.')) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 478 | group.add_argument( |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 479 | '-f', '--test-filter', dest='test_filter', |
| 480 | help='Filters tests googletest-style.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 481 | group.add_argument( |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 482 | '--package-filter', dest='package_filter', |
| 483 | help='Filters tests by package.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 484 | group.add_argument( |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 485 | '--runner-filter', dest='runner_filter', |
| 486 | help='Filters tests by runner class. Must be fully qualified.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 487 | group.add_argument( |
| 488 | '--sdk-version', dest='sdk_version', type=int, |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 489 | help='The Android SDK version.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 490 | AddCommonOptions(parser) |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 491 | |
| 492 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 493 | def AddMonkeyTestOptions(parser): |
| 494 | """Adds monkey test options to |parser|.""" |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 495 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 496 | group = parser.add_argument_group('Monkey Test Options') |
| 497 | group.add_argument( |
| 498 | '--package', required=True, choices=constants.PACKAGE_INFO.keys(), |
| 499 | metavar='PACKAGE', help='Package under test.') |
| 500 | group.add_argument( |
| 501 | '--event-count', default=10000, type=int, |
| 502 | help='Number of events to generate (default: %(default)s).') |
| 503 | group.add_argument( |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 504 | '--category', default='', |
[email protected] | fb81b98 | 2013-08-09 00:07:12 | [diff] [blame] | 505 | help='A list of allowed categories.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 506 | group.add_argument( |
| 507 | '--throttle', default=100, type=int, |
| 508 | help='Delay between events (ms) (default: %(default)s). ') |
| 509 | group.add_argument( |
| 510 | '--seed', type=int, |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 511 | help=('Seed value for pseudo-random generator. Same seed value generates ' |
| 512 | 'the same sequence of events. Seed is randomized by default.')) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 513 | group.add_argument( |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 514 | '--extra-args', default='', |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 515 | help=('String of other args to pass to the command verbatim.')) |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 516 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 517 | AddCommonOptions(parser) |
| 518 | AddDeviceOptions(parser) |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 519 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 520 | def ProcessMonkeyTestOptions(args): |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 521 | """Processes all monkey test options. |
| 522 | |
| 523 | Args: |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 524 | args: argparse.Namespace object. |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 525 | |
| 526 | Returns: |
| 527 | A MonkeyOptions named tuple which contains all options relevant to |
| 528 | monkey tests. |
| 529 | """ |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 530 | # TODO(jbudorick): Handle this directly in argparse with nargs='+' |
| 531 | category = args.category |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 532 | if category: |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 533 | category = args.category.split(',') |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 534 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 535 | # TODO(jbudorick): Get rid of MonkeyOptions. |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 536 | return monkey_test_options.MonkeyOptions( |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 537 | args.verbose_count, |
| 538 | args.package, |
| 539 | args.event_count, |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 540 | category, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 541 | args.throttle, |
| 542 | args.seed, |
| 543 | args.extra_args) |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 544 | |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 545 | def AddUirobotTestOptions(parser): |
| 546 | """Adds uirobot test options to |option_parser|.""" |
| 547 | group = parser.add_argument_group('Uirobot Test Options') |
| 548 | |
rnephew | efe44b4 | 2015-02-04 04:45:15 | [diff] [blame] | 549 | group.add_argument('--app-under-test', required=True, |
| 550 | help='APK to run tests on.') |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 551 | group.add_argument( |
| 552 | '--minutes', default=5, type=int, |
jbudorick | 676b120 | 2015-02-06 22:02:27 | [diff] [blame] | 553 | help='Number of minutes to run uirobot test [default: %(default)s].') |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 554 | |
| 555 | AddCommonOptions(parser) |
| 556 | AddDeviceOptions(parser) |
| 557 | AddRemoteDeviceOptions(parser) |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 558 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 559 | def AddPerfTestOptions(parser): |
| 560 | """Adds perf test options to |parser|.""" |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 561 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 562 | group = parser.add_argument_group('Perf Test Options') |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 563 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 564 | class SingleStepAction(argparse.Action): |
| 565 | def __call__(self, parser, namespace, values, option_string=None): |
| 566 | if values and not namespace.single_step: |
| 567 | parser.error('single step command provided, ' |
| 568 | 'but --single-step not specified.') |
| 569 | elif namespace.single_step and not values: |
| 570 | parser.error('--single-step specified, ' |
| 571 | 'but no single step command provided.') |
| 572 | setattr(namespace, self.dest, values) |
| 573 | |
| 574 | step_group = group.add_mutually_exclusive_group(required=True) |
| 575 | # TODO(jbudorick): Revise --single-step to use argparse.REMAINDER. |
| 576 | # This requires removing "--" from client calls. |
| 577 | step_group.add_argument( |
| 578 | '--single-step', action='store_true', |
[email protected] | def4bce | 2013-11-12 12:59:52 | [diff] [blame] | 579 | help='Execute the given command with retries, but only print the result ' |
| 580 | 'for the "most successful" round.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 581 | step_group.add_argument( |
[email protected] | 181a5c9 | 2013-09-06 17:11:46 | [diff] [blame] | 582 | '--steps', |
[email protected] | def4bce | 2013-11-12 12:59:52 | [diff] [blame] | 583 | help='JSON file containing the list of commands to run.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 584 | step_group.add_argument( |
| 585 | '--print-step', |
| 586 | help='The name of a previously executed perf step to print.') |
| 587 | |
| 588 | group.add_argument( |
peter | bd4e73d | 2014-12-03 15:47:36 | [diff] [blame] | 589 | '--output-json-list', |
| 590 | help='Write a simple list of names from --steps into the given file.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 591 | group.add_argument( |
peter | bd4e73d | 2014-12-03 15:47:36 | [diff] [blame] | 592 | '--collect-chartjson-data', |
| 593 | action='store_true', |
| 594 | help='Cache the chartjson output from each step for later use.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 595 | group.add_argument( |
peter | bd4e73d | 2014-12-03 15:47:36 | [diff] [blame] | 596 | '--output-chartjson-data', |
| 597 | default='', |
| 598 | help='Write out chartjson into the given file.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 599 | group.add_argument( |
perezju | 67cf7f1 | 2015-09-29 11:39:05 | [diff] [blame] | 600 | '--get-output-dir-archive', metavar='FILENAME', |
| 601 | help='Write the chached output directory archived by a step into the' |
| 602 | ' given ZIP file.') |
| 603 | group.add_argument( |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 604 | '--flaky-steps', |
| 605 | help=('A JSON file containing steps that are flaky ' |
| 606 | 'and will have its exit code ignored.')) |
| 607 | group.add_argument( |
[email protected] | 181a5c9 | 2013-09-06 17:11:46 | [diff] [blame] | 608 | '--no-timeout', action='store_true', |
| 609 | help=('Do not impose a timeout. Each perf step is responsible for ' |
| 610 | 'implementing the timeout logic.')) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 611 | group.add_argument( |
[email protected] | 650487c | 2013-09-30 11:40:49 | [diff] [blame] | 612 | '-f', '--test-filter', |
| 613 | help=('Test filter (will match against the names listed in --steps).')) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 614 | group.add_argument( |
| 615 | '--dry-run', action='store_true', |
[email protected] | 650487c | 2013-09-30 11:40:49 | [diff] [blame] | 616 | help='Just print the steps without executing.') |
jbudorick | 5cfff87 | 2015-07-01 18:46:13 | [diff] [blame] | 617 | # Uses 0.1 degrees C because that's what Android does. |
| 618 | group.add_argument( |
| 619 | '--max-battery-temp', type=int, |
| 620 | help='Only start tests when the battery is at or below the given ' |
| 621 | 'temperature (0.1 C)') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 622 | group.add_argument('single_step_command', nargs='*', action=SingleStepAction, |
| 623 | help='If --single-step is specified, the command to run.') |
rnephew | dde05da8 | 2015-07-09 20:31:01 | [diff] [blame] | 624 | group.add_argument('--min-battery-level', type=int, |
| 625 | help='Only starts tests when the battery is charged above ' |
| 626 | 'given level.') |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 627 | AddCommonOptions(parser) |
| 628 | AddDeviceOptions(parser) |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 629 | |
| 630 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 631 | def ProcessPerfTestOptions(args): |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 632 | """Processes all perf test options. |
| 633 | |
| 634 | Args: |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 635 | args: argparse.Namespace object. |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 636 | |
| 637 | Returns: |
| 638 | A PerfOptions named tuple which contains all options relevant to |
| 639 | perf tests. |
| 640 | """ |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 641 | # TODO(jbudorick): Move single_step handling down into the perf tests. |
| 642 | if args.single_step: |
| 643 | args.single_step = ' '.join(args.single_step_command) |
| 644 | # TODO(jbudorick): Get rid of PerfOptions. |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 645 | return perf_test_options.PerfOptions( |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 646 | args.steps, args.flaky_steps, args.output_json_list, |
| 647 | args.print_step, args.no_timeout, args.test_filter, |
| 648 | args.dry_run, args.single_step, args.collect_chartjson_data, |
perezju | 67cf7f1 | 2015-09-29 11:39:05 | [diff] [blame] | 649 | args.output_chartjson_data, args.get_output_dir_archive, |
| 650 | args.max_battery_temp, args.min_battery_level) |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 651 | |
| 652 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 653 | def AddPythonTestOptions(parser): |
| 654 | group = parser.add_argument_group('Python Test Options') |
| 655 | group.add_argument( |
| 656 | '-s', '--suite', dest='suite_name', metavar='SUITE_NAME', |
| 657 | choices=constants.PYTHON_UNIT_TEST_SUITES.keys(), |
| 658 | help='Name of the test suite to run.') |
| 659 | AddCommonOptions(parser) |
jbudorick | 256fd53 | 2014-10-24 01:50:13 | [diff] [blame] | 660 | |
| 661 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 662 | def _RunLinkerTests(args, devices): |
[email protected] | 6b6abac6d | 2013-10-03 11:56:38 | [diff] [blame] | 663 | """Subcommand of RunTestsCommands which runs linker tests.""" |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 664 | runner_factory, tests = linker_setup.Setup(args, devices) |
[email protected] | 6b6abac6d | 2013-10-03 11:56:38 | [diff] [blame] | 665 | |
| 666 | results, exit_code = test_dispatcher.RunTests( |
| 667 | tests, runner_factory, devices, shard=True, test_timeout=60, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 668 | num_retries=args.num_retries) |
[email protected] | 6b6abac6d | 2013-10-03 11:56:38 | [diff] [blame] | 669 | |
| 670 | report_results.LogFull( |
| 671 | results=results, |
| 672 | test_type='Linker test', |
[email protected] | 93c9f9b | 2014-02-10 16:19:22 | [diff] [blame] | 673 | test_package='ChromiumLinkerTest') |
[email protected] | 6b6abac6d | 2013-10-03 11:56:38 | [diff] [blame] | 674 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 675 | if args.json_results_file: |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 676 | json_results.GenerateJsonResultsFile([results], args.json_results_file) |
jbudorick | b8c4207 | 2014-12-01 18:07:54 | [diff] [blame] | 677 | |
[email protected] | 6b6abac6d | 2013-10-03 11:56:38 | [diff] [blame] | 678 | return exit_code |
| 679 | |
| 680 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 681 | def _RunInstrumentationTests(args, devices): |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 682 | """Subcommand of RunTestsCommands which runs instrumentation tests.""" |
jbudorick | 58b4d36 | 2015-09-08 16:44:59 | [diff] [blame] | 683 | logging.info('_RunInstrumentationTests(%s, %s)', str(args), str(devices)) |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 684 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 685 | instrumentation_options = ProcessInstrumentationOptions(args) |
| 686 | |
| 687 | if len(devices) > 1 and args.wait_for_debugger: |
[email protected] | f7148dd4 | 2013-08-20 14:24:57 | [diff] [blame] | 688 | logging.warning('Debugger can not be sharded, using first available device') |
| 689 | devices = devices[:1] |
| 690 | |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 691 | results = base_test_result.TestRunResults() |
| 692 | exit_code = 0 |
| 693 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 694 | if args.run_java_tests: |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 695 | java_runner_factory, java_tests = instrumentation_setup.Setup( |
mikecase | 526d68e | 2014-11-19 20:02:05 | [diff] [blame] | 696 | instrumentation_options, devices) |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 697 | else: |
| 698 | java_runner_factory = None |
| 699 | java_tests = None |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 700 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 701 | if args.run_python_tests: |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 702 | py_runner_factory, py_tests = host_driven_setup.InstrumentationSetup( |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 703 | args.host_driven_root, args.official_build, |
[email protected] | 37ee0c79 | 2013-08-06 19:10:13 | [diff] [blame] | 704 | instrumentation_options) |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 705 | else: |
| 706 | py_runner_factory = None |
| 707 | py_tests = None |
[email protected] | 37ee0c79 | 2013-08-06 19:10:13 | [diff] [blame] | 708 | |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 709 | results = [] |
| 710 | repetitions = (xrange(args.repeat + 1) if args.repeat >= 0 |
| 711 | else itertools.count()) |
| 712 | for _ in repetitions: |
| 713 | iteration_results = base_test_result.TestRunResults() |
| 714 | if java_tests: |
[email protected] | 3402002 | 2013-08-06 23:35:34 | [diff] [blame] | 715 | test_results, test_exit_code = test_dispatcher.RunTests( |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 716 | java_tests, java_runner_factory, devices, shard=True, |
| 717 | test_timeout=None, num_retries=args.num_retries) |
| 718 | iteration_results.AddTestRunResults(test_results) |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 719 | |
[email protected] | 3402002 | 2013-08-06 23:35:34 | [diff] [blame] | 720 | # Only allow exit code escalation |
| 721 | if test_exit_code and exit_code != constants.ERROR_EXIT_CODE: |
| 722 | exit_code = test_exit_code |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 723 | |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 724 | if py_tests: |
| 725 | test_results, test_exit_code = test_dispatcher.RunTests( |
| 726 | py_tests, py_runner_factory, devices, shard=True, test_timeout=None, |
| 727 | num_retries=args.num_retries) |
| 728 | iteration_results.AddTestRunResults(test_results) |
[email protected] | 4f777ca | 2014-08-08 01:45:59 | [diff] [blame] | 729 | |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 730 | # Only allow exit code escalation |
| 731 | if test_exit_code and exit_code != constants.ERROR_EXIT_CODE: |
| 732 | exit_code = test_exit_code |
| 733 | |
| 734 | results.append(iteration_results) |
| 735 | report_results.LogFull( |
| 736 | results=iteration_results, |
| 737 | test_type='Instrumentation', |
| 738 | test_package=os.path.basename(args.test_apk), |
| 739 | annotation=args.annotations, |
| 740 | flakiness_server=args.flakiness_dashboard_server) |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 741 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 742 | if args.json_results_file: |
| 743 | json_results.GenerateJsonResultsFile(results, args.json_results_file) |
jbudorick | b8c4207 | 2014-12-01 18:07:54 | [diff] [blame] | 744 | |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 745 | return exit_code |
| 746 | |
| 747 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 748 | def _RunUIAutomatorTests(args, devices): |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 749 | """Subcommand of RunTestsCommands which runs uiautomator tests.""" |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 750 | uiautomator_options = ProcessUIAutomatorOptions(args) |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 751 | |
jbudorick | dde688fb | 2015-08-27 03:00:17 | [diff] [blame] | 752 | runner_factory, tests = uiautomator_setup.Setup(uiautomator_options, devices) |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 753 | |
[email protected] | 37ee0c79 | 2013-08-06 19:10:13 | [diff] [blame] | 754 | results, exit_code = test_dispatcher.RunTests( |
[email protected] | f7148dd4 | 2013-08-20 14:24:57 | [diff] [blame] | 755 | tests, runner_factory, devices, shard=True, test_timeout=None, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 756 | num_retries=args.num_retries) |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 757 | |
| 758 | report_results.LogFull( |
| 759 | results=results, |
| 760 | test_type='UIAutomator', |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 761 | test_package=os.path.basename(args.test_jar), |
| 762 | annotation=args.annotations, |
| 763 | flakiness_server=args.flakiness_dashboard_server) |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 764 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 765 | if args.json_results_file: |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 766 | json_results.GenerateJsonResultsFile([results], args.json_results_file) |
jbudorick | b8c4207 | 2014-12-01 18:07:54 | [diff] [blame] | 767 | |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 768 | return exit_code |
| 769 | |
| 770 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 771 | def _RunJUnitTests(args): |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 772 | """Subcommand of RunTestsCommand which runs junit tests.""" |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 773 | runner_factory, tests = junit_setup.Setup(args) |
mikecase | c638a07 | 2015-04-01 16:35:35 | [diff] [blame] | 774 | results, exit_code = junit_dispatcher.RunTests(tests, runner_factory) |
| 775 | |
| 776 | report_results.LogFull( |
| 777 | results=results, |
| 778 | test_type='JUnit', |
| 779 | test_package=args.test_suite) |
| 780 | |
mikecase | 572401b | 2015-04-09 02:28:57 | [diff] [blame] | 781 | if args.json_results_file: |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 782 | json_results.GenerateJsonResultsFile([results], args.json_results_file) |
mikecase | 572401b | 2015-04-09 02:28:57 | [diff] [blame] | 783 | |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 784 | return exit_code |
| 785 | |
| 786 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 787 | def _RunMonkeyTests(args, devices): |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 788 | """Subcommand of RunTestsCommands which runs monkey tests.""" |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 789 | monkey_options = ProcessMonkeyTestOptions(args) |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 790 | |
| 791 | runner_factory, tests = monkey_setup.Setup(monkey_options) |
| 792 | |
| 793 | results, exit_code = test_dispatcher.RunTests( |
[email protected] | 181a5c9 | 2013-09-06 17:11:46 | [diff] [blame] | 794 | tests, runner_factory, devices, shard=False, test_timeout=None, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 795 | num_retries=args.num_retries) |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 796 | |
| 797 | report_results.LogFull( |
| 798 | results=results, |
| 799 | test_type='Monkey', |
[email protected] | 14b3b120 | 2013-08-15 22:25:28 | [diff] [blame] | 800 | test_package='Monkey') |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 801 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 802 | if args.json_results_file: |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 803 | json_results.GenerateJsonResultsFile([results], args.json_results_file) |
jbudorick | b8c4207 | 2014-12-01 18:07:54 | [diff] [blame] | 804 | |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 805 | return exit_code |
| 806 | |
| 807 | |
jbudorick | dde688fb | 2015-08-27 03:00:17 | [diff] [blame] | 808 | def _RunPerfTests(args, active_devices): |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 809 | """Subcommand of RunTestsCommands which runs perf tests.""" |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 810 | perf_options = ProcessPerfTestOptions(args) |
[email protected] | 61487ed | 2014-06-09 12:33:56 | [diff] [blame] | 811 | |
| 812 | # Just save a simple json with a list of test names. |
| 813 | if perf_options.output_json_list: |
| 814 | return perf_test_runner.OutputJsonList( |
| 815 | perf_options.steps, perf_options.output_json_list) |
| 816 | |
[email protected] | ad32f31 | 2013-11-13 04:03:29 | [diff] [blame] | 817 | # Just print the results from a single previously executed step. |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 818 | if perf_options.print_step: |
simonhatch | 9b9256d | 2015-01-07 18:03:42 | [diff] [blame] | 819 | return perf_test_runner.PrintTestOutput( |
perezju | 67cf7f1 | 2015-09-29 11:39:05 | [diff] [blame] | 820 | perf_options.print_step, perf_options.output_chartjson_data, |
| 821 | perf_options.get_output_dir_archive) |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 822 | |
jbudorick | dde688fb | 2015-08-27 03:00:17 | [diff] [blame] | 823 | runner_factory, tests, devices = perf_setup.Setup( |
| 824 | perf_options, active_devices) |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 825 | |
[email protected] | a72f075 | 2014-06-03 23:52:34 | [diff] [blame] | 826 | # shard=False means that each device will get the full list of tests |
| 827 | # and then each one will decide their own affinity. |
| 828 | # shard=True means each device will pop the next test available from a queue, |
| 829 | # which increases throughput but have no affinity. |
[email protected] | 86184c7b | 2013-08-15 15:06:57 | [diff] [blame] | 830 | results, _ = test_dispatcher.RunTests( |
[email protected] | a72f075 | 2014-06-03 23:52:34 | [diff] [blame] | 831 | tests, runner_factory, devices, shard=False, test_timeout=None, |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 832 | num_retries=args.num_retries) |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 833 | |
| 834 | report_results.LogFull( |
| 835 | results=results, |
| 836 | test_type='Perf', |
[email protected] | 865a47a | 2013-08-16 14:01:12 | [diff] [blame] | 837 | test_package='Perf') |
[email protected] | def4bce | 2013-11-12 12:59:52 | [diff] [blame] | 838 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 839 | if args.json_results_file: |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 840 | json_results.GenerateJsonResultsFile([results], args.json_results_file) |
jbudorick | b8c4207 | 2014-12-01 18:07:54 | [diff] [blame] | 841 | |
[email protected] | def4bce | 2013-11-12 12:59:52 | [diff] [blame] | 842 | if perf_options.single_step: |
| 843 | return perf_test_runner.PrintTestOutput('single_step') |
| 844 | |
[email protected] | 11ce845 | 2014-02-17 10:55:03 | [diff] [blame] | 845 | perf_test_runner.PrintSummary(tests) |
| 846 | |
[email protected] | 86184c7b | 2013-08-15 15:06:57 | [diff] [blame] | 847 | # Always return 0 on the sharding stage. Individual tests exit_code |
| 848 | # will be returned on the print_step stage. |
| 849 | return 0 |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 850 | |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 851 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 852 | def _RunPythonTests(args): |
jbudorick | 256fd53 | 2014-10-24 01:50:13 | [diff] [blame] | 853 | """Subcommand of RunTestsCommand which runs python unit tests.""" |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 854 | suite_vars = constants.PYTHON_UNIT_TEST_SUITES[args.suite_name] |
jbudorick | 256fd53 | 2014-10-24 01:50:13 | [diff] [blame] | 855 | suite_path = suite_vars['path'] |
| 856 | suite_test_modules = suite_vars['test_modules'] |
| 857 | |
| 858 | sys.path = [suite_path] + sys.path |
| 859 | try: |
| 860 | suite = unittest.TestSuite() |
| 861 | suite.addTests(unittest.defaultTestLoader.loadTestsFromName(m) |
| 862 | for m in suite_test_modules) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 863 | runner = unittest.TextTestRunner(verbosity=1+args.verbose_count) |
jbudorick | 256fd53 | 2014-10-24 01:50:13 | [diff] [blame] | 864 | return 0 if runner.run(suite).wasSuccessful() else 1 |
| 865 | finally: |
| 866 | sys.path = sys.path[1:] |
| 867 | |
| 868 | |
jbudorick | dde688fb | 2015-08-27 03:00:17 | [diff] [blame] | 869 | def _GetAttachedDevices(blacklist_file, test_device): |
[email protected] | f7148dd4 | 2013-08-20 14:24:57 | [diff] [blame] | 870 | """Get all attached devices. |
| 871 | |
| 872 | Args: |
| 873 | test_device: Name of a specific device to use. |
| 874 | |
| 875 | Returns: |
| 876 | A list of attached devices. |
| 877 | """ |
jbudorick | a583ba3 | 2015-09-11 17:23:19 | [diff] [blame] | 878 | blacklist = (device_blacklist.Blacklist(blacklist_file) |
| 879 | if blacklist_file |
| 880 | else None) |
jbudorick | dde688fb | 2015-08-27 03:00:17 | [diff] [blame] | 881 | |
jbudorick | dde688fb | 2015-08-27 03:00:17 | [diff] [blame] | 882 | attached_devices = device_utils.DeviceUtils.HealthyDevices(blacklist) |
aberent | 6a02a618 | 2015-04-29 11:07:55 | [diff] [blame] | 883 | if test_device: |
jbudorick | 4551d0dc | 2015-04-29 16:07:06 | [diff] [blame] | 884 | test_device = [d for d in attached_devices if d == test_device] |
| 885 | if not test_device: |
| 886 | raise device_errors.DeviceUnreachableError( |
| 887 | 'Did not find device %s among attached device. Attached devices: %s' |
| 888 | % (test_device, ', '.join(attached_devices))) |
| 889 | return test_device |
aberent | 6a02a618 | 2015-04-29 11:07:55 | [diff] [blame] | 890 | |
jbudorick | 4551d0dc | 2015-04-29 16:07:06 | [diff] [blame] | 891 | else: |
| 892 | if not attached_devices: |
| 893 | raise device_errors.NoDevicesError() |
| 894 | return sorted(attached_devices) |
[email protected] | f7148dd4 | 2013-08-20 14:24:57 | [diff] [blame] | 895 | |
| 896 | |
jbudorick | 58b4d36 | 2015-09-08 16:44:59 | [diff] [blame] | 897 | def RunTestsCommand(args, parser): # pylint: disable=too-many-return-statements |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 898 | """Checks test type and dispatches to the appropriate function. |
| 899 | |
| 900 | Args: |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 901 | args: argparse.Namespace object. |
| 902 | parser: argparse.ArgumentParser object. |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 903 | |
| 904 | Returns: |
| 905 | Integer indicated exit code. |
[email protected] | b387389 | 2013-07-10 04:57:10 | [diff] [blame] | 906 | |
| 907 | Raises: |
| 908 | Exception: Unknown command name passed in, or an exception from an |
| 909 | individual test runner. |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 910 | """ |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 911 | command = args.command |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 912 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 913 | ProcessCommonOptions(args) |
[email protected] | d82f025 | 2013-07-12 23:22:57 | [diff] [blame] | 914 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 915 | if args.enable_platform_mode: |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 916 | return RunTestsInPlatformMode(args, parser) |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 917 | |
| 918 | if command in constants.LOCAL_MACHINE_TESTS: |
jbudorick | 256fd53 | 2014-10-24 01:50:13 | [diff] [blame] | 919 | devices = [] |
| 920 | else: |
jbudorick | dde688fb | 2015-08-27 03:00:17 | [diff] [blame] | 921 | devices = _GetAttachedDevices(args.blacklist_file, args.test_device) |
[email protected] | f7148dd4 | 2013-08-20 14:24:57 | [diff] [blame] | 922 | |
[email protected] | c0662e09 | 2013-11-12 11:51:25 | [diff] [blame] | 923 | forwarder.Forwarder.RemoveHostLog() |
[email protected] | 6b11583b | 2013-11-21 16:18:40 | [diff] [blame] | 924 | if not ports.ResetTestServerPortAllocation(): |
| 925 | raise Exception('Failed to reset test server port.') |
[email protected] | c0662e09 | 2013-11-12 11:51:25 | [diff] [blame] | 926 | |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 927 | if command == 'gtest': |
jbudorick | 566592ab | 2015-09-21 15:32:47 | [diff] [blame] | 928 | return RunTestsInPlatformMode(args, parser) |
[email protected] | 6b6abac6d | 2013-10-03 11:56:38 | [diff] [blame] | 929 | elif command == 'linker': |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 930 | return _RunLinkerTests(args, devices) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 931 | elif command == 'instrumentation': |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 932 | return _RunInstrumentationTests(args, devices) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 933 | elif command == 'uiautomator': |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 934 | return _RunUIAutomatorTests(args, devices) |
jbudorick | 9a6b7b33 | 2014-09-20 00:01:07 | [diff] [blame] | 935 | elif command == 'junit': |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 936 | return _RunJUnitTests(args) |
[email protected] | 3dbdfa4 | 2013-08-08 01:08:14 | [diff] [blame] | 937 | elif command == 'monkey': |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 938 | return _RunMonkeyTests(args, devices) |
[email protected] | ec3170b | 2013-08-14 14:39:47 | [diff] [blame] | 939 | elif command == 'perf': |
jbudorick | dde688fb | 2015-08-27 03:00:17 | [diff] [blame] | 940 | return _RunPerfTests(args, devices) |
jbudorick | 256fd53 | 2014-10-24 01:50:13 | [diff] [blame] | 941 | elif command == 'python': |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 942 | return _RunPythonTests(args) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 943 | else: |
[email protected] | 6bc1bda2 | 2013-07-19 22:08:37 | [diff] [blame] | 944 | raise Exception('Unknown test type.') |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 945 | |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 946 | |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 947 | _SUPPORTED_IN_PLATFORM_MODE = [ |
| 948 | # TODO(jbudorick): Add support for more test types. |
jbudorick | 911be58d | 2015-01-13 02:51:06 | [diff] [blame] | 949 | 'gtest', |
| 950 | 'instrumentation', |
| 951 | 'uirobot', |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 952 | ] |
| 953 | |
| 954 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 955 | def RunTestsInPlatformMode(args, parser): |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 956 | |
jbudorick | 566592ab | 2015-09-21 15:32:47 | [diff] [blame] | 957 | def infra_error(message): |
| 958 | parser.exit(status=constants.INFRA_EXIT_CODE, message=message) |
jbudorick | b9b0ada | 2015-09-17 22:52:58 | [diff] [blame] | 959 | |
jbudorick | 566592ab | 2015-09-21 15:32:47 | [diff] [blame] | 960 | if args.command not in _SUPPORTED_IN_PLATFORM_MODE: |
| 961 | infra_error('%s is not yet supported in platform mode' % args.command) |
| 962 | |
| 963 | with environment_factory.CreateEnvironment(args, infra_error) as env: |
| 964 | with test_instance_factory.CreateTestInstance(args, infra_error) as test: |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 965 | with test_run_factory.CreateTestRun( |
jbudorick | 566592ab | 2015-09-21 15:32:47 | [diff] [blame] | 966 | args, env, test, infra_error) as test_run: |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 967 | results = [] |
| 968 | repetitions = (xrange(args.repeat + 1) if args.repeat >= 0 |
| 969 | else itertools.count()) |
| 970 | for _ in repetitions: |
| 971 | iteration_results = test_run.RunTests() |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 972 | |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 973 | if iteration_results is not None: |
jbudorick | d4f7798 | 2015-09-28 21:09:18 | [diff] [blame] | 974 | results.append(iteration_results) |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 975 | report_results.LogFull( |
| 976 | results=iteration_results, |
| 977 | test_type=test.TestType(), |
| 978 | test_package=test_run.TestPackage(), |
| 979 | annotation=getattr(args, 'annotations', None), |
| 980 | flakiness_server=getattr(args, 'flakiness_dashboard_server', |
| 981 | None)) |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 982 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 983 | if args.json_results_file: |
jbudorick | b8c4207 | 2014-12-01 18:07:54 | [diff] [blame] | 984 | json_results.GenerateJsonResultsFile( |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 985 | results, args.json_results_file) |
jbudorick | b8c4207 | 2014-12-01 18:07:54 | [diff] [blame] | 986 | |
jbudorick | eb7ea71c | 2015-09-28 16:40:20 | [diff] [blame] | 987 | return (0 if all(r.DidRunPass() for r in results) |
| 988 | else constants.ERROR_EXIT_CODE) |
jbudorick | 66dc372 | 2014-11-06 21:33:51 | [diff] [blame] | 989 | |
| 990 | |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 991 | CommandConfigTuple = collections.namedtuple( |
| 992 | 'CommandConfigTuple', |
| 993 | ['add_options_func', 'help_txt']) |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 994 | VALID_COMMANDS = { |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 995 | 'gtest': CommandConfigTuple( |
| 996 | AddGTestOptions, |
| 997 | 'googletest-based C++ tests'), |
| 998 | 'instrumentation': CommandConfigTuple( |
| 999 | AddInstrumentationTestOptions, |
| 1000 | 'InstrumentationTestCase-based Java tests'), |
| 1001 | 'uiautomator': CommandConfigTuple( |
| 1002 | AddUIAutomatorTestOptions, |
| 1003 | "Tests that run via Android's uiautomator command"), |
| 1004 | 'junit': CommandConfigTuple( |
| 1005 | AddJUnitTestOptions, |
| 1006 | 'JUnit4-based Java tests'), |
| 1007 | 'monkey': CommandConfigTuple( |
| 1008 | AddMonkeyTestOptions, |
| 1009 | "Tests based on Android's monkey"), |
| 1010 | 'perf': CommandConfigTuple( |
| 1011 | AddPerfTestOptions, |
| 1012 | 'Performance tests'), |
| 1013 | 'python': CommandConfigTuple( |
| 1014 | AddPythonTestOptions, |
| 1015 | 'Python tests based on unittest.TestCase'), |
| 1016 | 'linker': CommandConfigTuple( |
| 1017 | AddLinkerTestOptions, |
| 1018 | 'Linker tests'), |
rnephew | 5c49978 | 2014-12-12 19:08:55 | [diff] [blame] | 1019 | 'uirobot': CommandConfigTuple( |
| 1020 | AddUirobotTestOptions, |
| 1021 | 'Uirobot test'), |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 1022 | } |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 1023 | |
| 1024 | |
[email protected] | 7c53a60 | 2014-03-24 16:21:44 | [diff] [blame] | 1025 | def DumpThreadStacks(_signal, _frame): |
[email protected] | 71aec4b | 2013-11-20 00:35:24 | [diff] [blame] | 1026 | for thread in threading.enumerate(): |
| 1027 | reraiser_thread.LogThreadStack(thread) |
[email protected] | 83bb815 | 2013-11-19 15:02:21 | [diff] [blame] | 1028 | |
| 1029 | |
[email protected] | 7c53a60 | 2014-03-24 16:21:44 | [diff] [blame] | 1030 | def main(): |
[email protected] | 83bb815 | 2013-11-19 15:02:21 | [diff] [blame] | 1031 | signal.signal(signal.SIGUSR1, DumpThreadStacks) |
jam | a47ca85c | 2014-12-03 18:38:07 | [diff] [blame] | 1032 | |
| 1033 | parser = argparse.ArgumentParser() |
| 1034 | command_parsers = parser.add_subparsers(title='test types', |
| 1035 | dest='command') |
| 1036 | |
| 1037 | for test_type, config in sorted(VALID_COMMANDS.iteritems(), |
| 1038 | key=lambda x: x[0]): |
| 1039 | subparser = command_parsers.add_parser( |
| 1040 | test_type, usage='%(prog)s [options]', help=config.help_txt) |
| 1041 | config.add_options_func(subparser) |
| 1042 | |
| 1043 | args = parser.parse_args() |
mikecase | e7405102 | 2015-02-26 23:08:22 | [diff] [blame] | 1044 | |
| 1045 | try: |
| 1046 | return RunTestsCommand(args, parser) |
| 1047 | except base_error.BaseError as e: |
| 1048 | logging.exception('Error occurred.') |
| 1049 | if e.is_infra_error: |
| 1050 | return constants.INFRA_EXIT_CODE |
msw | ecce673 | 2015-06-06 00:31:33 | [diff] [blame] | 1051 | return constants.ERROR_EXIT_CODE |
mikecase | e7405102 | 2015-02-26 23:08:22 | [diff] [blame] | 1052 | except: # pylint: disable=W0702 |
| 1053 | logging.exception('Unrecognized error occurred.') |
| 1054 | return constants.ERROR_EXIT_CODE |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 1055 | |
[email protected] | fbe2932 | 2013-07-09 09:03:26 | [diff] [blame] | 1056 | |
| 1057 | if __name__ == '__main__': |
[email protected] | 7c53a60 | 2014-03-24 16:21:44 | [diff] [blame] | 1058 | sys.exit(main()) |