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 68c8b716f2f5dfac6819ab46b47cc38219c10dd7..9d41741d192441d89b6e5f4e03ad19b6112de79f 100755 |
--- a/build/android/pylib/device/device_utils_test.py |
+++ b/build/android/pylib/device/device_utils_test.py |
@@ -457,9 +457,17 @@ class DeviceUtilsRebootTest(DeviceUtilsNewImplTest): |
self.call.adb.Reboot(), |
(self.call.device.IsOnline(), True), |
(self.call.device.IsOnline(), False), |
- self.call.device.WaitUntilFullyBooted()): |
+ self.call.device.WaitUntilFullyBooted(wifi=False)): |
self.device.Reboot(block=True) |
+ def testReboot_blockUntilWifi(self): |
+ with self.assertCalls( |
+ self.call.adb.Reboot(), |
+ (self.call.device.IsOnline(), True), |
+ (self.call.device.IsOnline(), False), |
+ self.call.device.WaitUntilFullyBooted(wifi=True)): |
+ self.device.Reboot(block=True, wifi=True) |
+ |
class DeviceUtilsInstallTest(DeviceUtilsNewImplTest): |