| Index: build/android/pylib/instrumentation/setup.py
|
| diff --git a/build/android/pylib/instrumentation/setup.py b/build/android/pylib/instrumentation/setup.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..7a0501e5b5c302075c0a4f67e6cc65969d8de9fb
|
| --- /dev/null
|
| +++ b/build/android/pylib/instrumentation/setup.py
|
| @@ -0,0 +1,113 @@
|
| +# Copyright 2013 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +"""Generates test runner factory and tests for instrumentation tests."""
|
| +
|
| +import logging
|
| +import os
|
| +
|
| +from pylib import constants
|
| +from pylib import valgrind_tools
|
| +
|
| +from pylib.base import base_setup
|
| +from pylib.device import device_utils
|
| +from pylib.instrumentation import test_package
|
| +from pylib.instrumentation import test_runner
|
| +
|
| +DEVICE_DATA_DIR = 'chrome/test/data'
|
| +
|
| +ISOLATE_FILE_PATHS = {
|
| + 'AndroidWebViewTest': 'android_webview/android_webview_test_apk.isolate',
|
| + 'ChromeShellTest': 'chrome/chrome_shell_test_apk.isolate',
|
| + 'ContentShellTest': 'content/content_shell_test_apk.isolate',
|
| +}
|
| +
|
| +DEPS_EXCLUSION_LIST = []
|
| +
|
| +# TODO(mikecase): Remove this function and the constant DEVICE_DATA_DIR
|
| +# once all data deps are pushed to the same location on the device.
|
| +def _PushExtraSuiteDataDeps(device, test_apk):
|
| + """Pushes some extra data files/dirs needed by some test suite.
|
| +
|
| + Args:
|
| + test_apk: The test suite basename for which to return file paths.
|
| + """
|
| + if test_apk in ['ChromeTest', 'ContentShellTest']:
|
| + test_files = 'net/data/ssl/certificates'
|
| + host_device_file_tuple = [
|
| + (os.path.join(constants.DIR_SOURCE_ROOT, test_files),
|
| + os.path.join(device.GetExternalStoragePath(), test_files))]
|
| + device.PushChangedFiles(host_device_file_tuple)
|
| +
|
| +
|
| +# TODO(mikecase): Remove this function once everything uses
|
| +# base_setup.PushDataDeps to push data deps to the device.
|
| +def _PushDataDeps(device, test_options):
|
| + valgrind_tools.PushFilesForTool(test_options.tool, device)
|
| +
|
| + host_device_file_tuples = []
|
| + for dest_host_pair in test_options.test_data:
|
| + dst_src = dest_host_pair.split(':', 1)
|
| + dst_layer = dst_src[0]
|
| + host_src = dst_src[1]
|
| + host_test_files_path = os.path.join(constants.DIR_SOURCE_ROOT, host_src)
|
| + if os.path.exists(host_test_files_path):
|
| + host_device_file_tuples += [(
|
| + host_test_files_path,
|
| + '%s/%s/%s' % (
|
| + device.GetExternalStoragePath(),
|
| + DEVICE_DATA_DIR,
|
| + dst_layer))]
|
| + if host_device_file_tuples:
|
| + device.PushChangedFiles(host_device_file_tuples)
|
| +
|
| +
|
| +def Setup(test_options, devices):
|
| + """Create and return the test runner factory and tests.
|
| +
|
| + Args:
|
| + test_options: An InstrumentationOptions object.
|
| +
|
| + Returns:
|
| + A tuple of (TestRunnerFactory, tests).
|
| + """
|
| + if (test_options.coverage_dir and not
|
| + os.path.exists(test_options.coverage_dir)):
|
| + os.makedirs(test_options.coverage_dir)
|
| +
|
| + test_pkg = test_package.TestPackage(test_options.test_apk_path,
|
| + test_options.test_apk_jar_path,
|
| + test_options.test_support_apk_path)
|
| + tests = test_pkg.GetAllMatchingTests(
|
| + test_options.annotations,
|
| + test_options.exclude_annotations,
|
| + test_options.test_filter)
|
| + if not tests:
|
| + logging.error('No instrumentation tests to run with current args.')
|
| +
|
| + if test_options.test_data:
|
| + device_utils.DeviceUtils.parallel(devices).pMap(
|
| + _PushDataDeps, test_options)
|
| +
|
| + if test_options.isolate_file_path:
|
| + i = base_setup.GenerateDepsDirUsingIsolate(test_options.test_apk,
|
| + test_options.isolate_file_path,
|
| + ISOLATE_FILE_PATHS,
|
| + DEPS_EXCLUSION_LIST)
|
| + def push_data_deps_to_device_dir(device):
|
| + base_setup.PushDataDeps(device, device.GetExternalStoragePath(),
|
| + test_options)
|
| + device_utils.DeviceUtils.parallel(devices).pMap(
|
| + push_data_deps_to_device_dir)
|
| + if i:
|
| + i.Clear()
|
| +
|
| + device_utils.DeviceUtils.parallel(devices).pMap(
|
| + _PushExtraSuiteDataDeps, test_options.test_apk)
|
| +
|
| + def TestRunnerFactory(device, shard_index):
|
| + return test_runner.TestRunner(test_options, device, shard_index,
|
| + test_pkg)
|
| +
|
| + return (TestRunnerFactory, tests)
|
|
|