| Index: build/android/pylib/device/device_utils_test.py
|
| diff --git a/build/android/pylib/device/device_utils_test.py b/build/android/pylib/device/device_utils_test.py
|
| index b7e807e06d74cf063d247e10b99c93fe870b4b5c..2457e1737e4b927890ec4f6ce233e1b936b19ea0 100755
|
| --- a/build/android/pylib/device/device_utils_test.py
|
| +++ b/build/android/pylib/device/device_utils_test.py
|
| @@ -1467,6 +1467,31 @@ class DeviceUtilsSetChargingTest(DeviceUtilsTest):
|
| (self.call.device.GetCharging(), False)):
|
| self.device.SetCharging(False)
|
|
|
| + def testSetChargingDefault_enabled(self):
|
| + with self.assertCalls(
|
| + (self.call.device.FileExists(mock.ANY), False),
|
| + (self.call.device.FileExists(mock.ANY), False),
|
| + (self.call.device.RunShellCommand(mock.ANY, check_return=True), []),
|
| + (self.call.device.GetCharging(), False),
|
| + (self.call.device.RunShellCommand(mock.ANY, check_return=True), []),
|
| + (self.call.device.GetCharging(), True)):
|
| + self.device.SetCharging(True)
|
| + self.assertEqual(
|
| + 'dumpsys battery reset',
|
| + self.device._cache['charging_config']['enable_command'])
|
| +
|
| + def testSetChargingDefault_disabled(self):
|
| + with self.assertCalls(
|
| + (self.call.device.FileExists(mock.ANY), False),
|
| + (self.call.device.FileExists(mock.ANY), False),
|
| + (self.call.device.RunShellCommand(mock.ANY, check_return=True), []),
|
| + (self.call.device.GetCharging(), True),
|
| + (self.call.device.RunShellCommand(mock.ANY, check_return=True), []),
|
| + (self.call.device.GetCharging(), False)):
|
| + self.device.SetCharging(False)
|
| + self.assertEqual(
|
| + 'dumpsys battery set usb 0',
|
| + self.device._cache['charging_config']['disable_command'])
|
|
|
|
|
| class DeviceUtilsStrTest(DeviceUtilsTest):
|
|
|