build/android/test_runner.py - Issue 2392643003: Removes files from //build that we don't need

Unified Diff: build/android/test_runner.py

Issue 2392643003: Removes files from //build that we don't need (Closed)

Patch Set: Created 4 years, 2 months ago

Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.

Jump to:

View side-by-side diff with in-line comments

Index: build/android/test_runner.py

diff --git a/build/android/test_runner.py b/build/android/test_runner.py

deleted file mode 100755

index 1fc48ecf5050a3ec6d5354f8b573736e5b6e4c31..0000000000000000000000000000000000000000

--- a/build/android/test_runner.py

+++ /dev/null

@@ -1,1067 +0,0 @@

-#!/usr/bin/env python

-# Use of this source code is governed by a BSD-style license that can be

-# found in the LICENSE file.

-"""Runs all types of tests from one unified interface."""

-import argparse

-import collections

-import logging

-import os

-import shutil

-import signal

-import sys

-import threading

-import unittest

-from pylib import constants

-from pylib import forwarder

-from pylib import ports

-from pylib.base import base_test_result

-from pylib.base import environment_factory

-from pylib.base import test_dispatcher

-from pylib.base import test_instance_factory

-from pylib.base import test_run_factory

-from pylib.device import device_errors

-from pylib.device import device_utils

-from pylib.gtest import gtest_config

-# TODO(jbudorick): Remove this once we stop selectively enabling platform mode.

-from pylib.gtest import gtest_test_instance

-from pylib.gtest import setup as gtest_setup

-from pylib.gtest import test_options as gtest_test_options

-from pylib.linker import setup as linker_setup

-from pylib.host_driven import setup as host_driven_setup

-from pylib.instrumentation import setup as instrumentation_setup

-from pylib.instrumentation import test_options as instrumentation_test_options

-from pylib.junit import setup as junit_setup

-from pylib.junit import test_dispatcher as junit_dispatcher

-from pylib.monkey import setup as monkey_setup

-from pylib.monkey import test_options as monkey_test_options

-from pylib.perf import setup as perf_setup

-from pylib.perf import test_options as perf_test_options

-from pylib.perf import test_runner as perf_test_runner

-from pylib.results import json_results

-from pylib.results import report_results

-from pylib.uiautomator import setup as uiautomator_setup

-from pylib.uiautomator import test_options as uiautomator_test_options

-from pylib.utils import apk_helper

-from pylib.utils import base_error

-from pylib.utils import reraiser_thread

-from pylib.utils import run_tests_helper

-def AddCommonOptions(parser):

- """Adds all common options to |parser|."""

- group = parser.add_argument_group('Common Options')

- default_build_type = os.environ.get('BUILDTYPE', 'Debug')

- debug_or_release_group = group.add_mutually_exclusive_group()

- debug_or_release_group.add_argument(

- '--debug', action='store_const', const='Debug', dest='build_type',

- default=default_build_type,

- help=('If set, run test suites under out/Debug. '

- 'Default is env var BUILDTYPE or Debug.'))

- debug_or_release_group.add_argument(

- '--release', action='store_const', const='Release', dest='build_type',

- help=('If set, run test suites under out/Release. '

- 'Default is env var BUILDTYPE or Debug.'))

- group.add_argument('--build-directory', dest='build_directory',

- help=('Path to the directory in which build files are'

- ' located (should not include build type)'))

- group.add_argument('--output-directory', dest='output_directory',

- help=('Path to the directory in which build files are'

- ' located (must include build type). This will take'

- ' precedence over --debug, --release and'

- ' --build-directory'))

- group.add_argument('--num_retries', dest='num_retries', type=int, default=2,

- help=('Number of retries for a test before '

- 'giving up (default: %(default)s).'))

- group.add_argument('-v',

- '--verbose',

- dest='verbose_count',

- default=0,

- action='count',

- help='Verbose level (multiple times for more)')

- group.add_argument('--flakiness-dashboard-server',

- dest='flakiness_dashboard_server',

- help=('Address of the server that is hosting the '

- 'Chrome for Android flakiness dashboard.'))

- group.add_argument('--enable-platform-mode', action='store_true',

- help=('Run the test scripts in platform mode, which '

- 'conceptually separates the test runner from the '

- '"device" (local or remote, real or emulated) on '

- 'which the tests are running. [experimental]'))

- group.add_argument('-e', '--environment', default='local',

- choices=constants.VALID_ENVIRONMENTS,

- help='Test environment to run in (default: %(default)s).')

- group.add_argument('--adb-path',

- help=('Specify the absolute path of the adb binary that '

- 'should be used.'))

- group.add_argument('--json-results-file', dest='json_results_file',

- help='If set, will dump results in JSON form '

- 'to specified file.')

