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 414d856b27aca294ba3ca6d60a6e42685f6efd3d..3806e266a4f2ea564a13bd9cd777ed538c8add91 100755 |
--- a/build/android/pylib/device/device_utils_test.py |
+++ b/build/android/pylib/device/device_utils_test.py |
@@ -1551,9 +1551,10 @@ class DeviceUtilsParallelTest(mock_calls.TestCase): |
def testParallel_default(self): |
test_serials = ['0123456789abcdef', 'fedcba9876543210'] |
- with self.assertCall( |
- mock.call.pylib.device.adb_wrapper.AdbWrapper.GetDevices(), |
- [_AdbWrapperMock(serial) for serial in test_serials]): |
+ with self.assertCalls( |
+ (mock.call.pylib.device.device_filter.DefaultFilters(), None), |
+ (mock.call.pylib.device.adb_wrapper.AdbWrapper.Devices(filters=None), |
+ [_AdbWrapperMock(serial) for serial in test_serials])): |
parallel_devices = device_utils.DeviceUtils.parallel() |
for serial, device in zip(test_serials, parallel_devices.pGet(None)): |
self.assertTrue( |
@@ -1562,8 +1563,10 @@ class DeviceUtilsParallelTest(mock_calls.TestCase): |
'Expected a DeviceUtils object with serial %s' % serial) |
def testParallel_noDevices(self): |
- with self.assertCall( |
- mock.call.pylib.device.adb_wrapper.AdbWrapper.GetDevices(), []): |
+ with self.assertCalls( |
+ (mock.call.pylib.device.device_filter.DefaultFilters(), None), |
+ (mock.call.pylib.device.adb_wrapper.AdbWrapper.Devices(filters=None), |
+ [])): |
with self.assertRaises(device_errors.NoDevicesError): |
device_utils.DeviceUtils.parallel() |