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 ed66591daa5c6ac2df287ffaca938020d47fa37e..df5f826ff0f205343e188a30d2f560d10c085221 100644 |
--- a/build/android/pylib/device/battery_utils.py |
+++ b/build/android/pylib/device/battery_utils.py |
@@ -289,12 +289,14 @@ class BatteryUtils(object): |
raise device_errors.DeviceVersionError('Device must be L or higher.') |
self._device.RunShellCommand( |
- ['dumpsys', 'battery', 'reset'], check_return=True) |
+ ['dumpsys', 'battery', 'set', 'usb', '1'], check_return=True) |
+ self._device.RunShellCommand( |
+ ['dumpsys', 'battery', 'set', 'ac', '1'], check_return=True) |
self._device.RunShellCommand( |
['dumpsys', 'batterystats', '--reset'], check_return=True) |
battery_data = self._device.RunShellCommand( |
['dumpsys', 'batterystats', '--charged', '--checkin'], |
- check_return=True) |
+ check_return=True, large_output=True) |
ROW_TYPE_INDEX = 3 |
PWI_POWER_INDEX = 5 |
for line in battery_data: |