| #!/usr/bin/env python |
| # |
| # Copyright 2016 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. |
| |
| """Runs the CTS test APKs stored in GS.""" |
| |
| import argparse |
| import json |
| import os |
| import shutil |
| import sys |
| import tempfile |
| |
| |
| sys.path.append(os.path.join( |
| os.path.dirname(__file__), os.pardir, os.pardir, 'build', 'android')) |
| import devil_chromium # pylint: disable=import-error |
| from devil.utils import cmd_helper # pylint: disable=import-error |
| |
| sys.path.append(os.path.join( |
| os.path.dirname(__file__), os.pardir, os.pardir, 'build')) |
| import find_depot_tools # pylint: disable=import-error |
| |
| _CTS_BUCKET = 'gs://chromium-cts' |
| |
| _GSUTIL_PATH = os.path.join(find_depot_tools.DEPOT_TOOLS_PATH, 'gsutil.py') |
| _TEST_RUNNER_PATH = os.path.join( |
| os.path.dirname(__file__), os.pardir, os.pardir, |
| 'build', 'android', 'test_runner.py') |
| |
| _EXPECTED_FAILURES_FILE = os.path.join( |
| os.path.dirname(__file__), 'cts_config', 'expected_failure_on_bot.json') |
| _WEBVIEW_CTS_GCS_PATH_FILE = os.path.join( |
| os.path.dirname(__file__), 'cts_config', 'webview_cts_gcs_path.json') |
| |
| |
| def GetCtsPath(arch, platform): |
| """Gets relative path the CTS APK is stored at.""" |
| with open(_WEBVIEW_CTS_GCS_PATH_FILE) as f: |
| cts_gcs_path_info = json.load(f) |
| try: |
| return cts_gcs_path_info[arch][platform]['apk'] |
| except KeyError: |
| raise Exception('No CTS test available for arch:%s, android:%s' % |
| (arch, platform)) |
| |
| |
| def GetExpectedFailures(): |
| """Gets list of tests expected to fail.""" |
| with open(_EXPECTED_FAILURES_FILE) as f: |
| expected_failures_info = json.load(f) |
| expected_failures = [] |
| for class_name, methods in expected_failures_info.iteritems(): |
| expected_failures.extend(['%s#%s' % (class_name, m['name']) |
| for m in methods]) |
| return expected_failures |
| |
| |
| def DownloadAndRunCTS(args, test_runner_args): |
| base_cts_dir = None |
| delete_cts_dir = False |
| try: |
| relative_cts_path = GetCtsPath(args.arch, args.platform) |
| |
| if args.apk_dir: |
| base_cts_dir = args.apk_dir |
| else: |
| base_cts_dir = tempfile.mkdtemp() |
| delete_cts_dir = True |
| |
| local_cts_path = os.path.join(base_cts_dir, relative_cts_path) |
| google_storage_cts_path = '%s/%s' % (_CTS_BUCKET, relative_cts_path) |
| |
| # Download CTS APK if needed. |
| if not os.path.exists(local_cts_path): |
| if cmd_helper.RunCmd( |
| [_GSUTIL_PATH, 'cp', google_storage_cts_path, local_cts_path]): |
| raise Exception('Error downloading CTS from Google Storage.') |
| |
| test_runner_args += ['--test-apk', local_cts_path] |
| # TODO(mikecase): This doesn't work at all with the |
| # --gtest-filter test runner option currently. The |
| # filter options will just override eachother. |
| if args.skip_expected_failures: |
| test_runner_args += ['-f=-%s' % ':'.join(GetExpectedFailures())] |
| return cmd_helper.RunCmd( |
| [_TEST_RUNNER_PATH, 'instrumentation'] + test_runner_args) |
| finally: |
| if delete_cts_dir and base_cts_dir: |
| shutil.rmtree(base_cts_dir) |
| |
| |
| def main(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument( |
| '--arch', |
| choices=['arm64'], |
| default='arm64', |
| help='Arch for CTS tests.') |
| parser.add_argument( |
| '--platform', |
| choices=['L', 'M', 'N'], |
| required=True, |
| help='Android platform version for CTS tests.') |
| parser.add_argument( |
| '--skip-expected-failures', |
| action='store_true', |
| help='Option to skip all tests that are expected to fail.') |
| parser.add_argument( |
| '--apk-dir', |
| help='Directory to load/save CTS APKs. Will try to load CTS APK ' |
| 'from this directory before downloading from Google Storage ' |
| 'and will then cache APK here.') |
| |
| args, test_runner_args = parser.parse_known_args() |
| devil_chromium.Initialize() |
| |
| return DownloadAndRunCTS(args, test_runner_args) |
| |
| |
| if __name__ == '__main__': |
| sys.exit(main()) |