-def ProcessCommonOptions(args):

- """Processes and handles all common options."""

- run_tests_helper.SetLogLevel(args.verbose_count)

- constants.SetBuildType(args.build_type)

- if args.build_directory:

- constants.SetBuildDirectory(args.build_directory)

- if args.output_directory:

- constants.SetOutputDirectory(args.output_directory)

- if args.adb_path:

- constants.SetAdbPath(args.adb_path)

- # Some things such as Forwarder require ADB to be in the environment path.

- adb_dir = os.path.dirname(constants.GetAdbPath())

- if adb_dir and adb_dir not in os.environ['PATH'].split(os.pathsep):

- os.environ['PATH'] = adb_dir + os.pathsep + os.environ['PATH']

-def AddRemoteDeviceOptions(parser):

- group = parser.add_argument_group('Remote Device Options')

- group.add_argument('--trigger',

- help=('Only triggers the test if set. Stores test_run_id '

- 'in given file path. '))

- group.add_argument('--collect',

- help=('Only collects the test results if set. '

- 'Gets test_run_id from given file path.'))

- group.add_argument('--remote-device', action='append',

- help='Device type to run test on.')

- group.add_argument('--results-path',

- help='File path to download results to.')

- group.add_argument('--api-protocol',

- help='HTTP protocol to use. (http or https)')

- group.add_argument('--api-address',

- help='Address to send HTTP requests.')

- group.add_argument('--api-port',

- help='Port to send HTTP requests to.')

- group.add_argument('--runner-type',

- help='Type of test to run as.')

- group.add_argument('--runner-package',

- help='Package name of test.')

- group.add_argument('--device-type',

- choices=constants.VALID_DEVICE_TYPES,

- help=('Type of device to run on. iOS or android'))

- group.add_argument('--device-oem', action='append',

- help='Device OEM to run on.')

- group.add_argument('--remote-device-file',

- help=('File with JSON to select remote device. '

- 'Overrides all other flags.'))

- group.add_argument('--remote-device-timeout', type=int,

- help='Times to retry finding remote device')

- group.add_argument('--network-config', type=int,

- help='Integer that specifies the network environment '

- 'that the tests will be run in.')

- device_os_group = group.add_mutually_exclusive_group()

- device_os_group.add_argument('--remote-device-minimum-os',

- help='Minimum OS on device.')

- device_os_group.add_argument('--remote-device-os', action='append',

- help='OS to have on the device.')

- api_secret_group = group.add_mutually_exclusive_group()

- api_secret_group.add_argument('--api-secret', default='',

- help='API secret for remote devices.')

- api_secret_group.add_argument('--api-secret-file', default='',

- help='Path to file that contains API secret.')

- api_key_group = group.add_mutually_exclusive_group()

- api_key_group.add_argument('--api-key', default='',

- help='API key for remote devices.')

- api_key_group.add_argument('--api-key-file', default='',

- help='Path to file that contains API key.')

-def AddDeviceOptions(parser):

- """Adds device options to |parser|."""

- group = parser.add_argument_group(title='Device Options')

- group.add_argument('--tool',

- dest='tool',

- help=('Run the test under a tool '

- '(use --tool help to list them)'))

- group.add_argument('-d', '--device', dest='test_device',

- help=('Target device for the test suite '

- 'to run on.'))

-def AddGTestOptions(parser):

- """Adds gtest options to |parser|."""

- gtest_suites = list(gtest_config.STABLE_TEST_SUITES

- + gtest_config.EXPERIMENTAL_TEST_SUITES)

- group = parser.add_argument_group('GTest Options')

- group.add_argument('-s', '--suite', dest='suite_name',

- nargs='+', metavar='SUITE_NAME', required=True,

- help=('Executable name of the test suite to run. '

- 'Available suites include (but are not limited to): '

- '%s' % ', '.join('"%s"' % s for s in gtest_suites)))

- group.add_argument('--gtest_also_run_disabled_tests',

- '--gtest-also-run-disabled-tests',

- dest='run_disabled', action='store_true',

- help='Also run disabled tests if applicable.')

- group.add_argument('-a', '--test-arguments', dest='test_arguments',

- default='',

- help='Additional arguments to pass to the test.')

- group.add_argument('-t', dest='timeout', type=int, default=60,

- help='Timeout to wait for each test '

- '(default: %(default)s).')

- group.add_argument('--isolate_file_path',

- '--isolate-file-path',

- dest='isolate_file_path',

- help='.isolate file path to override the default '

- 'path')

- group.add_argument('--app-data-file', action='append', dest='app_data_files',

- help='A file path relative to the app data directory '

- 'that should be saved to the host.')

- group.add_argument('--app-data-file-dir',

- help='Host directory to which app data files will be'

- ' saved. Used with --app-data-file.')

