| Index: build/android/pylib/local/device/local_device_test_run.py
|
| diff --git a/build/android/pylib/local/device/local_device_test_run.py b/build/android/pylib/local/device/local_device_test_run.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..e79b3880a51e5322aa6eac559b8ba567b5a17bbf
|
| --- /dev/null
|
| +++ b/build/android/pylib/local/device/local_device_test_run.py
|
| @@ -0,0 +1,85 @@
|
| +# Copyright 2014 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.
|
| +
|
| +
|
| +from pylib import valgrind_tools
|
| +from pylib.base import base_test_result
|
| +from pylib.base import test_run
|
| +from pylib.base import test_collection
|
| +
|
| +
|
| +class LocalDeviceTestRun(test_run.TestRun):
|
| +
|
| + def __init__(self, env, test_instance):
|
| + super(LocalDeviceTestRun, self).__init__(env, test_instance)
|
| + self._tools = {}
|
| +
|
| + #override
|
| + def RunTests(self):
|
| + tests = self._GetTests()
|
| +
|
| + def run_tests_on_device(dev, tests):
|
| + r = base_test_result.TestRunResults()
|
| + for test in tests:
|
| + result = self._RunTest(dev, test)
|
| + if isinstance(result, base_test_result.BaseTestResult):
|
| + r.AddResult(result)
|
| + elif isinstance(result, list):
|
| + r.AddResults(result)
|
| + else:
|
| + raise Exception('Unexpected result type: %s' % type(result).__name__)
|
| + if isinstance(tests, test_collection.TestCollection):
|
| + tests.test_completed()
|
| + return r
|
| +
|
| + tries = 0
|
| + results = base_test_result.TestRunResults()
|
| + fail_results = []
|
| + while tries < self._env.max_tries and tests:
|
| + if self._ShouldShard():
|
| + tc = test_collection.TestCollection(self._CreateShards(tests))
|
| + try_results = self._env.parallel_devices.pMap(
|
| + run_tests_on_device, tc).pGet(None)
|
| + else:
|
| + try_results = self._env.parallel_devices.pMap(
|
| + run_tests_on_device, tests).pGet(None)
|
| + fail_results = []
|
| + for try_result in try_results:
|
| + for result in try_result.GetAll():
|
| + if result.GetType() in (base_test_result.ResultType.PASS,
|
| + base_test_result.ResultType.SKIP):
|
| + results.AddResult(result)
|
| + else:
|
| + fail_results.append(result)
|
| +
|
| + results_names = set(r.GetName() for r in results.GetAll())
|
| + tests = [t for t in tests if t not in results_names]
|
| + tries += 1
|
| +
|
| + if tests:
|
| + results.AddResults(
|
| + base_test_result.BaseTestResult(
|
| + t, base_test_result.ResultType.UNKNOWN)
|
| + for t in tests)
|
| + if fail_results:
|
| + results.AddResults(fail_results)
|
| + return results
|
| +
|
| + def GetTool(self, device):
|
| + if not str(device) in self._tools:
|
| + self._tools[str(device)] = valgrind_tools.CreateTool(
|
| + self._env.tool, device)
|
| + return self._tools[str(device)]
|
| +
|
| + def _CreateShards(self, tests):
|
| + raise NotImplementedError
|
| +
|
| + def _GetTests(self):
|
| + raise NotImplementedError
|
| +
|
| + def _RunTest(self, device, test):
|
| + raise NotImplementedError
|
| +
|
| + def _ShouldShard(self):
|
| + raise NotImplementedError
|
|
|