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..55fe60125d0732fef87b7516dad0a8a135fe63bc 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, ignore=['GetDeviceSerial']) |
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')): |
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) |