- group.add_argument('--delete-stale-data', dest='delete_stale_data',

- action='store_true',

- help='Delete stale test data on the device.')

- filter_group = group.add_mutually_exclusive_group()

- filter_group.add_argument('-f', '--gtest_filter', '--gtest-filter',

- dest='test_filter',

- help='googletest-style filter string.')

- filter_group.add_argument('--gtest-filter-file', dest='test_filter_file',

- help='Path to file that contains googletest-style '

- 'filter strings. (Lines will be joined with '

- '":" to create a single filter string.)')

- AddDeviceOptions(parser)

- AddCommonOptions(parser)

- AddRemoteDeviceOptions(parser)

-def AddLinkerTestOptions(parser):

- group = parser.add_argument_group('Linker Test Options')

- group.add_argument('-f', '--gtest-filter', dest='test_filter',

- help='googletest-style filter string.')

- AddCommonOptions(parser)

- AddDeviceOptions(parser)

-def AddJavaTestOptions(argument_group):

- """Adds the Java test options to |option_parser|."""

- argument_group.add_argument(

- '-f', '--test-filter', dest='test_filter',

- help=('Test filter (if not fully qualified, will run all matches).'))

- argument_group.add_argument(

- '-A', '--annotation', dest='annotation_str',

- help=('Comma-separated list of annotations. Run only tests with any of '

- 'the given annotations. An annotation can be either a key or a '

- 'key-values pair. A test that has no annotation is considered '

- '"SmallTest".'))

- argument_group.add_argument(

- '-E', '--exclude-annotation', dest='exclude_annotation_str',

- help=('Comma-separated list of annotations. Exclude tests with these '

- 'annotations.'))

- argument_group.add_argument(

- '--screenshot', dest='screenshot_failures', action='store_true',

- help='Capture screenshots of test failures')

- argument_group.add_argument(

- '--save-perf-json', action='store_true',

- help='Saves the JSON file for each UI Perf test.')

- argument_group.add_argument(

- '--official-build', action='store_true', help='Run official build tests.')

- argument_group.add_argument(

- '--test_data', '--test-data', action='append', default=[],

- help=('Each instance defines a directory of test data that should be '

- 'copied to the target(s) before running the tests. The argument '

- 'should be of the form <target>:<source>, <target> is relative to '

- 'the device data directory, and <source> is relative to the '

- 'chromium build directory.'))

- argument_group.add_argument(

- '--disable-dalvik-asserts', dest='set_asserts', action='store_false',

- default=True, help='Removes the dalvik.vm.enableassertions property')

-def ProcessJavaTestOptions(args):

- """Processes options/arguments and populates |options| with defaults."""

- # TODO(jbudorick): Handle most of this function in argparse.

- if args.annotation_str:

- args.annotations = args.annotation_str.split(',')

- elif args.test_filter:

- args.annotations = []

- else:

- args.annotations = ['Smoke', 'SmallTest', 'MediumTest', 'LargeTest',

- 'EnormousTest', 'IntegrationTest']

- if args.exclude_annotation_str:

- args.exclude_annotations = args.exclude_annotation_str.split(',')

- else:

- args.exclude_annotations = []

-def AddInstrumentationTestOptions(parser):

- """Adds Instrumentation test options to |parser|."""

- parser.usage = '%(prog)s [options]'

- group = parser.add_argument_group('Instrumentation Test Options')

- AddJavaTestOptions(group)

- java_or_python_group = group.add_mutually_exclusive_group()

- java_or_python_group.add_argument(

- '-j', '--java-only', action='store_false',

- dest='run_python_tests', default=True, help='Run only the Java tests.')

- java_or_python_group.add_argument(

- '-p', '--python-only', action='store_false',

- dest='run_java_tests', default=True,

- help='Run only the host-driven tests.')

- group.add_argument('--host-driven-root',

- help='Root of the host-driven tests.')

- group.add_argument('-w', '--wait_debugger', dest='wait_for_debugger',

- action='store_true',

- help='Wait for debugger.')

- group.add_argument('--apk-under-test', dest='apk_under_test',

- help=('the name of the apk under test.'))

- group.add_argument('--test-apk', dest='test_apk', required=True,

- help=('The name of the apk containing the tests '

- '(without the .apk extension; '

- 'e.g. "ContentShellTest").'))

- group.add_argument('--support-apk', dest='test_support_apk_path',

- help=('The path to an optional support apk to be '

- 'installed alongside the test apk. The '

- 'path should be relative to the output '

- 'directory (--output-directory).'))

- group.add_argument('--coverage-dir',

- help=('Directory in which to place all generated '

- 'EMMA coverage files.'))

