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 performance tests.""" | 5 """Generates test runner factory and tests for performance tests.""" |
6 | 6 |
7 import json | 7 import json |
8 import fnmatch | 8 import fnmatch |
9 import logging | 9 import logging |
10 import os | 10 import os |
11 import shutil | 11 import shutil |
12 | 12 |
13 from pylib import android_commands | |
14 from pylib import constants | 13 from pylib import constants |
15 from pylib import forwarder | 14 from pylib import forwarder |
16 from pylib.device import device_list | 15 from pylib.device import device_list |
| 16 from pylib.device import device_utils |
17 from pylib.perf import test_runner | 17 from pylib.perf import test_runner |
18 from pylib.utils import test_environment | 18 from pylib.utils import test_environment |
19 | 19 |
20 | 20 |
21 def _GetAllDevices(): | 21 def _GetAllDevices(): |
22 devices_path = os.path.join(os.environ.get('CHROMIUM_OUT_DIR', 'out'), | 22 devices_path = os.path.join(os.environ.get('CHROMIUM_OUT_DIR', 'out'), |
23 device_list.LAST_DEVICES_FILENAME) | 23 device_list.LAST_DEVICES_FILENAME) |
24 try: | 24 try: |
25 devices = device_list.GetPersistentDeviceList(devices_path) | 25 devices = [device_utils.DeviceUtils(s) |
| 26 for s in device_list.GetPersistentDeviceList(devices_path)] |
26 except IOError as e: | 27 except IOError as e: |
27 logging.error('Unable to find %s [%s]', devices_path, e) | 28 logging.error('Unable to find %s [%s]', devices_path, e) |
28 devices = android_commands.GetAttachedDevices() | 29 devices = device_utils.DeviceUtils.HealthyDevices() |
29 return sorted(devices) | 30 return sorted(devices) |
30 | 31 |
31 | 32 |
32 def _GetStepsDictFromSingleStep(test_options): | 33 def _GetStepsDictFromSingleStep(test_options): |
33 # Running a single command, build the tests structure. | 34 # Running a single command, build the tests structure. |
34 steps_dict = { | 35 steps_dict = { |
35 'version': 1, | 36 'version': 1, |
36 'steps': { | 37 'steps': { |
37 'single_step': { | 38 'single_step': { |
38 'device_affinity': 0, | 39 'device_affinity': 0, |
(...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
87 if test_options.flaky_steps: | 88 if test_options.flaky_steps: |
88 with file(test_options.flaky_steps, 'r') as f: | 89 with file(test_options.flaky_steps, 'r') as f: |
89 flaky_steps = json.load(f) | 90 flaky_steps = json.load(f) |
90 | 91 |
91 def TestRunnerFactory(device, shard_index): | 92 def TestRunnerFactory(device, shard_index): |
92 return test_runner.TestRunner( | 93 return test_runner.TestRunner( |
93 test_options, device, shard_index, len(all_devices), | 94 test_options, device, shard_index, len(all_devices), |
94 steps_dict, flaky_steps) | 95 steps_dict, flaky_steps) |
95 | 96 |
96 return (TestRunnerFactory, sorted_step_names, all_devices) | 97 return (TestRunnerFactory, sorted_step_names, all_devices) |
OLD | NEW |