| Index: build/android/test_runner.py
|
| diff --git a/build/android/test_runner.py b/build/android/test_runner.py
|
| index c54ed28d324054eae3c5aee72aa1442d80bf1f36..f3dd764e554929b40f364460259038599549d4e6 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,16 +868,16 @@ 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)))
|
| +
|
| + if not attached_devices:
|
| + raise device_errors.NoDevicesError()
|
|
|
| return sorted(attached_devices)
|
|
|
|
|