- group.add_argument('--device-flags', dest='device_flags', default='',

- help='The relative filepath to a file containing '

- 'command-line flags to set on the device')

- group.add_argument('--device-flags-file', default='',

- help='The relative filepath to a file containing '

- 'command-line flags to set on the device')

- group.add_argument('--isolate_file_path',

- '--isolate-file-path',

- dest='isolate_file_path',

- help='.isolate file path to override the default '

- 'path')

- group.add_argument('--delete-stale-data', dest='delete_stale_data',

- action='store_true',

- help='Delete stale test data on the device.')

- AddCommonOptions(parser)

- AddDeviceOptions(parser)

- AddRemoteDeviceOptions(parser)

-def ProcessInstrumentationOptions(args):

- """Processes options/arguments and populate |options| with defaults.

- Args:

- args: argparse.Namespace object.

- Returns:

- An InstrumentationOptions named tuple which contains all options relevant to

- instrumentation tests.

- """

- ProcessJavaTestOptions(args)

- if not args.host_driven_root:

- args.run_python_tests = False

- args.test_apk_path = os.path.join(

- constants.GetOutDirectory(),

- constants.SDK_BUILD_APKS_DIR,

- '%s.apk' % args.test_apk)

- args.test_apk_jar_path = os.path.join(

- constants.GetOutDirectory(),

- constants.SDK_BUILD_TEST_JAVALIB_DIR,

- '%s.jar' % args.test_apk)

- args.test_runner = apk_helper.GetInstrumentationName(args.test_apk_path)

- # TODO(jbudorick): Get rid of InstrumentationOptions.

- return instrumentation_test_options.InstrumentationOptions(

- args.tool,

- args.annotations,

- args.exclude_annotations,

- args.test_filter,

- args.test_data,

- args.save_perf_json,

- args.screenshot_failures,

- args.wait_for_debugger,

- args.coverage_dir,

- args.test_apk,

- args.test_apk_path,

- args.test_apk_jar_path,

- args.test_runner,

- args.test_support_apk_path,

- args.device_flags,

- args.isolate_file_path,

- args.set_asserts,

- args.delete_stale_data

- )

-def AddUIAutomatorTestOptions(parser):

- """Adds UI Automator test options to |parser|."""

- group = parser.add_argument_group('UIAutomator Test Options')

- AddJavaTestOptions(group)

- group.add_argument(

- '--package', required=True, choices=constants.PACKAGE_INFO.keys(),

- metavar='PACKAGE', help='Package under test.')

- group.add_argument(

- '--test-jar', dest='test_jar', required=True,

- help=('The name of the dexed jar containing the tests (without the '

- '.dex.jar extension). Alternatively, this can be a full path '

- 'to the jar.'))

- AddCommonOptions(parser)

- AddDeviceOptions(parser)

-def ProcessUIAutomatorOptions(args):

- """Processes UIAutomator options/arguments.

- Args:

- args: argparse.Namespace object.

- Returns:

- A UIAutomatorOptions named tuple which contains all options relevant to

- uiautomator tests.

- """

- ProcessJavaTestOptions(args)

- if os.path.exists(args.test_jar):

- # The dexed JAR is fully qualified, assume the info JAR lives along side.

- args.uiautomator_jar = args.test_jar

- else:

- args.uiautomator_jar = os.path.join(

- constants.GetOutDirectory(),

- constants.SDK_BUILD_JAVALIB_DIR,

- '%s.dex.jar' % args.test_jar)

- args.uiautomator_info_jar = (

- args.uiautomator_jar[:args.uiautomator_jar.find('.dex.jar')] +

- '_java.jar')

- return uiautomator_test_options.UIAutomatorOptions(

- args.tool,

- args.annotations,

- args.exclude_annotations,

- args.test_filter,

- args.test_data,

- args.save_perf_json,

- args.screenshot_failures,

- args.uiautomator_jar,

- args.uiautomator_info_jar,

- args.package,

- args.set_asserts)

-def AddJUnitTestOptions(parser):

- """Adds junit test options to |parser|."""

- group = parser.add_argument_group('JUnit Test Options')

- group.add_argument(

- '-s', '--test-suite', dest='test_suite', required=True,

- help=('JUnit test suite to run.'))

- group.add_argument(

- '-f', '--test-filter', dest='test_filter',

- help='Filters tests googletest-style.')

- group.add_argument(

- '--package-filter', dest='package_filter',

- help='Filters tests by package.')

- group.add_argument(

- '--runner-filter', dest='runner_filter',

- help='Filters tests by runner class. Must be fully qualified.')

- group.add_argument(

- '--sdk-version', dest='sdk_version', type=int,

- help='The Android SDK version.')

- AddCommonOptions(parser)

-def AddMonkeyTestOptions(parser):

