blob: 2221e45b704c8d82bf0e7dce61e2550708a9efc1 [file] [log] [blame]
[email protected]fbe29322013-07-09 09:03:261#!/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]181a5c92013-09-06 17:11:467"""Runs all types of tests from one unified interface."""
[email protected]fbe29322013-07-09 09:03:268
9import collections
[email protected]f7148dd42013-08-20 14:24:5710import logging
[email protected]fbe29322013-07-09 09:03:2611import optparse
12import os
[email protected]6bc1bda22013-07-19 22:08:3713import shutil
[email protected]83bb8152013-11-19 15:02:2114import signal
[email protected]fbe29322013-07-09 09:03:2615import sys
[email protected]83bb8152013-11-19 15:02:2116import threading
[email protected]e9ee6ec32014-02-11 01:47:4117import traceback
[email protected]fbe29322013-07-09 09:03:2618
[email protected]f7148dd42013-08-20 14:24:5719from pylib import android_commands
[email protected]fbe29322013-07-09 09:03:2620from pylib import constants
[email protected]c0662e092013-11-12 11:51:2521from pylib import forwarder
[email protected]fbe29322013-07-09 09:03:2622from pylib import ports
23from pylib.base import base_test_result
[email protected]6bc1bda22013-07-19 22:08:3724from pylib.base import test_dispatcher
[email protected]6bc1bda22013-07-19 22:08:3725from pylib.gtest import gtest_config
[email protected]2a684222013-08-01 16:59:2226from pylib.gtest import setup as gtest_setup
27from pylib.gtest import test_options as gtest_test_options
[email protected]6b6abac6d2013-10-03 11:56:3828from pylib.linker import setup as linker_setup
[email protected]37ee0c792013-08-06 19:10:1329from pylib.host_driven import setup as host_driven_setup
[email protected]6bc1bda22013-07-19 22:08:3730from pylib.instrumentation import setup as instrumentation_setup
[email protected]2a684222013-08-01 16:59:2231from pylib.instrumentation import test_options as instrumentation_test_options
[email protected]3dbdfa42013-08-08 01:08:1432from pylib.monkey import setup as monkey_setup
33from pylib.monkey import test_options as monkey_test_options
[email protected]ec3170b2013-08-14 14:39:4734from pylib.perf import setup as perf_setup
35from pylib.perf import test_options as perf_test_options
36from pylib.perf import test_runner as perf_test_runner
[email protected]6bc1bda22013-07-19 22:08:3737from pylib.uiautomator import setup as uiautomator_setup
[email protected]2a684222013-08-01 16:59:2238from pylib.uiautomator import test_options as uiautomator_test_options
[email protected]803f65a72013-08-20 19:11:3039from pylib.utils import command_option_parser
[email protected]6bc1bda22013-07-19 22:08:3740from pylib.utils import report_results
[email protected]71aec4b2013-11-20 00:35:2441from pylib.utils import reraiser_thread
[email protected]6bc1bda22013-07-19 22:08:3742from pylib.utils import run_tests_helper
[email protected]fbe29322013-07-09 09:03:2643
44
[email protected]fbe29322013-07-09 09:03:2645def AddCommonOptions(option_parser):
46 """Adds all common options to |option_parser|."""
47
[email protected]dfffbcbc2013-09-17 22:06:0148 group = optparse.OptionGroup(option_parser, 'Common Options')
49 default_build_type = os.environ.get('BUILDTYPE', 'Debug')
50 group.add_option('--debug', action='store_const', const='Debug',
51 dest='build_type', default=default_build_type,
52 help=('If set, run test suites under out/Debug. '
53 'Default is env var BUILDTYPE or Debug.'))
54 group.add_option('--release', action='store_const',
55 const='Release', dest='build_type',
56 help=('If set, run test suites under out/Release.'
57 ' Default is env var BUILDTYPE or Debug.'))
58 group.add_option('-c', dest='cleanup_test_files',
59 help='Cleanup test files on the device after run',
60 action='store_true')
61 group.add_option('--num_retries', dest='num_retries', type='int',
62 default=2,
63 help=('Number of retries for a test before '
64 'giving up.'))
65 group.add_option('-v',
66 '--verbose',
67 dest='verbose_count',
68 default=0,
69 action='count',
70 help='Verbose level (multiple times for more)')
71 group.add_option('--tool',
72 dest='tool',
73 help=('Run the test under a tool '
74 '(use --tool help to list them)'))
75 group.add_option('--flakiness-dashboard-server',
76 dest='flakiness_dashboard_server',
77 help=('Address of the server that is hosting the '
78 'Chrome for Android flakiness dashboard.'))
79 group.add_option('--skip-deps-push', dest='push_deps',
80 action='store_false', default=True,
81 help=('Do not push dependencies to the device. '
82 'Use this at own risk for speeding up test '
83 'execution on local machine.'))
84 group.add_option('-d', '--device', dest='test_device',
85 help=('Target device for the test suite '
86 'to run on.'))
87 option_parser.add_option_group(group)
[email protected]fbe29322013-07-09 09:03:2688
89
90def ProcessCommonOptions(options):
91 """Processes and handles all common options."""
[email protected]fbe29322013-07-09 09:03:2692 run_tests_helper.SetLogLevel(options.verbose_count)
[email protected]14b3b1202013-08-15 22:25:2893 constants.SetBuildType(options.build_type)
[email protected]fbe29322013-07-09 09:03:2694
95
[email protected]fbe29322013-07-09 09:03:2696def AddGTestOptions(option_parser):
97 """Adds gtest options to |option_parser|."""
98
99 option_parser.usage = '%prog gtest [options]'
[email protected]dfffbcbc2013-09-17 22:06:01100 option_parser.commands_dict = {}
[email protected]fbe29322013-07-09 09:03:26101 option_parser.example = '%prog gtest -s base_unittests'
102
[email protected]6bc1bda22013-07-19 22:08:37103 # TODO(gkanwar): Make this option required
104 option_parser.add_option('-s', '--suite', dest='suite_name',
[email protected]fbe29322013-07-09 09:03:26105 help=('Executable name of the test suite to run '
106 '(use -s help to list them).'))
[email protected]c53dc4332013-11-20 04:38:03107 option_parser.add_option('-f', '--gtest_filter', '--gtest-filter',
108 dest='test_filter',
[email protected]9e689252013-07-30 20:14:36109 help='googletest-style filter string.')
[email protected]c53dc4332013-11-20 04:38:03110 option_parser.add_option('--gtest_also_run_disabled_tests',
111 '--gtest-also-run-disabled-tests',
[email protected]dfffbcbc2013-09-17 22:06:01112 dest='run_disabled', action='store_true',
113 help='Also run disabled tests if applicable.')
114 option_parser.add_option('-a', '--test-arguments', dest='test_arguments',
115 default='',
[email protected]9e689252013-07-30 20:14:36116 help='Additional arguments to pass to the test.')
117 option_parser.add_option('-t', dest='timeout',
118 help='Timeout to wait for each test',
119 type='int',
120 default=60)
[email protected]fbe29322013-07-09 09:03:26121 # TODO(gkanwar): Move these to Common Options once we have the plumbing
122 # in our other test types to handle these commands
[email protected]fbe29322013-07-09 09:03:26123 AddCommonOptions(option_parser)
124
125
[email protected]6b6abac6d2013-10-03 11:56:38126def AddLinkerTestOptions(option_parser):
127 option_parser.usage = '%prog linker'
128 option_parser.commands_dict = {}
129 option_parser.example = '%prog linker'
130
[email protected]98c4feef2013-10-08 01:19:05131 option_parser.add_option('-f', '--gtest-filter', dest='test_filter',
132 help='googletest-style filter string.')
[email protected]6b6abac6d2013-10-03 11:56:38133 AddCommonOptions(option_parser)
134
135
[email protected]6bc1bda22013-07-19 22:08:37136def ProcessGTestOptions(options):
137 """Intercept test suite help to list test suites.
138
139 Args:
140 options: Command line options.
[email protected]6bc1bda22013-07-19 22:08:37141 """
142 if options.suite_name == 'help':
143 print 'Available test suites are:'
[email protected]9e689252013-07-30 20:14:36144 for test_suite in (gtest_config.STABLE_TEST_SUITES +
145 gtest_config.EXPERIMENTAL_TEST_SUITES):
146 print test_suite
[email protected]2a684222013-08-01 16:59:22147 sys.exit(0)
[email protected]6bc1bda22013-07-19 22:08:37148
149 # Convert to a list, assuming all test suites if nothing was specified.
150 # TODO(gkanwar): Require having a test suite
151 if options.suite_name:
152 options.suite_name = [options.suite_name]
153 else:
[email protected]9e689252013-07-30 20:14:36154 options.suite_name = [s for s in gtest_config.STABLE_TEST_SUITES]
[email protected]6bc1bda22013-07-19 22:08:37155
156
[email protected]fbe29322013-07-09 09:03:26157def AddJavaTestOptions(option_parser):
158 """Adds the Java test options to |option_parser|."""
159
[email protected]dfffbcbc2013-09-17 22:06:01160 option_parser.add_option('-f', '--test-filter', dest='test_filter',
[email protected]fbe29322013-07-09 09:03:26161 help=('Test filter (if not fully qualified, '
162 'will run all matches).'))
163 option_parser.add_option(
164 '-A', '--annotation', dest='annotation_str',
165 help=('Comma-separated list of annotations. Run only tests with any of '
166 'the given annotations. An annotation can be either a key or a '
167 'key-values pair. A test that has no annotation is considered '
168 '"SmallTest".'))
169 option_parser.add_option(
170 '-E', '--exclude-annotation', dest='exclude_annotation_str',
171 help=('Comma-separated list of annotations. Exclude tests with these '
172 'annotations.'))
[email protected]fbe29322013-07-09 09:03:26173 option_parser.add_option('--screenshot', dest='screenshot_failures',
174 action='store_true',
175 help='Capture screenshots of test failures')
176 option_parser.add_option('--save-perf-json', action='store_true',
177 help='Saves the JSON file for each UI Perf test.')
[email protected]37ee0c792013-08-06 19:10:13178 option_parser.add_option('--official-build', action='store_true',
179 help='Run official build tests.')
[email protected]fbe29322013-07-09 09:03:26180 option_parser.add_option('--test_data', action='append', default=[],
181 help=('Each instance defines a directory of test '
182 'data that should be copied to the target(s) '
183 'before running the tests. The argument '
184 'should be of the form <target>:<source>, '
185 '<target> is relative to the device data'
186 'directory, and <source> is relative to the '
187 'chromium build directory.'))
188
189
[email protected]e9ee6ec32014-02-11 01:47:41190def ProcessJavaTestOptions(options, error_func):
[email protected]fbe29322013-07-09 09:03:26191 """Processes options/arguments and populates |options| with defaults."""
192
[email protected]fbe29322013-07-09 09:03:26193 if options.annotation_str:
194 options.annotations = options.annotation_str.split(',')
195 elif options.test_filter:
196 options.annotations = []
197 else:
[email protected]6bc1bda22013-07-19 22:08:37198 options.annotations = ['Smoke', 'SmallTest', 'MediumTest', 'LargeTest',
199 'EnormousTest']
[email protected]fbe29322013-07-09 09:03:26200
201 if options.exclude_annotation_str:
202 options.exclude_annotations = options.exclude_annotation_str.split(',')
203 else:
204 options.exclude_annotations = []
205
[email protected]fbe29322013-07-09 09:03:26206
207def AddInstrumentationTestOptions(option_parser):
208 """Adds Instrumentation test options to |option_parser|."""
209
210 option_parser.usage = '%prog instrumentation [options]'
[email protected]dfffbcbc2013-09-17 22:06:01211 option_parser.commands_dict = {}
[email protected]fb7ab5e82013-07-26 18:31:20212 option_parser.example = ('%prog instrumentation '
[email protected]fbe29322013-07-09 09:03:26213 '--test-apk=ChromiumTestShellTest')
214
215 AddJavaTestOptions(option_parser)
216 AddCommonOptions(option_parser)
217
[email protected]dfffbcbc2013-09-17 22:06:01218 option_parser.add_option('-j', '--java-only', action='store_true',
[email protected]37ee0c792013-08-06 19:10:13219 default=False, help='Run only the Java tests.')
[email protected]dfffbcbc2013-09-17 22:06:01220 option_parser.add_option('-p', '--python-only', action='store_true',
[email protected]37ee0c792013-08-06 19:10:13221 default=False,
222 help='Run only the host-driven tests.')
[email protected]a69e85bc2013-08-16 18:07:26223 option_parser.add_option('--host-driven-root',
[email protected]37ee0c792013-08-06 19:10:13224 help='Root of the host-driven tests.')
[email protected]fbe29322013-07-09 09:03:26225 option_parser.add_option('-w', '--wait_debugger', dest='wait_for_debugger',
226 action='store_true',
227 help='Wait for debugger.')
[email protected]fbe29322013-07-09 09:03:26228 option_parser.add_option(
229 '--test-apk', dest='test_apk',
230 help=('The name of the apk containing the tests '
[email protected]ae68d4a2013-09-24 21:57:15231 '(without the .apk extension; e.g. "ContentShellTest").'))
[email protected]803f65a72013-08-20 19:11:30232 option_parser.add_option('--coverage-dir',
233 help=('Directory in which to place all generated '
234 'EMMA coverage files.'))
[email protected]fbe29322013-07-09 09:03:26235
236
237def ProcessInstrumentationOptions(options, error_func):
[email protected]2a684222013-08-01 16:59:22238 """Processes options/arguments and populate |options| with defaults.
239
240 Args:
241 options: optparse.Options object.
242 error_func: Function to call with the error message in case of an error.
243
244 Returns:
245 An InstrumentationOptions named tuple which contains all options relevant to
246 instrumentation tests.
247 """
[email protected]fbe29322013-07-09 09:03:26248
[email protected]e9ee6ec32014-02-11 01:47:41249 ProcessJavaTestOptions(options, error_func)
[email protected]fbe29322013-07-09 09:03:26250
[email protected]37ee0c792013-08-06 19:10:13251 if options.java_only and options.python_only:
252 error_func('Options java_only (-j) and python_only (-p) '
253 'are mutually exclusive.')
254 options.run_java_tests = True
255 options.run_python_tests = True
256 if options.java_only:
257 options.run_python_tests = False
258 elif options.python_only:
259 options.run_java_tests = False
260
[email protected]67954f822013-08-14 18:09:08261 if not options.host_driven_root:
[email protected]37ee0c792013-08-06 19:10:13262 options.run_python_tests = False
263
[email protected]fbe29322013-07-09 09:03:26264 if not options.test_apk:
265 error_func('--test-apk must be specified.')
266
[email protected]ae68d4a2013-09-24 21:57:15267
268 options.test_apk_path = os.path.join(constants.GetOutDirectory(),
269 constants.SDK_BUILD_APKS_DIR,
270 '%s.apk' % options.test_apk)
271 options.test_apk_jar_path = os.path.join(
272 constants.GetOutDirectory(),
273 constants.SDK_BUILD_TEST_JAVALIB_DIR,
274 '%s.jar' % options.test_apk)
[email protected]fbe29322013-07-09 09:03:26275
[email protected]2a684222013-08-01 16:59:22276 return instrumentation_test_options.InstrumentationOptions(
[email protected]2a684222013-08-01 16:59:22277 options.tool,
278 options.cleanup_test_files,
279 options.push_deps,
280 options.annotations,
281 options.exclude_annotations,
282 options.test_filter,
283 options.test_data,
284 options.save_perf_json,
285 options.screenshot_failures,
[email protected]2a684222013-08-01 16:59:22286 options.wait_for_debugger,
[email protected]803f65a72013-08-20 19:11:30287 options.coverage_dir,
[email protected]2a684222013-08-01 16:59:22288 options.test_apk,
289 options.test_apk_path,
290 options.test_apk_jar_path)
291
[email protected]fbe29322013-07-09 09:03:26292
293def AddUIAutomatorTestOptions(option_parser):
294 """Adds UI Automator test options to |option_parser|."""
295
296 option_parser.usage = '%prog uiautomator [options]'
[email protected]dfffbcbc2013-09-17 22:06:01297 option_parser.commands_dict = {}
[email protected]fbe29322013-07-09 09:03:26298 option_parser.example = (
299 '%prog uiautomator --test-jar=chromium_testshell_uiautomator_tests'
[email protected]a8886c8a92013-10-08 17:29:30300 ' --package=chromium_test_shell')
[email protected]fbe29322013-07-09 09:03:26301 option_parser.add_option(
[email protected]a8886c8a92013-10-08 17:29:30302 '--package',
303 help=('Package under test. Possible values: %s' %
304 constants.PACKAGE_INFO.keys()))
[email protected]fbe29322013-07-09 09:03:26305 option_parser.add_option(
306 '--test-jar', dest='test_jar',
307 help=('The name of the dexed jar containing the tests (without the '
308 '.dex.jar extension). Alternatively, this can be a full path '
309 'to the jar.'))
310
311 AddJavaTestOptions(option_parser)
312 AddCommonOptions(option_parser)
313
314
315def ProcessUIAutomatorOptions(options, error_func):
[email protected]2a684222013-08-01 16:59:22316 """Processes UIAutomator options/arguments.
317
318 Args:
319 options: optparse.Options object.
320 error_func: Function to call with the error message in case of an error.
321
322 Returns:
323 A UIAutomatorOptions named tuple which contains all options relevant to
[email protected]3dbdfa42013-08-08 01:08:14324 uiautomator tests.
[email protected]2a684222013-08-01 16:59:22325 """
[email protected]fbe29322013-07-09 09:03:26326
[email protected]e9ee6ec32014-02-11 01:47:41327 ProcessJavaTestOptions(options, error_func)
[email protected]fbe29322013-07-09 09:03:26328
[email protected]a8886c8a92013-10-08 17:29:30329 if not options.package:
330 error_func('--package is required.')
331
332 if options.package not in constants.PACKAGE_INFO:
333 error_func('Invalid package.')
[email protected]fbe29322013-07-09 09:03:26334
335 if not options.test_jar:
336 error_func('--test-jar must be specified.')
337
338 if os.path.exists(options.test_jar):
339 # The dexed JAR is fully qualified, assume the info JAR lives along side.
340 options.uiautomator_jar = options.test_jar
341 else:
342 options.uiautomator_jar = os.path.join(
[email protected]ae68d4a2013-09-24 21:57:15343 constants.GetOutDirectory(),
344 constants.SDK_BUILD_JAVALIB_DIR,
[email protected]fbe29322013-07-09 09:03:26345 '%s.dex.jar' % options.test_jar)
346 options.uiautomator_info_jar = (
347 options.uiautomator_jar[:options.uiautomator_jar.find('.dex.jar')] +
348 '_java.jar')
349
[email protected]2a684222013-08-01 16:59:22350 return uiautomator_test_options.UIAutomatorOptions(
[email protected]2a684222013-08-01 16:59:22351 options.tool,
352 options.cleanup_test_files,
353 options.push_deps,
354 options.annotations,
355 options.exclude_annotations,
356 options.test_filter,
357 options.test_data,
358 options.save_perf_json,
359 options.screenshot_failures,
[email protected]2a684222013-08-01 16:59:22360 options.uiautomator_jar,
361 options.uiautomator_info_jar,
[email protected]a8886c8a92013-10-08 17:29:30362 options.package)
[email protected]2a684222013-08-01 16:59:22363
[email protected]fbe29322013-07-09 09:03:26364
[email protected]3dbdfa42013-08-08 01:08:14365def AddMonkeyTestOptions(option_parser):
366 """Adds monkey test options to |option_parser|."""
[email protected]fb81b982013-08-09 00:07:12367
368 option_parser.usage = '%prog monkey [options]'
[email protected]dfffbcbc2013-09-17 22:06:01369 option_parser.commands_dict = {}
[email protected]fb81b982013-08-09 00:07:12370 option_parser.example = (
[email protected]a8886c8a92013-10-08 17:29:30371 '%prog monkey --package=chromium_test_shell')
[email protected]fb81b982013-08-09 00:07:12372
[email protected]3dbdfa42013-08-08 01:08:14373 option_parser.add_option(
[email protected]a8886c8a92013-10-08 17:29:30374 '--package',
375 help=('Package under test. Possible values: %s' %
376 constants.PACKAGE_INFO.keys()))
[email protected]3dbdfa42013-08-08 01:08:14377 option_parser.add_option(
378 '--event-count', default=10000, type='int',
379 help='Number of events to generate [default: %default].')
380 option_parser.add_option(
381 '--category', default='',
[email protected]fb81b982013-08-09 00:07:12382 help='A list of allowed categories.')
[email protected]3dbdfa42013-08-08 01:08:14383 option_parser.add_option(
384 '--throttle', default=100, type='int',
385 help='Delay between events (ms) [default: %default]. ')
386 option_parser.add_option(
387 '--seed', type='int',
388 help=('Seed value for pseudo-random generator. Same seed value generates '
389 'the same sequence of events. Seed is randomized by default.'))
390 option_parser.add_option(
391 '--extra-args', default='',
392 help=('String of other args to pass to the command verbatim '
393 '[default: "%default"].'))
394
395 AddCommonOptions(option_parser)
396
397
398def ProcessMonkeyTestOptions(options, error_func):
399 """Processes all monkey test options.
400
401 Args:
402 options: optparse.Options object.
403 error_func: Function to call with the error message in case of an error.
404
405 Returns:
406 A MonkeyOptions named tuple which contains all options relevant to
407 monkey tests.
408 """
[email protected]a8886c8a92013-10-08 17:29:30409 if not options.package:
410 error_func('--package is required.')
411
412 if options.package not in constants.PACKAGE_INFO:
413 error_func('Invalid package.')
[email protected]3dbdfa42013-08-08 01:08:14414
415 category = options.category
416 if category:
417 category = options.category.split(',')
418
419 return monkey_test_options.MonkeyOptions(
[email protected]3dbdfa42013-08-08 01:08:14420 options.verbose_count,
[email protected]a8886c8a92013-10-08 17:29:30421 options.package,
[email protected]3dbdfa42013-08-08 01:08:14422 options.event_count,
423 category,
424 options.throttle,
425 options.seed,
426 options.extra_args)
427
428
[email protected]ec3170b2013-08-14 14:39:47429def AddPerfTestOptions(option_parser):
430 """Adds perf test options to |option_parser|."""
431
432 option_parser.usage = '%prog perf [options]'
[email protected]dfffbcbc2013-09-17 22:06:01433 option_parser.commands_dict = {}
[email protected]def4bce2013-11-12 12:59:52434 option_parser.example = ('%prog perf '
[email protected]ad32f312013-11-13 04:03:29435 '[--single-step -- command args] or '
[email protected]def4bce2013-11-12 12:59:52436 '[--steps perf_steps.json] or '
[email protected]ad32f312013-11-13 04:03:29437 '[--print-step step]')
[email protected]ec3170b2013-08-14 14:39:47438
[email protected]181a5c92013-09-06 17:11:46439 option_parser.add_option(
[email protected]def4bce2013-11-12 12:59:52440 '--single-step',
[email protected]ad32f312013-11-13 04:03:29441 action='store_true',
[email protected]def4bce2013-11-12 12:59:52442 help='Execute the given command with retries, but only print the result '
443 'for the "most successful" round.')
444 option_parser.add_option(
[email protected]181a5c92013-09-06 17:11:46445 '--steps',
[email protected]def4bce2013-11-12 12:59:52446 help='JSON file containing the list of commands to run.')
[email protected]181a5c92013-09-06 17:11:46447 option_parser.add_option(
448 '--flaky-steps',
449 help=('A JSON file containing steps that are flaky '
450 'and will have its exit code ignored.'))
451 option_parser.add_option(
452 '--print-step',
453 help='The name of a previously executed perf step to print.')
454 option_parser.add_option(
455 '--no-timeout', action='store_true',
456 help=('Do not impose a timeout. Each perf step is responsible for '
457 'implementing the timeout logic.'))
[email protected]650487c2013-09-30 11:40:49458 option_parser.add_option(
459 '-f', '--test-filter',
460 help=('Test filter (will match against the names listed in --steps).'))
461 option_parser.add_option(
462 '--dry-run',
463 action='store_true',
464 help='Just print the steps without executing.')
[email protected]ec3170b2013-08-14 14:39:47465 AddCommonOptions(option_parser)
466
467
[email protected]ad32f312013-11-13 04:03:29468def ProcessPerfTestOptions(options, args, error_func):
[email protected]ec3170b2013-08-14 14:39:47469 """Processes all perf test options.
470
471 Args:
472 options: optparse.Options object.
473 error_func: Function to call with the error message in case of an error.
474
475 Returns:
476 A PerfOptions named tuple which contains all options relevant to
477 perf tests.
478 """
[email protected]def4bce2013-11-12 12:59:52479 # Only one of steps, print_step or single_step must be provided.
480 count = len(filter(None,
481 [options.steps, options.print_step, options.single_step]))
482 if count != 1:
483 error_func('Please specify one of: --steps, --print-step, --single-step.')
[email protected]ad32f312013-11-13 04:03:29484 single_step = None
485 if options.single_step:
486 single_step = ' '.join(args[2:])
[email protected]ec3170b2013-08-14 14:39:47487 return perf_test_options.PerfOptions(
[email protected]181a5c92013-09-06 17:11:46488 options.steps, options.flaky_steps, options.print_step,
[email protected]def4bce2013-11-12 12:59:52489 options.no_timeout, options.test_filter, options.dry_run,
[email protected]ad32f312013-11-13 04:03:29490 single_step)
[email protected]ec3170b2013-08-14 14:39:47491
492
[email protected]e9ee6ec32014-02-11 01:47:41493def _RunGTests(options, error_func, devices):
[email protected]6bc1bda22013-07-19 22:08:37494 """Subcommand of RunTestsCommands which runs gtests."""
[email protected]2a684222013-08-01 16:59:22495 ProcessGTestOptions(options)
[email protected]6bc1bda22013-07-19 22:08:37496
497 exit_code = 0
498 for suite_name in options.suite_name:
[email protected]2a684222013-08-01 16:59:22499 # TODO(gkanwar): Move this into ProcessGTestOptions once we require -s for
500 # the gtest command.
501 gtest_options = gtest_test_options.GTestOptions(
[email protected]2a684222013-08-01 16:59:22502 options.tool,
503 options.cleanup_test_files,
504 options.push_deps,
505 options.test_filter,
[email protected]dfffbcbc2013-09-17 22:06:01506 options.run_disabled,
[email protected]2a684222013-08-01 16:59:22507 options.test_arguments,
508 options.timeout,
509 suite_name)
[email protected]f7148dd42013-08-20 14:24:57510 runner_factory, tests = gtest_setup.Setup(gtest_options, devices)
[email protected]6bc1bda22013-07-19 22:08:37511
512 results, test_exit_code = test_dispatcher.RunTests(
[email protected]f7148dd42013-08-20 14:24:57513 tests, runner_factory, devices, shard=True, test_timeout=None,
[email protected]6bc1bda22013-07-19 22:08:37514 num_retries=options.num_retries)
515
516 if test_exit_code and exit_code != constants.ERROR_EXIT_CODE:
517 exit_code = test_exit_code
518
519 report_results.LogFull(
520 results=results,
521 test_type='Unit test',
522 test_package=suite_name,
[email protected]6bc1bda22013-07-19 22:08:37523 flakiness_server=options.flakiness_dashboard_server)
524
525 if os.path.isdir(constants.ISOLATE_DEPS_DIR):
526 shutil.rmtree(constants.ISOLATE_DEPS_DIR)
527
528 return exit_code
529
530
[email protected]e9ee6ec32014-02-11 01:47:41531def _RunLinkerTests(options, error_func, devices):
[email protected]6b6abac6d2013-10-03 11:56:38532 """Subcommand of RunTestsCommands which runs linker tests."""
533 runner_factory, tests = linker_setup.Setup(options, devices)
534
535 results, exit_code = test_dispatcher.RunTests(
536 tests, runner_factory, devices, shard=True, test_timeout=60,
537 num_retries=options.num_retries)
538
539 report_results.LogFull(
540 results=results,
541 test_type='Linker test',
[email protected]93c9f9b2014-02-10 16:19:22542 test_package='ChromiumLinkerTest')
[email protected]6b6abac6d2013-10-03 11:56:38543
544 return exit_code
545
546
[email protected]f7148dd42013-08-20 14:24:57547def _RunInstrumentationTests(options, error_func, devices):
[email protected]6bc1bda22013-07-19 22:08:37548 """Subcommand of RunTestsCommands which runs instrumentation tests."""
[email protected]2a684222013-08-01 16:59:22549 instrumentation_options = ProcessInstrumentationOptions(options, error_func)
[email protected]6bc1bda22013-07-19 22:08:37550
[email protected]f7148dd42013-08-20 14:24:57551 if len(devices) > 1 and options.wait_for_debugger:
552 logging.warning('Debugger can not be sharded, using first available device')
553 devices = devices[:1]
554
[email protected]6bc1bda22013-07-19 22:08:37555 results = base_test_result.TestRunResults()
556 exit_code = 0
557
558 if options.run_java_tests:
[email protected]2a684222013-08-01 16:59:22559 runner_factory, tests = instrumentation_setup.Setup(instrumentation_options)
[email protected]6bc1bda22013-07-19 22:08:37560
561 test_results, exit_code = test_dispatcher.RunTests(
[email protected]f7148dd42013-08-20 14:24:57562 tests, runner_factory, devices, shard=True, test_timeout=None,
[email protected]6bc1bda22013-07-19 22:08:37563 num_retries=options.num_retries)
564
565 results.AddTestRunResults(test_results)
566
567 if options.run_python_tests:
[email protected]37ee0c792013-08-06 19:10:13568 runner_factory, tests = host_driven_setup.InstrumentationSetup(
[email protected]67954f822013-08-14 18:09:08569 options.host_driven_root, options.official_build,
[email protected]37ee0c792013-08-06 19:10:13570 instrumentation_options)
571
[email protected]34020022013-08-06 23:35:34572 if tests:
573 test_results, test_exit_code = test_dispatcher.RunTests(
[email protected]f7148dd42013-08-20 14:24:57574 tests, runner_factory, devices, shard=True, test_timeout=None,
[email protected]34020022013-08-06 23:35:34575 num_retries=options.num_retries)
[email protected]6bc1bda22013-07-19 22:08:37576
[email protected]34020022013-08-06 23:35:34577 results.AddTestRunResults(test_results)
[email protected]6bc1bda22013-07-19 22:08:37578
[email protected]34020022013-08-06 23:35:34579 # Only allow exit code escalation
580 if test_exit_code and exit_code != constants.ERROR_EXIT_CODE:
581 exit_code = test_exit_code
[email protected]6bc1bda22013-07-19 22:08:37582
583 report_results.LogFull(
584 results=results,
585 test_type='Instrumentation',
586 test_package=os.path.basename(options.test_apk),
587 annotation=options.annotations,
[email protected]6bc1bda22013-07-19 22:08:37588 flakiness_server=options.flakiness_dashboard_server)
589
590 return exit_code
591
592
[email protected]f7148dd42013-08-20 14:24:57593def _RunUIAutomatorTests(options, error_func, devices):
[email protected]6bc1bda22013-07-19 22:08:37594 """Subcommand of RunTestsCommands which runs uiautomator tests."""
[email protected]2a684222013-08-01 16:59:22595 uiautomator_options = ProcessUIAutomatorOptions(options, error_func)
[email protected]6bc1bda22013-07-19 22:08:37596
[email protected]37ee0c792013-08-06 19:10:13597 runner_factory, tests = uiautomator_setup.Setup(uiautomator_options)
[email protected]6bc1bda22013-07-19 22:08:37598
[email protected]37ee0c792013-08-06 19:10:13599 results, exit_code = test_dispatcher.RunTests(
[email protected]f7148dd42013-08-20 14:24:57600 tests, runner_factory, devices, shard=True, test_timeout=None,
[email protected]37ee0c792013-08-06 19:10:13601 num_retries=options.num_retries)
[email protected]6bc1bda22013-07-19 22:08:37602
603 report_results.LogFull(
604 results=results,
605 test_type='UIAutomator',
606 test_package=os.path.basename(options.test_jar),
607 annotation=options.annotations,
[email protected]6bc1bda22013-07-19 22:08:37608 flakiness_server=options.flakiness_dashboard_server)
609
610 return exit_code
611
612
[email protected]f7148dd42013-08-20 14:24:57613def _RunMonkeyTests(options, error_func, devices):
[email protected]3dbdfa42013-08-08 01:08:14614 """Subcommand of RunTestsCommands which runs monkey tests."""
615 monkey_options = ProcessMonkeyTestOptions(options, error_func)
616
617 runner_factory, tests = monkey_setup.Setup(monkey_options)
618
619 results, exit_code = test_dispatcher.RunTests(
[email protected]181a5c92013-09-06 17:11:46620 tests, runner_factory, devices, shard=False, test_timeout=None,
621 num_retries=options.num_retries)
[email protected]3dbdfa42013-08-08 01:08:14622
623 report_results.LogFull(
624 results=results,
625 test_type='Monkey',
[email protected]14b3b1202013-08-15 22:25:28626 test_package='Monkey')
[email protected]3dbdfa42013-08-08 01:08:14627
628 return exit_code
629
630
[email protected]ad32f312013-11-13 04:03:29631def _RunPerfTests(options, args, error_func, devices):
[email protected]ec3170b2013-08-14 14:39:47632 """Subcommand of RunTestsCommands which runs perf tests."""
[email protected]ad32f312013-11-13 04:03:29633 perf_options = ProcessPerfTestOptions(options, args, error_func)
634 # Just print the results from a single previously executed step.
[email protected]ec3170b2013-08-14 14:39:47635 if perf_options.print_step:
636 return perf_test_runner.PrintTestOutput(perf_options.print_step)
637
638 runner_factory, tests = perf_setup.Setup(perf_options)
639
[email protected]86184c7b2013-08-15 15:06:57640 results, _ = test_dispatcher.RunTests(
[email protected]181a5c92013-09-06 17:11:46641 tests, runner_factory, devices, shard=True, test_timeout=None,
642 num_retries=options.num_retries)
[email protected]ec3170b2013-08-14 14:39:47643
644 report_results.LogFull(
645 results=results,
646 test_type='Perf',
[email protected]865a47a2013-08-16 14:01:12647 test_package='Perf')
[email protected]def4bce2013-11-12 12:59:52648
649 if perf_options.single_step:
650 return perf_test_runner.PrintTestOutput('single_step')
651
[email protected]11ce8452014-02-17 10:55:03652 perf_test_runner.PrintSummary(tests)
653
[email protected]86184c7b2013-08-15 15:06:57654 # Always return 0 on the sharding stage. Individual tests exit_code
655 # will be returned on the print_step stage.
656 return 0
[email protected]ec3170b2013-08-14 14:39:47657
[email protected]3dbdfa42013-08-08 01:08:14658
[email protected]f7148dd42013-08-20 14:24:57659def _GetAttachedDevices(test_device=None):
660 """Get all attached devices.
661
662 Args:
663 test_device: Name of a specific device to use.
664
665 Returns:
666 A list of attached devices.
667 """
668 attached_devices = []
669
670 attached_devices = android_commands.GetAttachedDevices()
671 if test_device:
672 assert test_device in attached_devices, (
673 'Did not find device %s among attached device. Attached devices: %s'
674 % (test_device, ', '.join(attached_devices)))
675 attached_devices = [test_device]
676
677 assert attached_devices, 'No devices attached.'
678
679 return sorted(attached_devices)
680
681
[email protected]fbe29322013-07-09 09:03:26682def RunTestsCommand(command, options, args, option_parser):
683 """Checks test type and dispatches to the appropriate function.
684
685 Args:
686 command: String indicating the command that was received to trigger
687 this function.
688 options: optparse options dictionary.
689 args: List of extra args from optparse.
690 option_parser: optparse.OptionParser object.
691
692 Returns:
693 Integer indicated exit code.
[email protected]b3873892013-07-10 04:57:10694
695 Raises:
696 Exception: Unknown command name passed in, or an exception from an
697 individual test runner.
[email protected]fbe29322013-07-09 09:03:26698 """
699
[email protected]d82f0252013-07-12 23:22:57700 # Check for extra arguments
[email protected]ad32f312013-11-13 04:03:29701 if len(args) > 2 and command != 'perf':
[email protected]d82f0252013-07-12 23:22:57702 option_parser.error('Unrecognized arguments: %s' % (' '.join(args[2:])))
703 return constants.ERROR_EXIT_CODE
[email protected]ad32f312013-11-13 04:03:29704 if command == 'perf':
705 if ((options.single_step and len(args) <= 2) or
706 (not options.single_step and len(args) > 2)):
707 option_parser.error('Unrecognized arguments: %s' % (' '.join(args)))
708 return constants.ERROR_EXIT_CODE
[email protected]d82f0252013-07-12 23:22:57709
[email protected]fbe29322013-07-09 09:03:26710 ProcessCommonOptions(options)
711
[email protected]f7148dd42013-08-20 14:24:57712 devices = _GetAttachedDevices(options.test_device)
713
[email protected]c0662e092013-11-12 11:51:25714 forwarder.Forwarder.RemoveHostLog()
[email protected]6b11583b2013-11-21 16:18:40715 if not ports.ResetTestServerPortAllocation():
716 raise Exception('Failed to reset test server port.')
[email protected]c0662e092013-11-12 11:51:25717
[email protected]fbe29322013-07-09 09:03:26718 if command == 'gtest':
[email protected]e9ee6ec32014-02-11 01:47:41719 return _RunGTests(options, option_parser.error, devices)
[email protected]6b6abac6d2013-10-03 11:56:38720 elif command == 'linker':
[email protected]e9ee6ec32014-02-11 01:47:41721 return _RunLinkerTests(options, option_parser.error, devices)
[email protected]fbe29322013-07-09 09:03:26722 elif command == 'instrumentation':
[email protected]f7148dd42013-08-20 14:24:57723 return _RunInstrumentationTests(options, option_parser.error, devices)
[email protected]fbe29322013-07-09 09:03:26724 elif command == 'uiautomator':
[email protected]f7148dd42013-08-20 14:24:57725 return _RunUIAutomatorTests(options, option_parser.error, devices)
[email protected]3dbdfa42013-08-08 01:08:14726 elif command == 'monkey':
[email protected]f7148dd42013-08-20 14:24:57727 return _RunMonkeyTests(options, option_parser.error, devices)
[email protected]ec3170b2013-08-14 14:39:47728 elif command == 'perf':
[email protected]ad32f312013-11-13 04:03:29729 return _RunPerfTests(options, args, option_parser.error, devices)
[email protected]fbe29322013-07-09 09:03:26730 else:
[email protected]6bc1bda22013-07-19 22:08:37731 raise Exception('Unknown test type.')
[email protected]fbe29322013-07-09 09:03:26732
[email protected]fbe29322013-07-09 09:03:26733
[email protected]e9ee6ec32014-02-11 01:47:41734def HelpCommand(command, options, args, option_parser):
[email protected]fbe29322013-07-09 09:03:26735 """Display help for a certain command, or overall help.
736
737 Args:
738 command: String indicating the command that was received to trigger
739 this function.
[email protected]e9ee6ec32014-02-11 01:47:41740 options: optparse options dictionary.
[email protected]fbe29322013-07-09 09:03:26741 args: List of extra args from optparse.
742 option_parser: optparse.OptionParser object.
743
744 Returns:
745 Integer indicated exit code.
746 """
747 # If we don't have any args, display overall help
748 if len(args) < 3:
749 option_parser.print_help()
750 return 0
[email protected]d82f0252013-07-12 23:22:57751 # If we have too many args, print an error
752 if len(args) > 3:
753 option_parser.error('Unrecognized arguments: %s' % (' '.join(args[3:])))
754 return constants.ERROR_EXIT_CODE
[email protected]fbe29322013-07-09 09:03:26755
756 command = args[2]
757
758 if command not in VALID_COMMANDS:
759 option_parser.error('Unrecognized command.')
760
761 # Treat the help command as a special case. We don't care about showing a
762 # specific help page for itself.
763 if command == 'help':
764 option_parser.print_help()
765 return 0
766
767 VALID_COMMANDS[command].add_options_func(option_parser)
768 option_parser.usage = '%prog ' + command + ' [options]'
[email protected]dfffbcbc2013-09-17 22:06:01769 option_parser.commands_dict = {}
[email protected]fbe29322013-07-09 09:03:26770 option_parser.print_help()
771
772 return 0
773
774
775# Define a named tuple for the values in the VALID_COMMANDS dictionary so the
776# syntax is a bit prettier. The tuple is two functions: (add options, run
777# command).
778CommandFunctionTuple = collections.namedtuple(
779 'CommandFunctionTuple', ['add_options_func', 'run_command_func'])
780VALID_COMMANDS = {
781 'gtest': CommandFunctionTuple(AddGTestOptions, RunTestsCommand),
[email protected]fbe29322013-07-09 09:03:26782 'instrumentation': CommandFunctionTuple(
783 AddInstrumentationTestOptions, RunTestsCommand),
784 'uiautomator': CommandFunctionTuple(
785 AddUIAutomatorTestOptions, RunTestsCommand),
[email protected]3dbdfa42013-08-08 01:08:14786 'monkey': CommandFunctionTuple(
787 AddMonkeyTestOptions, RunTestsCommand),
[email protected]ec3170b2013-08-14 14:39:47788 'perf': CommandFunctionTuple(
789 AddPerfTestOptions, RunTestsCommand),
[email protected]6b6abac6d2013-10-03 11:56:38790 'linker': CommandFunctionTuple(
791 AddLinkerTestOptions, RunTestsCommand),
[email protected]fbe29322013-07-09 09:03:26792 'help': CommandFunctionTuple(lambda option_parser: None, HelpCommand)
793 }
794
795
[email protected]e9ee6ec32014-02-11 01:47:41796def DumpThreadStacks(signal, frame):
[email protected]71aec4b2013-11-20 00:35:24797 for thread in threading.enumerate():
798 reraiser_thread.LogThreadStack(thread)
[email protected]83bb8152013-11-19 15:02:21799
800
[email protected]e9ee6ec32014-02-11 01:47:41801def main(argv):
[email protected]83bb8152013-11-19 15:02:21802 signal.signal(signal.SIGUSR1, DumpThreadStacks)
[email protected]803f65a72013-08-20 19:11:30803 option_parser = command_option_parser.CommandOptionParser(
804 commands_dict=VALID_COMMANDS)
805 return command_option_parser.ParseAndExecute(option_parser)
[email protected]fbe29322013-07-09 09:03:26806
[email protected]fbe29322013-07-09 09:03:26807
808if __name__ == '__main__':
[email protected]e9ee6ec32014-02-11 01:47:41809 sys.exit(main(sys.argv))