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 cca89c725e6e1608b1d674ebfa319709af5ca6fa..46800c3569236416682fbd5c2cf87fdcd621a4c0 100644 |
--- a/build/android/pylib/device/battery_utils.py |
+++ b/build/android/pylib/device/battery_utils.py |
@@ -237,8 +237,12 @@ class BatteryUtils(object): |
def battery_updates_disabled(): |
return self.GetCharging() is False |
+ if (self._device.build_version_sdk < |
jbudorick
2015/04/08 17:43:11
Please add this exception to the Raises: section o
fmeawad
2015/04/08 18:00:23
Done.
|
+ constants.ANDROID_SDK_VERSION_CODES.LOLLIPOP): |
+ raise device_errors.DeviceVersionError('Device must be L or higher.') |
+ |
self._device.RunShellCommand( |
- ['dumpsys', 'battery', 'set', 'usb', '1'], check_return=True) |
+ ['dumpsys', 'battery', 'reset'], check_return=True) |
self._device.RunShellCommand( |
['dumpsys', 'batterystats', '--reset'], check_return=True) |
battery_data = self._device.RunShellCommand( |
@@ -252,6 +256,8 @@ class BatteryUtils(object): |
and l[PWI_POWER_INDEX] != 0): |
raise device_errors.CommandFailedError( |
'Non-zero pmi value found after reset.') |
+ self._device.RunShellCommand(['dumpsys', 'battery', 'set', 'ac', '0'], |
+ check_return=True) |
self._device.RunShellCommand(['dumpsys', 'battery', 'set', 'usb', '0'], |
check_return=True) |
timeout_retry.WaitFor(battery_updates_disabled, wait_period=1) |
@@ -267,8 +273,10 @@ class BatteryUtils(object): |
def battery_updates_enabled(): |
return self.GetCharging() is True |
- self._device.RunShellCommand(['dumpsys', 'battery', 'set', 'usb', '1'], |
- check_return=True) |
+ if (self._device.build_version_sdk < |
jbudorick
2015/04/08 17:43:11
same here
fmeawad
2015/04/08 18:00:23
Done.
|
+ constants.ANDROID_SDK_VERSION_CODES.LOLLIPOP): |
+ raise device_errors.DeviceVersionError('Device must be L or higher.') |
+ |
self._device.RunShellCommand(['dumpsys', 'battery', 'reset'], |
check_return=True) |
timeout_retry.WaitFor(battery_updates_enabled, wait_period=1) |