summaryrefslogtreecommitdiffstats
path: root/build/android
diff options
context:
space:
mode:
Diffstat (limited to 'build/android')
-rw-r--r--build/android/pylib/device/device_utils.py2
-rwxr-xr-xbuild/android/pylib/device/device_utils_test.py2
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():