| Index: build/android/pylib/device/battery_utils_test.py
|
| diff --git a/build/android/pylib/device/battery_utils_test.py b/build/android/pylib/device/battery_utils_test.py
|
| index 72799b8b5d109ea8fc3c3cdc89ffc66304b4bad0..74220625f77146c2152d3dfb9c6bf45c563c5140 100755
|
| --- a/build/android/pylib/device/battery_utils_test.py
|
| +++ b/build/android/pylib/device/battery_utils_test.py
|
| @@ -104,13 +104,16 @@ class BatteryUtilsSetChargingTest(BatteryUtilsTest):
|
|
|
| class BatteryUtilsSetBatteryMeasurementTest(BatteryUtilsTest):
|
|
|
| + @mock.patch('time.sleep', mock.Mock())
|
| def testBatteryMeasurement(self):
|
| with self.assertCalls(
|
| (self.call.device.RunShellCommand(
|
| mock.ANY, retries=0, single_line=True,
|
| timeout=10, check_return=True), '22'),
|
| (self.call.device.RunShellCommand(
|
| - ['dumpsys', 'battery', 'reset'], check_return=True), []),
|
| + ['dumpsys', 'battery', 'set', 'usb', '1'], check_return=True), []),
|
| + (self.call.device.RunShellCommand(
|
| + ['dumpsys', 'battery', 'set', 'ac', '1'], check_return=True), []),
|
| (self.call.device.RunShellCommand(
|
| ['dumpsys', 'batterystats', '--reset'], check_return=True), []),
|
| (self.call.device.RunShellCommand(
|
|
|