diff options
Diffstat (limited to 'build/android')
-rw-r--r-- | build/android/pylib/device/device_utils.py | 2 | ||||
-rwxr-xr-x | build/android/pylib/device/device_utils_test.py | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/build/android/pylib/device/device_utils.py b/build/android/pylib/device/device_utils.py index 1faca07..9f538e1 100644 --- a/build/android/pylib/device/device_utils.py +++ b/build/android/pylib/device/device_utils.py @@ -1554,6 +1554,8 @@ class DeviceUtils(object): def battery_updates_enabled(): return self.GetCharging() is True + self.RunShellCommand(['dumpsys', 'battery', 'set', 'usb', '1'], + check_return=True) self.RunShellCommand(['dumpsys', 'battery', 'reset'], check_return=True) timeout_retry.WaitFor(battery_updates_enabled, wait_period=1) diff --git a/build/android/pylib/device/device_utils_test.py b/build/android/pylib/device/device_utils_test.py index 4317684c..98e3539 100755 --- a/build/android/pylib/device/device_utils_test.py +++ b/build/android/pylib/device/device_utils_test.py @@ -1553,6 +1553,8 @@ class DeviceUtilsSetBatteryMeasurementTest(DeviceUtilsTest): ['dumpsys', 'battery', 'set', 'usb', '0'], check_return=True), []), (self.call.device.GetCharging(), False), (self.call.device.RunShellCommand( + ['dumpsys', 'battery', 'set', 'usb', '1'], check_return=True), []), + (self.call.device.RunShellCommand( ['dumpsys', 'battery', 'reset'], check_return=True), []), (self.call.device.GetCharging(), True)): with self.device.BatteryMeasurement(): |