Chromium Code Reviews| Index: build/android/test_runner.py |
| diff --git a/build/android/test_runner.py b/build/android/test_runner.py |
| index c54ed28d324054eae3c5aee72aa1442d80bf1f36..ebd5b39928bee12c2ffcc89a3b852f77c9329e2e 100755 |
| --- a/build/android/test_runner.py |
| +++ b/build/android/test_runner.py |
| @@ -16,7 +16,6 @@ import sys |
| import threading |
| import unittest |
| -from pylib import android_commands |
| from pylib import constants |
| from pylib import forwarder |
| from pylib import ports |
| @@ -25,6 +24,8 @@ from pylib.base import environment_factory |
| from pylib.base import test_dispatcher |
| from pylib.base import test_instance_factory |
| from pylib.base import test_run_factory |
| +from pylib.device import device_errors |
| +from pylib.device import device_utils |
| from pylib.gtest import gtest_config |
| from pylib.gtest import setup as gtest_setup |
| from pylib.gtest import test_options as gtest_test_options |
| @@ -867,18 +868,19 @@ def _GetAttachedDevices(test_device=None): |
| Returns: |
| A list of attached devices. |
| """ |
| - attached_devices = [] |
| - |
| - attached_devices = android_commands.GetAttachedDevices() |
| + attached_devices = device_utils.DeviceUtils.HealthyDevices() |
| if test_device: |
| - assert test_device in attached_devices, ( |
| - 'Did not find device %s among attached device. Attached devices: %s' |
| - % (test_device, ', '.join(attached_devices))) |
| - attached_devices = [test_device] |
| - |
| - assert attached_devices, 'No devices attached.' |
| + test_device = [d for d in attached_devices if d == test_device] |
| + if not test_device: |
| + raise device_errors.DeviceUnreachableError( |
| + 'Did not find device %s among attached device. Attached devices: %s' |
| + % (test_device, ', '.join(attached_devices))) |
| + return test_device |
|
perezju
2015/04/29 14:30:11
nit: is than an extra empty line?
jbudorick
2015/04/29 14:33:07
It's here intentionally, if that's what you're ask
|
| - return sorted(attached_devices) |
| + else: |
| + if not attached_devices: |
| + raise device_errors.NoDevicesError() |
| + return sorted(attached_devices) |
| def RunTestsCommand(args, parser): |