- """Adds monkey test options to |parser|."""

- group = parser.add_argument_group('Monkey Test Options')

- group.add_argument(

- '--package', required=True, choices=constants.PACKAGE_INFO.keys(),

- metavar='PACKAGE', help='Package under test.')

- group.add_argument(

- '--event-count', default=10000, type=int,

- help='Number of events to generate (default: %(default)s).')

- group.add_argument(

- '--category', default='',

- help='A list of allowed categories.')

- group.add_argument(

- '--throttle', default=100, type=int,

- help='Delay between events (ms) (default: %(default)s). ')

- group.add_argument(

- '--seed', type=int,

- help=('Seed value for pseudo-random generator. Same seed value generates '

- 'the same sequence of events. Seed is randomized by default.'))

- group.add_argument(

- '--extra-args', default='',

- help=('String of other args to pass to the command verbatim.'))

- AddCommonOptions(parser)

- AddDeviceOptions(parser)

-def ProcessMonkeyTestOptions(args):

- """Processes all monkey test options.

- Args:

- args: argparse.Namespace object.

- Returns:

- A MonkeyOptions named tuple which contains all options relevant to

- monkey tests.

- """

- # TODO(jbudorick): Handle this directly in argparse with nargs='+'

- category = args.category

- if category:

- category = args.category.split(',')

- # TODO(jbudorick): Get rid of MonkeyOptions.

- return monkey_test_options.MonkeyOptions(

- args.verbose_count,

- args.package,

- args.event_count,

- category,

- args.throttle,

- args.seed,

- args.extra_args)

-def AddUirobotTestOptions(parser):

- """Adds uirobot test options to |option_parser|."""

- group = parser.add_argument_group('Uirobot Test Options')

- group.add_argument('--app-under-test', required=True,

- help='APK to run tests on.')

- group.add_argument(

- '--minutes', default=5, type=int,

- help='Number of minutes to run uirobot test [default: %(default)s].')

- AddCommonOptions(parser)

- AddDeviceOptions(parser)

- AddRemoteDeviceOptions(parser)

-def AddPerfTestOptions(parser):

- """Adds perf test options to |parser|."""

- group = parser.add_argument_group('Perf Test Options')

- class SingleStepAction(argparse.Action):

- def __call__(self, parser, namespace, values, option_string=None):

- if values and not namespace.single_step:

- parser.error('single step command provided, '

- 'but --single-step not specified.')

- elif namespace.single_step and not values:

- parser.error('--single-step specified, '

- 'but no single step command provided.')

- setattr(namespace, self.dest, values)

- step_group = group.add_mutually_exclusive_group(required=True)

- # TODO(jbudorick): Revise --single-step to use argparse.REMAINDER.

- # This requires removing "--" from client calls.

- step_group.add_argument(

- '--single-step', action='store_true',

- help='Execute the given command with retries, but only print the result '

- 'for the "most successful" round.')

- step_group.add_argument(

- '--steps',

- help='JSON file containing the list of commands to run.')

- step_group.add_argument(

- '--print-step',

- help='The name of a previously executed perf step to print.')

- group.add_argument(

- '--output-json-list',

- help='Write a simple list of names from --steps into the given file.')

- group.add_argument(

- '--collect-chartjson-data',

- action='store_true',

- help='Cache the chartjson output from each step for later use.')

- group.add_argument(

- '--output-chartjson-data',

- default='',

- help='Write out chartjson into the given file.')

- group.add_argument(

- '--flaky-steps',

- help=('A JSON file containing steps that are flaky '

- 'and will have its exit code ignored.'))

- group.add_argument(

- '--no-timeout', action='store_true',

- help=('Do not impose a timeout. Each perf step is responsible for '

- 'implementing the timeout logic.'))

- group.add_argument(

- '-f', '--test-filter',

- help=('Test filter (will match against the names listed in --steps).'))

- group.add_argument(

- '--dry-run', action='store_true',

- help='Just print the steps without executing.')

- # Uses 0.1 degrees C because that's what Android does.

- group.add_argument(

- '--max-battery-temp', type=int,

- help='Only start tests when the battery is at or below the given '

- 'temperature (0.1 C)')

- group.add_argument('single_step_command', nargs='*', action=SingleStepAction,

- help='If --single-step is specified, the command to run.')

- group.add_argument('--min-battery-level', type=int,

- help='Only starts tests when the battery is charged above '

- 'given level.')

- AddCommonOptions(parser)

- AddDeviceOptions(parser)

-def ProcessPerfTestOptions(args):

- """Processes all perf test options.

- Args:

- args: argparse.Namespace object.

- Returns:

- A PerfOptions named tuple which contains all options relevant to

- perf tests.

- """

- # TODO(jbudorick): Move single_step handling down into the perf tests.

