OLD | NEW |
1 # Copyright 2013 The Chromium Authors. All rights reserved. | 1 # Copyright 2013 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 """Generates test runner factory and tests for instrumentation tests.""" | 5 """Generates test runner factory and tests for instrumentation tests.""" |
6 | 6 |
7 import logging | 7 import logging |
8 import os | 8 import os |
9 | 9 |
10 from pylib import android_commands | 10 from pylib import android_commands |
11 from pylib import constants | 11 from pylib import constants |
12 from pylib.base import base_test_result | 12 from pylib.base import base_test_result |
13 from pylib.utils import report_results | 13 from pylib.utils import report_results |
14 | 14 |
15 import test_package | 15 import test_package |
16 import test_runner | 16 import test_runner |
17 | 17 |
18 | 18 |
19 def Setup(test_apk_path, test_apk_jar_path, annotations, exclude_annotations, | 19 def Setup(test_apk_path, test_apk_jar_path, annotations, exclude_annotations, |
20 test_filter, build_type, test_data, install_apk, save_perf_json, | 20 test_filter, build_type, test_data, save_perf_json, |
21 screenshot_failures, tool, wait_for_debugger, disable_assertions, | 21 screenshot_failures, tool, wait_for_debugger, disable_assertions, |
22 push_deps, cleanup_test_files): | 22 push_deps, cleanup_test_files): |
23 """Create and return the test runner factory and tests. | 23 """Create and return the test runner factory and tests. |
24 | 24 |
25 Args: | 25 Args: |
26 test_apk_path: Path to the test apk file. | 26 test_apk_path: Path to the test apk file. |
27 test_apk_jar_path: Path to the jar associated with the test apk. | 27 test_apk_jar_path: Path to the jar associated with the test apk. |
28 annotations: Annotations for the tests. | 28 annotations: Annotations for the tests. |
29 exclude_annotations: Any annotations to exclude from running. | 29 exclude_annotations: Any annotations to exclude from running. |
30 test_filter: Filter string for tests. | 30 test_filter: Filter string for tests. |
31 build_type: 'Release' or 'Debug'. | 31 build_type: 'Release' or 'Debug'. |
32 test_data: Location of the test data. | 32 test_data: Location of the test data. |
33 install_apk: Re-installs the apk if opted. | |
34 save_perf_json: Whether or not to save the JSON file from UI perf tests. | 33 save_perf_json: Whether or not to save the JSON file from UI perf tests. |
35 screenshot_failures: Take a screenshot for a test failure | 34 screenshot_failures: Take a screenshot for a test failure |
36 tool: Name of the Valgrind tool. | 35 tool: Name of the Valgrind tool. |
37 wait_for_debugger: blocks until the debugger is connected. | 36 wait_for_debugger: blocks until the debugger is connected. |
38 disable_assertions: Whether to disable java assertions on the device. | 37 disable_assertions: Whether to disable java assertions on the device. |
39 push_deps: If True, push all dependencies to the device. | 38 push_deps: If True, push all dependencies to the device. |
40 cleanup_test_files: Whether or not to cleanup test files on device. | 39 cleanup_test_files: Whether or not to cleanup test files on device. |
41 | 40 |
42 Returns: | 41 Returns: |
43 A tuple of (TestRunnerFactory, tests). | 42 A tuple of (TestRunnerFactory, tests). |
44 """ | 43 """ |
45 test_pkg = test_package.TestPackage(test_apk_path, test_apk_jar_path) | 44 test_pkg = test_package.TestPackage(test_apk_path, test_apk_jar_path) |
46 tests = test_pkg._GetAllMatchingTests(annotations, exclude_annotations, | 45 tests = test_pkg._GetAllMatchingTests(annotations, exclude_annotations, |
47 test_filter) | 46 test_filter) |
48 if not tests: | 47 if not tests: |
49 logging.error('No instrumentation tests to run with current args.') | 48 logging.error('No instrumentation tests to run with current args.') |
50 | 49 |
51 def TestRunnerFactory(device, shard_index): | 50 def TestRunnerFactory(device, shard_index): |
52 return test_runner.TestRunner( | 51 return test_runner.TestRunner( |
53 build_type, test_data, install_apk, save_perf_json, screenshot_failures, | 52 build_type, test_data, save_perf_json, screenshot_failures, |
54 tool, wait_for_debugger, disable_assertions, push_deps, | 53 tool, wait_for_debugger, disable_assertions, push_deps, |
55 cleanup_test_files, device, shard_index, test_pkg, []) | 54 cleanup_test_files, device, shard_index, test_pkg, []) |
56 | 55 |
57 return (TestRunnerFactory, tests) | 56 return (TestRunnerFactory, tests) |
OLD | NEW |