Index: build/android/devil/android/battery_utils_test.py |
diff --git a/build/android/devil/android/battery_utils_test.py b/build/android/devil/android/battery_utils_test.py |
index 734b68d1b20f00ec4de17b1b45a7b07a747c6e28..0d26b964f0c67e3c08a2b7d0cc7c0b1cd30ec146 100755 |
--- a/build/android/devil/android/battery_utils_test.py |
+++ b/build/android/devil/android/battery_utils_test.py |
@@ -244,13 +244,44 @@ class BatteryUtilsGetPowerData(BatteryUtilsTest): |
class BatteryUtilsChargeDevice(BatteryUtilsTest): |
@mock.patch('time.sleep', mock.Mock()) |
- def testChargeDeviceToLevel(self): |
+ def testChargeDeviceToLevel_pass(self): |
with self.assertCalls( |
(self.call.battery.SetCharging(True)), |
(self.call.battery.GetBatteryInfo(), {'level': '50'}), |
(self.call.battery.GetBatteryInfo(), {'level': '100'})): |
self.battery.ChargeDeviceToLevel(95) |
+ @mock.patch('time.sleep', mock.Mock()) |
+ def testChargeDeviceToLevel_failureSame(self): |
+ with self.assertCalls( |
+ (self.call.battery.SetCharging(True)), |
+ (self.call.battery.GetBatteryInfo(), {'level': '50'}), |
+ (self.call.battery.GetBatteryInfo(), {'level': '50'}), |
+ |
+ (self.call.battery.GetBatteryInfo(), {'level': '50'})): |
+ with self.assertRaises(device_errors.DeviceChargingError): |
+ old_max = battery_utils._MAX_CHARGE_ERROR |
+ try: |
+ battery_utils._MAX_CHARGE_ERROR = 2 |
+ self.battery.ChargeDeviceToLevel(95) |
+ finally: |
+ battery_utils._MAX_CHARGE_ERROR = old_max |
+ |
+ @mock.patch('time.sleep', mock.Mock()) |
+ def testChargeDeviceToLevel_failureDischarge(self): |
+ with self.assertCalls( |
+ (self.call.battery.SetCharging(True)), |
+ (self.call.battery.GetBatteryInfo(), {'level': '50'}), |
+ (self.call.battery.GetBatteryInfo(), {'level': '49'}), |
+ (self.call.battery.GetBatteryInfo(), {'level': '48'})): |
+ with self.assertRaises(device_errors.DeviceChargingError): |
+ old_max = battery_utils._MAX_CHARGE_ERROR |
+ try: |
+ battery_utils._MAX_CHARGE_ERROR = 2 |
+ self.battery.ChargeDeviceToLevel(95) |
+ finally: |
+ battery_utils._MAX_CHARGE_ERROR = old_max |
+ |
class BatteryUtilsDischargeDevice(BatteryUtilsTest): |