Chromium Code Reviews| Index: build/android/pylib/device/battery_utils.py |
| diff --git a/build/android/pylib/device/battery_utils.py b/build/android/pylib/device/battery_utils.py |
| index 96e3bf660c5841642ba3fc0fb88e560b11314df5..b6b781c9e1378e2ef143b4b02dd8f1f3cc1f1357 100644 |
| --- a/build/android/pylib/device/battery_utils.py |
| +++ b/build/android/pylib/device/battery_utils.py |
| @@ -246,3 +246,24 @@ class BatteryUtils(object): |
| 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) |