| 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
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..bda668778897e57188937e91e4b6a380e42d268b
|
| --- /dev/null
|
| +++ b/build/android/pylib/uiautomator/test_runner.py
|
| @@ -0,0 +1,89 @@
|
| +# Copyright (c) 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.
|
| +
|
| +"""Class for running uiautomator tests on a single device."""
|
| +
|
| +from pylib import constants
|
| +from pylib import flag_changer
|
| +from pylib.device import intent
|
| +from pylib.instrumentation import test_options as instr_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, test_options, device, shard_index, test_pkg):
|
| + """Create a new TestRunner.
|
| +
|
| + Args:
|
| + test_options: A UIAutomatorOptions object.
|
| + device: Attached android device.
|
| + shard_index: Shard index.
|
| + test_pkg: A TestPackage object.
|
| + """
|
| + # Create an InstrumentationOptions object to pass to the super class
|
| + instrumentation_options = instr_test_options.InstrumentationOptions(
|
| + test_options.tool,
|
| + test_options.annotations,
|
| + test_options.exclude_annotations,
|
| + test_options.test_filter,
|
| + test_options.test_data,
|
| + test_options.save_perf_json,
|
| + test_options.screenshot_failures,
|
| + wait_for_debugger=False,
|
| + coverage_dir=None,
|
| + test_apk=None,
|
| + test_apk_path=None,
|
| + test_apk_jar_path=None,
|
| + test_runner=None,
|
| + test_support_apk_path=None,
|
| + device_flags=None,
|
| + isolate_file_path=None,
|
| + set_asserts=test_options.set_asserts,
|
| + delete_stale_data=False)
|
| + super(TestRunner, self).__init__(instrumentation_options, device,
|
| + shard_index, test_pkg)
|
| +
|
| + cmdline_file = constants.PACKAGE_INFO[test_options.package].cmdline_file
|
| + self.flags = None
|
| + if cmdline_file:
|
| + self.flags = flag_changer.FlagChanger(self.device, cmdline_file)
|
| + self._package = constants.PACKAGE_INFO[test_options.package].package
|
| + self._activity = constants.PACKAGE_INFO[test_options.package].activity
|
| +
|
| + #override
|
| + def InstallTestPackage(self):
|
| + self.test_pkg.Install(self.device)
|
| +
|
| + #override
|
| + def _RunTest(self, test, timeout):
|
| + self.device.ClearApplicationState(self._package)
|
| + if self.flags:
|
| + annotations = self.test_pkg.GetTestAnnotations(test)
|
| + if 'FirstRunExperience' == annotations.get('Feature', None):
|
| + self.flags.RemoveFlags(['--disable-fre'])
|
| + else:
|
| + self.flags.AddFlags(['--disable-fre'])
|
| + self.device.StartActivity(
|
| + intent.Intent(action='android.intent.action.MAIN',
|
| + activity=self._activity,
|
| + package=self._package),
|
| + blocking=True,
|
| + force_stop=True)
|
| + cmd = ['uiautomator', 'runtest',
|
| + self.test_pkg.UIAUTOMATOR_PATH + self.test_pkg.GetPackageName(),
|
| + '-e', 'class', test,
|
| + '-e', 'test_package', self._package]
|
| + return self.device.RunShellCommand(cmd, timeout=timeout, retries=0)
|
| +
|
| + #override
|
| + def _GenerateTestResult(self, test, _result_code, _result_bundle, statuses,
|
| + start_ms, duration_ms):
|
| + # uiautomator emits its summary status with INSTRUMENTATION_STATUS_CODE,
|
| + # not INSTRUMENTATION_CODE, so we have to drop if off the list of statuses.
|
| + summary_code, summary_bundle = statuses[-1]
|
| + return super(TestRunner, self)._GenerateTestResult(
|
| + test, summary_code, summary_bundle, statuses[:-1], start_ms,
|
| + duration_ms)
|
|
|