- if args.single_step:

- args.single_step = ' '.join(args.single_step_command)

- # TODO(jbudorick): Get rid of PerfOptions.

- return perf_test_options.PerfOptions(

- args.steps, args.flaky_steps, args.output_json_list,

- args.print_step, args.no_timeout, args.test_filter,

- args.dry_run, args.single_step, args.collect_chartjson_data,

- args.output_chartjson_data, args.max_battery_temp, args.min_battery_level)

-def AddPythonTestOptions(parser):

- group = parser.add_argument_group('Python Test Options')

- group.add_argument(

- '-s', '--suite', dest='suite_name', metavar='SUITE_NAME',

- choices=constants.PYTHON_UNIT_TEST_SUITES.keys(),

- help='Name of the test suite to run.')

- AddCommonOptions(parser)

-def _RunGTests(args, devices):

- """Subcommand of RunTestsCommands which runs gtests."""

- exit_code = 0

- for suite_name in args.suite_name:

- # TODO(jbudorick): Either deprecate multi-suite or move its handling down

- # into the gtest code.

- gtest_options = gtest_test_options.GTestOptions(

- args.tool,

- args.test_filter,

- args.run_disabled,

- args.test_arguments,

- args.timeout,

- args.isolate_file_path,

- suite_name,

- args.app_data_files,

- args.app_data_file_dir,

- args.delete_stale_data)

- runner_factory, tests = gtest_setup.Setup(gtest_options, devices)

- results, test_exit_code = test_dispatcher.RunTests(

- tests, runner_factory, devices, shard=True, test_timeout=None,

- num_retries=args.num_retries)

- if test_exit_code and exit_code != constants.ERROR_EXIT_CODE:

- exit_code = test_exit_code

- report_results.LogFull(

- results=results,

- test_type='Unit test',

- test_package=suite_name,

- flakiness_server=args.flakiness_dashboard_server)

- if args.json_results_file:

- json_results.GenerateJsonResultsFile(results, args.json_results_file)

- return exit_code

-def _RunLinkerTests(args, devices):

- """Subcommand of RunTestsCommands which runs linker tests."""

- runner_factory, tests = linker_setup.Setup(args, devices)

- results, exit_code = test_dispatcher.RunTests(

- tests, runner_factory, devices, shard=True, test_timeout=60,

- num_retries=args.num_retries)

- report_results.LogFull(

- results=results,

- test_type='Linker test',

- test_package='ChromiumLinkerTest')

- if args.json_results_file:

- json_results.GenerateJsonResultsFile(results, args.json_results_file)

- return exit_code

-def _RunInstrumentationTests(args, devices):

- """Subcommand of RunTestsCommands which runs instrumentation tests."""

- logging.info('_RunInstrumentationTests(%s, %s)' % (str(args), str(devices)))

- instrumentation_options = ProcessInstrumentationOptions(args)

- if len(devices) > 1 and args.wait_for_debugger:

- logging.warning('Debugger can not be sharded, using first available device')

- devices = devices[:1]

- results = base_test_result.TestRunResults()

- exit_code = 0

- if args.run_java_tests:

- runner_factory, tests = instrumentation_setup.Setup(

- instrumentation_options, devices)

- test_results, exit_code = test_dispatcher.RunTests(

- tests, runner_factory, devices, shard=True, test_timeout=None,

- num_retries=args.num_retries)

- results.AddTestRunResults(test_results)

- if args.run_python_tests:

- runner_factory, tests = host_driven_setup.InstrumentationSetup(

- args.host_driven_root, args.official_build,

- instrumentation_options)

- if tests:

- test_results, test_exit_code = test_dispatcher.RunTests(

- tests, runner_factory, devices, shard=True, test_timeout=None,

- num_retries=args.num_retries)

- results.AddTestRunResults(test_results)

- # Only allow exit code escalation

- if test_exit_code and exit_code != constants.ERROR_EXIT_CODE:

- exit_code = test_exit_code

- if args.device_flags:

- args.device_flags = os.path.join(constants.DIR_SOURCE_ROOT,

- args.device_flags)

- report_results.LogFull(

- results=results,

- test_type='Instrumentation',

- test_package=os.path.basename(args.test_apk),

- annotation=args.annotations,

- flakiness_server=args.flakiness_dashboard_server)

- if args.json_results_file:

- json_results.GenerateJsonResultsFile(results, args.json_results_file)

- return exit_code

-def _RunUIAutomatorTests(args, devices):

- """Subcommand of RunTestsCommands which runs uiautomator tests."""

- uiautomator_options = ProcessUIAutomatorOptions(args)

- runner_factory, tests = uiautomator_setup.Setup(uiautomator_options)

