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): |