Index: build/android/pylib/local/device/local_device_environment.py |
diff --git a/build/android/pylib/local/device/local_device_environment.py b/build/android/pylib/local/device/local_device_environment.py |
index 72acb6e2ec44e71ce1922b88225888035123a68d..0d02ca3e2f8eb81e91bfff8543e488a800b9db25 100644 |
--- a/build/android/pylib/local/device/local_device_environment.py |
+++ b/build/android/pylib/local/device/local_device_environment.py |
@@ -13,26 +13,24 @@ class LocalDeviceEnvironment(environment.Environment): |
def __init__(self, args, _error_func): |
super(LocalDeviceEnvironment, self).__init__() |
- self._device = args.test_device |
+ self._device_serial = args.test_device |
self._devices = [] |
self._max_tries = 1 + args.num_retries |
self._tool_name = args.tool |
#override |
def SetUp(self): |
- # TODO(jbudorick): This can be refined to support filters etc. |
- available_devices = adb_wrapper.AdbWrapper.GetDevices() |
+ available_devices = device_utils.DeviceUtils.HealthyDevices() |
if not available_devices: |
raise device_errors.NoDevicesError |
- if self._device: |
- if self._device not in available_devices: |
+ if self._device_serial: |
+ self._devices = [d for d in available_devices |
+ if d.adb.GetDeviceSerial == self._device_serial] |
+ if not self._devices: |
raise device_errors.DeviceUnreachableError( |
- 'Could not find device %r' % self._device) |
- self._devices = [device_utils.DeviceUtils(self._device)] |
+ 'Could not find device %r' % self._device_serial) |
else: |
- self._devices = [ |
- device_utils.DeviceUtils(s) |
- for s in available_devices] |
+ self._devices = available_devices |
@property |
def devices(self): |