[devil] Detect if device is not charging properly.

devil/devil/android/battery_utils_test.py

 #!/usr/bin/env python
 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """
 Unit tests for the contents of battery_utils.py
 """
 
 # pylint: disable=protected-access,unused-argument
@@ skipping 228 matching lines @@
           'test_package1': {'uid': '1000', 'data': [1.0]},
           'test_package2': {'uid': '1001', 'data': [2.0]}
         }
       }
       self.assertEqual(data, check)
 
 
 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):
 
   @mock.patch('time.sleep', mock.Mock())
   def testDischargeDevice_exact(self):
     with self.assertCalls(
         (self.call.battery.GetBatteryInfo(), {'level': '100'}),
         (self.call.battery._HardwareSetCharging(False)),
         (self.call.battery._HardwareSetCharging(True)),
         (self.call.battery.GetBatteryInfo(), {'level': '99'})):
@@ skipping 370 matching lines @@
               ['9,1000,l,pwi,uid,0.0327']),
           (self.call.device.RunShellCommand(
               ['dumpsys', 'battery', 'reset'], check_return=True), [])):
         with self.assertRaises(device_errors.CommandFailedError):
           self.battery._ClearPowerData()
 
 
 if __name__ == '__main__':
   logging.getLogger().setLevel(logging.DEBUG)
   unittest.main(verbosity=2)