| Index: build/android/pylib/uiautomator/test_runner.py
|
| diff --git a/build/android/pylib/uiautomator/test_runner.py b/build/android/pylib/uiautomator/test_runner.py
|
| index 309fa1f034ffb84440ebeced2b293e7c4b4662ad..39ed5ec829ad43f2829474a2f8e1393d9a800476 100644
|
| --- a/build/android/pylib/uiautomator/test_runner.py
|
| +++ b/build/android/pylib/uiautomator/test_runner.py
|
| @@ -4,32 +4,48 @@
|
|
|
| """Class for running uiautomator tests on a single device."""
|
|
|
| +from pylib.instrumentation import test_options
|
| from pylib.instrumentation import test_runner as instr_test_runner
|
|
|
|
|
| class TestRunner(instr_test_runner.TestRunner):
|
| """Responsible for running a series of tests connected to a single device."""
|
|
|
| - def __init__(self, package_name, build_type, test_data, save_perf_json,
|
| - screenshot_failures, tool, wait_for_debugger,
|
| - disable_assertions, push_deps, cleanup_test_files, device,
|
| - shard_index, test_pkg, ports_to_forward):
|
| + def __init__(self, test_options, device, shard_index, test_pkg,
|
| + ports_to_forward):
|
| """Create a new TestRunner.
|
|
|
| Args:
|
| - package_name: Application package name under test.
|
| - See the super class for all other args.
|
| + test_options: A UIAutomatorOptions object.
|
| + device: Attached android device.
|
| + shard_index: Shard index.
|
| + test_pkg: A TestPackage object.
|
| + ports_to_forward: A list of port numbers for which to set up forwarders.
|
| + Can be optionally requested by a test case.
|
| """
|
| - super(TestRunner, self).__init__(
|
| - build_type, test_data, save_perf_json, screenshot_failures, tool,
|
| - wait_for_debugger, disable_assertions, push_deps, cleanup_test_files,
|
| - device, shard_index, test_pkg, ports_to_forward)
|
| -
|
| - self.package_name = package_name
|
| + # Create an InstrumentationOptions object to pass to the super class
|
| + instrumentation_options = test_options.InstrumentationOptions(
|
| + test_options.build_type,
|
| + test_options.tool,
|
| + test_options.cleanup_test_files,
|
| + test_options.push_deps,
|
| + test_options.annotations,
|
| + test_options.exclude_annotations,
|
| + test_options.test_filter,
|
| + test_options.test_data,
|
| + test_options.save_perf_json,
|
| + test_options.screenshot_failures,
|
| + test_options.disable_assertions,
|
| + wait_for_debugger=False,
|
| + test_apk=None)
|
| + super(TestRunner, self).__init__(instrumentation_options, device,
|
| + shard_index, test_pkg, ports_to_forward)
|
| +
|
| + self.package_name = test_options.package_name
|
|
|
| #override
|
| def InstallTestPackage(self):
|
| - self.test_pkg.Install(self.adb)
|
| + self.test_pkg.Install(self.adb)
|
|
|
| #override
|
| def PushDataDeps(self):
|
|
|