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 b7e807e06d74cf063d247e10b99c93fe870b4b5c..0e9fcc86d061a0eb2476e2dc6c7f7edffeecc752 100755 |
--- a/build/android/pylib/device/device_utils_test.py |
+++ b/build/android/pylib/device/device_utils_test.py |
@@ -1468,6 +1468,17 @@ class DeviceUtilsSetChargingTest(DeviceUtilsTest): |
self.device.SetCharging(False) |
+class DeviceUtilsSetBatteryMeasurementTest(DeviceUtilsTest): |
+ |
+ def testBatteryMeasurement(self): |
+ with self.assertCalls( |
+ (self.call.device.RunShellCommand(mock.ANY, check_return=True), []), |
+ (self.call.device.GetCharging(), False), |
+ (self.call.device.RunShellCommand(mock.ANY, check_return=True), []), |
+ (self.call.device.GetCharging(), True)): |
+ with self.device.BatteryMeasurement(): |
+ pass |
+ |
class DeviceUtilsStrTest(DeviceUtilsTest): |