- results, exit_code = test_dispatcher.RunTests(

- tests, runner_factory, devices, shard=True, test_timeout=None,

- num_retries=args.num_retries)

- report_results.LogFull(

- results=results,

- test_type='UIAutomator',

- test_package=os.path.basename(args.test_jar),

- annotation=args.annotations,

- flakiness_server=args.flakiness_dashboard_server)

- if args.json_results_file:

- json_results.GenerateJsonResultsFile(results, args.json_results_file)

- return exit_code

-def _RunJUnitTests(args):

- """Subcommand of RunTestsCommand which runs junit tests."""

- runner_factory, tests = junit_setup.Setup(args)

- results, exit_code = junit_dispatcher.RunTests(tests, runner_factory)

- report_results.LogFull(

- results=results,

- test_type='JUnit',

- test_package=args.test_suite)

- if args.json_results_file:

- json_results.GenerateJsonResultsFile(results, args.json_results_file)

- return exit_code

-def _RunMonkeyTests(args, devices):

- """Subcommand of RunTestsCommands which runs monkey tests."""

- monkey_options = ProcessMonkeyTestOptions(args)

- runner_factory, tests = monkey_setup.Setup(monkey_options)

- results, exit_code = test_dispatcher.RunTests(

- tests, runner_factory, devices, shard=False, test_timeout=None,

- num_retries=args.num_retries)

- report_results.LogFull(

- results=results,

- test_type='Monkey',

- test_package='Monkey')

- if args.json_results_file:

- json_results.GenerateJsonResultsFile(results, args.json_results_file)

- return exit_code

-def _RunPerfTests(args):

- """Subcommand of RunTestsCommands which runs perf tests."""

- perf_options = ProcessPerfTestOptions(args)

- # Just save a simple json with a list of test names.

- if perf_options.output_json_list:

- return perf_test_runner.OutputJsonList(

- perf_options.steps, perf_options.output_json_list)

- # Just print the results from a single previously executed step.

- if perf_options.print_step:

- return perf_test_runner.PrintTestOutput(

- perf_options.print_step, perf_options.output_chartjson_data)

- runner_factory, tests, devices = perf_setup.Setup(perf_options)

- # shard=False means that each device will get the full list of tests

- # and then each one will decide their own affinity.

- # shard=True means each device will pop the next test available from a queue,

- # which increases throughput but have no affinity.

- results, _ = test_dispatcher.RunTests(

- tests, runner_factory, devices, shard=False, test_timeout=None,

- num_retries=args.num_retries)

- report_results.LogFull(

- results=results,

- test_type='Perf',

- test_package='Perf')

- if args.json_results_file:

- json_results.GenerateJsonResultsFile(results, args.json_results_file)

- if perf_options.single_step:

- return perf_test_runner.PrintTestOutput('single_step')

- perf_test_runner.PrintSummary(tests)

- # Always return 0 on the sharding stage. Individual tests exit_code

- # will be returned on the print_step stage.

- return 0

-def _RunPythonTests(args):

- """Subcommand of RunTestsCommand which runs python unit tests."""

- suite_vars = constants.PYTHON_UNIT_TEST_SUITES[args.suite_name]

- suite_path = suite_vars['path']

- suite_test_modules = suite_vars['test_modules']

- sys.path = [suite_path] + sys.path

- try:

- suite = unittest.TestSuite()

- suite.addTests(unittest.defaultTestLoader.loadTestsFromName(m)

- for m in suite_test_modules)

- runner = unittest.TextTestRunner(verbosity=1+args.verbose_count)

- return 0 if runner.run(suite).wasSuccessful() else 1

- finally:

- sys.path = sys.path[1:]

-def _GetAttachedDevices(test_device=None):

- """Get all attached devices.

- Args:

- test_device: Name of a specific device to use.

- Returns:

- A list of attached devices.

- """

- attached_devices = device_utils.DeviceUtils.HealthyDevices()

- if test_device:

- test_device = [d for d in attached_devices if d == test_device]

- if not test_device:

- raise device_errors.DeviceUnreachableError(

- 'Did not find device %s among attached device. Attached devices: %s'

- % (test_device, ', '.join(attached_devices)))

- return test_device

- else:

- if not attached_devices:

- raise device_errors.NoDevicesError()

- return sorted(attached_devices)

-def RunTestsCommand(args, parser):

- """Checks test type and dispatches to the appropriate function.

- Args:

- args: argparse.Namespace object.

- parser: argparse.ArgumentParser object.

- Returns:

- Integer indicated exit code.

- Raises:

- Exception: Unknown command name passed in, or an exception from an

- individual test runner.

- """

- command = args.command

- ProcessCommonOptions(args)

- if args.enable_platform_mode:

- return RunTestsInPlatformMode(args, parser)

