Chromium Code Reviews| 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 4f47021e2dcb7ba66cc02c45f68dc3ac9597289e..24d380e11d966f3d93af657543f5373d05ebd481 100755 |
| --- a/build/android/pylib/device/device_utils_test.py |
| +++ b/build/android/pylib/device/device_utils_test.py |
| @@ -43,28 +43,30 @@ class DeviceUtilsTest(unittest.TestCase): |
| def testInitWithStr(self): |
| serial_as_str = str('0123456789abcdef') |
| d = device_utils.DeviceUtils('0123456789abcdef') |
| - self.assertEqual(serial_as_str, d.old_interface.GetDevice()) |
| + self.assertEqual(serial_as_str, d.adb.GetDeviceSerial()) |
| def testInitWithUnicode(self): |
| serial_as_unicode = unicode('fedcba9876543210') |
| d = device_utils.DeviceUtils(serial_as_unicode) |
| - self.assertEqual(serial_as_unicode, d.old_interface.GetDevice()) |
| + self.assertEqual(serial_as_unicode, d.adb.GetDeviceSerial()) |
| def testInitWithAdbWrapper(self): |
| serial = '123456789abcdef0' |
| a = adb_wrapper.AdbWrapper(serial) |
| d = device_utils.DeviceUtils(a) |
| - self.assertEqual(serial, d.old_interface.GetDevice()) |
| + self.assertEqual(serial, d.adb.GetDeviceSerial()) |
| def testInitWithAndroidCommands(self): |
| serial = '0fedcba987654321' |
| a = android_commands.AndroidCommands(device=serial) |
| d = device_utils.DeviceUtils(a) |
| - self.assertEqual(serial, d.old_interface.GetDevice()) |
| + self.assertEqual(serial, d.adb.GetDeviceSerial()) |
| - def testInitWithNone(self): |
| - d = device_utils.DeviceUtils(None) |
| - self.assertIsNone(d.old_interface.GetDevice()) |
| + def testInitWithMissing_fails(self): |
| + with self.assertRaises(ValueError): |
| + device_utils.DeviceUtils(None) |
| + with self.assertRaises(ValueError): |
| + device_utils.DeviceUtils('') |
| class MockTempFile(object): |
| @@ -246,7 +248,7 @@ class DeviceUtilsNewImplTest(mock_calls.TestCase): |
| self.adb.GetDeviceSerial.return_value = test_serial |
| self.device = device_utils.DeviceUtils( |
| self.adb, default_timeout=10, default_retries=0) |
| - self.watchMethodCalls(self.call.adb) |
| + self.watchMethodCalls(self.call.adb, exclude=['GetDeviceSerial']) |
|
perezju
2014/12/02 11:13:18
GetDeviceSerial is now excluded from the methods w
|
| def ShellError(self, output=None, exit_code=1): |
| def action(cmd, *args, **kwargs): |
| @@ -1552,23 +1554,13 @@ class DeviceUtilsGetMemoryUsageForPidTest(DeviceUtilsOldImplTest): |
| self.assertEqual({}, self.device.GetMemoryUsageForPid(4321)) |
| -class DeviceUtilsStrTest(DeviceUtilsOldImplTest): |
| +class DeviceUtilsStrTest(DeviceUtilsNewImplTest): |
| - def testStr_noAdbCalls(self): |
| - with self.assertNoAdbCalls(): |
| - self.assertEqual('0123456789abcdef', str(self.device)) |
| - |
| - def testStr_noSerial(self): |
| - self.device = device_utils.DeviceUtils(None) |
| - with self.assertCalls('adb get-serialno', '0123456789abcdef'): |
| + def testStr_returnsSerial(self): |
| + with self.assertCalls( |
| + (self.call.adb.GetDeviceSerial(), '0123456789abcdef')): |
|
perezju
2014/12/02 11:13:18
... except if we really want to watch it
|
| self.assertEqual('0123456789abcdef', str(self.device)) |
| - def testStr_noSerial_noDevices(self): |
| - self.device = device_utils.DeviceUtils(None) |
| - with self.assertCalls('adb get-serialno', 'unknown'), ( |
| - self.assertRaises(device_errors.NoDevicesError)): |
| - str(self.device) |
| - |
| if __name__ == '__main__': |
| logging.getLogger().setLevel(logging.DEBUG) |