Index: build/android/pylib/device/battery_utils_test.py |
diff --git a/build/android/pylib/device/battery_utils_test.py b/build/android/pylib/device/battery_utils_test.py |
index 72799b8b5d109ea8fc3c3cdc89ffc66304b4bad0..74220625f77146c2152d3dfb9c6bf45c563c5140 100755 |
--- a/build/android/pylib/device/battery_utils_test.py |
+++ b/build/android/pylib/device/battery_utils_test.py |
@@ -104,13 +104,16 @@ class BatteryUtilsSetChargingTest(BatteryUtilsTest): |
class BatteryUtilsSetBatteryMeasurementTest(BatteryUtilsTest): |
+ @mock.patch('time.sleep', mock.Mock()) |
def testBatteryMeasurement(self): |
with self.assertCalls( |
(self.call.device.RunShellCommand( |
mock.ANY, retries=0, single_line=True, |
timeout=10, check_return=True), '22'), |
(self.call.device.RunShellCommand( |
- ['dumpsys', 'battery', 'reset'], check_return=True), []), |
+ ['dumpsys', 'battery', 'set', 'usb', '1'], check_return=True), []), |
+ (self.call.device.RunShellCommand( |
+ ['dumpsys', 'battery', 'set', 'ac', '1'], check_return=True), []), |
(self.call.device.RunShellCommand( |
['dumpsys', 'batterystats', '--reset'], check_return=True), []), |
(self.call.device.RunShellCommand( |