| Index: build/android/pylib/device/device_utils_test.py
|
| diff --git a/build/android/pylib/device/device_utils_test.py b/build/android/pylib/device/device_utils_test.py
|
| index 1a783c314716af8784b8b5a939a4fbc3ec699fbf..15d7355160dfe234115c2b966ac05104b544748c 100755
|
| --- a/build/android/pylib/device/device_utils_test.py
|
| +++ b/build/android/pylib/device/device_utils_test.py
|
| @@ -24,6 +24,7 @@ from pylib import cmd_helper
|
| from pylib import constants
|
| from pylib import device_signal
|
| from pylib.device import adb_wrapper
|
| +from pylib.device import device_blacklist
|
| from pylib.device import device_errors
|
| from pylib.device import device_utils
|
| from pylib.device import intent
|
| @@ -1867,25 +1868,6 @@ class DeviceUtilsClientCache(DeviceUtilsTest):
|
| self.assertEqual(client_cache_two, {})
|
|
|
|
|
| -class DeviceUtilsParallelTest(mock_calls.TestCase):
|
| -
|
| - def testParallel_default(self):
|
| - test_serials = ['0123456789abcdef', 'fedcba9876543210']
|
| - with self.assertCall(
|
| - mock.call.pylib.device.device_utils.DeviceUtils.HealthyDevices(),
|
| - [device_utils.DeviceUtils(s) for s in test_serials]):
|
| - parallel_devices = device_utils.DeviceUtils.parallel()
|
| - for serial, device in zip(test_serials, parallel_devices.pGet(None)):
|
| - self.assertTrue(isinstance(device, device_utils.DeviceUtils))
|
| - self.assertEquals(serial, device.adb.GetDeviceSerial())
|
| -
|
| - def testParallel_noDevices(self):
|
| - with self.assertCall(
|
| - mock.call.pylib.device.device_utils.DeviceUtils.HealthyDevices(), []):
|
| - with self.assertRaises(device_errors.NoDevicesError):
|
| - device_utils.DeviceUtils.parallel()
|
| -
|
| -
|
| class DeviceUtilsHealthyDevicesTest(mock_calls.TestCase):
|
|
|
| def _createAdbWrapperMock(self, serial, is_ready=True):
|
| @@ -1893,25 +1875,25 @@ class DeviceUtilsHealthyDevicesTest(mock_calls.TestCase):
|
| adb.is_ready = is_ready
|
| return adb
|
|
|
| - def testHealthyDevices_default(self):
|
| + def testHealthyDevices_emptyBlacklist(self):
|
| test_serials = ['0123456789abcdef', 'fedcba9876543210']
|
| with self.assertCalls(
|
| - (mock.call.pylib.device.device_blacklist.ReadBlacklist(), []),
|
| (mock.call.pylib.device.adb_wrapper.AdbWrapper.Devices(),
|
| [self._createAdbWrapperMock(s) for s in test_serials])):
|
| - devices = device_utils.DeviceUtils.HealthyDevices()
|
| + blacklist = mock.NonCallableMock(**{'Read.return_value': []})
|
| + devices = device_utils.DeviceUtils.HealthyDevices(blacklist)
|
| for serial, device in zip(test_serials, devices):
|
| self.assertTrue(isinstance(device, device_utils.DeviceUtils))
|
| self.assertEquals(serial, device.adb.GetDeviceSerial())
|
|
|
| - def testHealthyDevices_blacklisted(self):
|
| + def testHealthyDevices_blacklist(self):
|
| test_serials = ['0123456789abcdef', 'fedcba9876543210']
|
| with self.assertCalls(
|
| - (mock.call.pylib.device.device_blacklist.ReadBlacklist(),
|
| - ['fedcba9876543210']),
|
| (mock.call.pylib.device.adb_wrapper.AdbWrapper.Devices(),
|
| [self._createAdbWrapperMock(s) for s in test_serials])):
|
| - devices = device_utils.DeviceUtils.HealthyDevices()
|
| + blacklist = mock.NonCallableMock(
|
| + **{'Read.return_value': ['fedcba9876543210']})
|
| + devices = device_utils.DeviceUtils.HealthyDevices(blacklist)
|
| self.assertEquals(1, len(devices))
|
| self.assertTrue(isinstance(devices[0], device_utils.DeviceUtils))
|
| self.assertEquals('0123456789abcdef', devices[0].adb.GetDeviceSerial())
|
|
|