- if command in constants.LOCAL_MACHINE_TESTS:

- devices = []

- else:

- devices = _GetAttachedDevices(args.test_device)

- forwarder.Forwarder.RemoveHostLog()

- if not ports.ResetTestServerPortAllocation():

- raise Exception('Failed to reset test server port.')

- if command == 'gtest':

- if args.suite_name[0] in gtest_test_instance.BROWSER_TEST_SUITES:

- return RunTestsInPlatformMode(args, parser)

- return _RunGTests(args, devices)

- elif command == 'linker':

- return _RunLinkerTests(args, devices)

- elif command == 'instrumentation':

- return _RunInstrumentationTests(args, devices)

- elif command == 'uiautomator':

- return _RunUIAutomatorTests(args, devices)

- elif command == 'junit':

- return _RunJUnitTests(args)

- elif command == 'monkey':

- return _RunMonkeyTests(args, devices)

- elif command == 'perf':

- return _RunPerfTests(args)

- elif command == 'python':

- return _RunPythonTests(args)

- else:

- raise Exception('Unknown test type.')

-_SUPPORTED_IN_PLATFORM_MODE = [

- # TODO(jbudorick): Add support for more test types.

- 'gtest',

- 'instrumentation',

- 'uirobot',

-def RunTestsInPlatformMode(args, parser):

- if args.command not in _SUPPORTED_IN_PLATFORM_MODE:

- parser.error('%s is not yet supported in platform mode' % args.command)

- with environment_factory.CreateEnvironment(args, parser.error) as env:

- with test_instance_factory.CreateTestInstance(args, parser.error) as test:

- with test_run_factory.CreateTestRun(

- args, env, test, parser.error) as test_run:

- results = test_run.RunTests()

- if args.environment == 'remote_device' and args.trigger:

- return 0 # Not returning results, only triggering.

- report_results.LogFull(

- results=results,

- test_type=test.TestType(),

- test_package=test_run.TestPackage(),

- annotation=getattr(args, 'annotations', None),

- flakiness_server=getattr(args, 'flakiness_dashboard_server', None))

- if args.json_results_file:

- json_results.GenerateJsonResultsFile(

- results, args.json_results_file)

- return 0 if results.DidRunPass() else constants.ERROR_EXIT_CODE

-CommandConfigTuple = collections.namedtuple(

- 'CommandConfigTuple',

- ['add_options_func', 'help_txt'])

-VALID_COMMANDS = {

- 'gtest': CommandConfigTuple(

- AddGTestOptions,

- 'googletest-based C++ tests'),

- 'instrumentation': CommandConfigTuple(

- AddInstrumentationTestOptions,

- 'InstrumentationTestCase-based Java tests'),

- 'uiautomator': CommandConfigTuple(

- AddUIAutomatorTestOptions,

- "Tests that run via Android's uiautomator command"),

- 'junit': CommandConfigTuple(

- AddJUnitTestOptions,

- 'JUnit4-based Java tests'),

- 'monkey': CommandConfigTuple(

- AddMonkeyTestOptions,

- "Tests based on Android's monkey"),

- 'perf': CommandConfigTuple(

- AddPerfTestOptions,

- 'Performance tests'),

- 'python': CommandConfigTuple(

- AddPythonTestOptions,

- 'Python tests based on unittest.TestCase'),

- 'linker': CommandConfigTuple(

- AddLinkerTestOptions,

- 'Linker tests'),

- 'uirobot': CommandConfigTuple(

- AddUirobotTestOptions,

- 'Uirobot test'),

-def DumpThreadStacks(_signal, _frame):

- for thread in threading.enumerate():

- reraiser_thread.LogThreadStack(thread)

-def main():

- signal.signal(signal.SIGUSR1, DumpThreadStacks)

- parser = argparse.ArgumentParser()

- command_parsers = parser.add_subparsers(title='test types',

- dest='command')

- for test_type, config in sorted(VALID_COMMANDS.iteritems(),

- key=lambda x: x[0]):

- subparser = command_parsers.add_parser(

- test_type, usage='%(prog)s [options]', help=config.help_txt)

- config.add_options_func(subparser)

- args = parser.parse_args()

- try:

- return RunTestsCommand(args, parser)

- except base_error.BaseError as e:

- logging.exception('Error occurred.')

- if e.is_infra_error:

- return constants.INFRA_EXIT_CODE

- return constants.ERROR_EXIT_CODE

- except: # pylint: disable=W0702

- logging.exception('Unrecognized error occurred.')

- return constants.ERROR_EXIT_CODE

-if __name__ == '__main__':

- sys.exit(main())

« no previous file with comments | « build/android/test_runner.gypi ('k') | build/android/tests/symbolize/Makefile » ('j') | no next file with comments »