[Android] Convert to BatteryUtils

build/android/pylib/device/battery_utils.py

 # Copyright 2015 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.
 
 """Provides a variety of device interactions with power.
 """
 # pylint: disable=unused-argument
 
 import collections
 import contextlib
@@ skipping 228 matching lines @@
       device_errors.CommandFailedError: If device is not L or higher.
     """
     if (self._device.build_version_sdk <
         constants.ANDROID_SDK_VERSION_CODES.LOLLIPOP):
       raise device_errors.DeviceVersionError('Device must be L or higher.')
     try:
       self.DisableBatteryUpdates(timeout=timeout, retries=retries)
       yield
     finally:
       self.EnableBatteryUpdates(timeout=timeout, retries=retries)
+
+  def ChargeDeviceToLevel(self, level, wait_period=60):
+    """ Enables charging and waits for device to be charged to given level.

[jbudorick, 2015/04/01 23:14:20]

nit: no space before Enables

[rnephew (Wrong account), 2015/04/01 23:19:58]

Done.

+
+    Args:
+      level: level of charge to wait for.
+      wait_period: time to wait between checking.

[jbudorick, 2015/04/01 23:14:20]

nit: specify the units, i.e., "time in seconds ..."

[rnephew (Wrong account), 2015/04/01 23:19:58]

Done.

+    """
+    self.SetCharging(True)
+
+    def device_charged():
+      battery_level = self.GetBatteryInfo().get('level')
+      if battery_level is None:
+        logging.warning('Unable to find current battery level.')
+        battery_level = 100
+      else:
+        logging.info('current battery level: %s', battery_level)
+        battery_level = int(battery_level)
+      return battery_level >= level
+
+    timeout_retry.WaitFor(device_charged, wait_period=wait_period)