diff options
Diffstat (limited to 'catapult/devil/devil/android/battery_utils_test.py')
| -rwxr-xr-x | catapult/devil/devil/android/battery_utils_test.py | 508 |
1 files changed, 268 insertions, 240 deletions
diff --git a/catapult/devil/devil/android/battery_utils_test.py b/catapult/devil/devil/android/battery_utils_test.py index 07c74967..7bfb9551 100755 --- a/catapult/devil/devil/android/battery_utils_test.py +++ b/catapult/devil/devil/android/battery_utils_test.py @@ -2,7 +2,6 @@ # Copyright 2014 The Chromium Authors. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. - """ Unit tests for the contents of battery_utils.py """ @@ -34,46 +33,64 @@ _DUMPSYS_OUTPUT = [ class BatteryUtilsTest(mock_calls.TestCase): _NEXUS_5 = { - 'name': 'Nexus 5', - 'witness_file': '/sys/kernel/debug/bq24192/INPUT_SRC_CONT', - 'enable_command': ( - 'echo 0x4A > /sys/kernel/debug/bq24192/INPUT_SRC_CONT && ' - 'echo 1 > /sys/class/power_supply/usb/online'), - 'disable_command': ( - 'echo 0xCA > /sys/kernel/debug/bq24192/INPUT_SRC_CONT && ' - 'chmod 644 /sys/class/power_supply/usb/online && ' - 'echo 0 > /sys/class/power_supply/usb/online'), - 'charge_counter': None, - 'voltage': None, - 'current': None, + 'name': + 'Nexus 5', + 'witness_file': + '/sys/kernel/debug/bq24192/INPUT_SRC_CONT', + 'enable_command': ( + 'echo 0x4A > /sys/kernel/debug/bq24192/INPUT_SRC_CONT && ' + 'echo 1 > /sys/class/power_supply/usb/online'), + 'disable_command': ( + 'echo 0xCA > /sys/kernel/debug/bq24192/INPUT_SRC_CONT && ' + 'chmod 644 /sys/class/power_supply/usb/online && ' + 'echo 0 > /sys/class/power_supply/usb/online'), + 'charge_counter': + None, + 'voltage': + None, + 'current': + None, } _NEXUS_6 = { - 'name': 'Nexus 6', - 'witness_file': None, - 'enable_command': None, - 'disable_command': None, - 'charge_counter': ( - '/sys/class/power_supply/max170xx_battery/charge_counter_ext'), - 'voltage': '/sys/class/power_supply/max170xx_battery/voltage_now', - 'current': '/sys/class/power_supply/max170xx_battery/current_now', + 'name': + 'Nexus 6', + 'witness_file': + None, + 'enable_command': + None, + 'disable_command': + None, + 'charge_counter': ( + '/sys/class/power_supply/max170xx_battery/charge_counter_ext'), + 'voltage': + '/sys/class/power_supply/max170xx_battery/voltage_now', + 'current': + '/sys/class/power_supply/max170xx_battery/current_now', } _NEXUS_10 = { - 'name': 'Nexus 10', - 'witness_file': None, - 'enable_command': None, - 'disable_command': None, - 'charge_counter': ( - '/sys/class/power_supply/ds2784-fuelgauge/charge_counter_ext'), - 'voltage': '/sys/class/power_supply/ds2784-fuelgauge/voltage_now', - 'current': '/sys/class/power_supply/ds2784-fuelgauge/current_now', + 'name': + 'Nexus 10', + 'witness_file': + None, + 'enable_command': + None, + 'disable_command': + None, + 'charge_counter': ( + '/sys/class/power_supply/ds2784-fuelgauge/charge_counter_ext'), + 'voltage': + '/sys/class/power_supply/ds2784-fuelgauge/voltage_now', + 'current': + '/sys/class/power_supply/ds2784-fuelgauge/current_now', } def ShellError(self, output=None, status=1): def action(cmd, *args, **kwargs): - raise device_errors.AdbShellCommandFailedError( - cmd, output, status, str(self.device)) + raise device_errors.AdbShellCommandFailedError(cmd, output, status, + str(self.device)) + if output is None: output = 'Permission denied\n' return action @@ -88,7 +105,6 @@ class BatteryUtilsTest(mock_calls.TestCase): class BatteryUtilsInitTest(unittest.TestCase): - def testInitWithDeviceUtil(self): serial = '0fedcba987654321' d = device_utils.DeviceUtils(serial) @@ -103,78 +119,75 @@ class BatteryUtilsInitTest(unittest.TestCase): class BatteryUtilsSetChargingTest(BatteryUtilsTest): - @mock.patch('time.sleep', mock.Mock()) def testHardwareSetCharging_enabled(self): self.battery._cache['profile'] = self._NEXUS_5 - with self.assertCalls( - (self.call.device.RunShellCommand( - mock.ANY, shell=True, check_return=True, as_root=True, - large_output=True), []), - (self.call.battery.GetCharging(), False), - (self.call.battery.GetCharging(), True)): + with self.assertCalls((self.call.device.RunShellCommand( + mock.ANY, + shell=True, + check_return=True, + as_root=True, + large_output=True), []), (self.call.battery.GetCharging(), False), + (self.call.battery.GetCharging(), True)): self.battery._HardwareSetCharging(True) def testHardwareSetCharging_alreadyEnabled(self): self.battery._cache['profile'] = self._NEXUS_5 - with self.assertCalls( - (self.call.device.RunShellCommand( - mock.ANY, shell=True, check_return=True, as_root=True, - large_output=True), []), - (self.call.battery.GetCharging(), True)): + with self.assertCalls((self.call.device.RunShellCommand( + mock.ANY, + shell=True, + check_return=True, + as_root=True, + large_output=True), []), (self.call.battery.GetCharging(), True)): self.battery._HardwareSetCharging(True) @mock.patch('time.sleep', mock.Mock()) def testHardwareSetCharging_disabled(self): self.battery._cache['profile'] = self._NEXUS_5 - with self.assertCalls( - (self.call.device.RunShellCommand( - mock.ANY, shell=True, check_return=True, as_root=True, - large_output=True), []), - (self.call.battery.GetCharging(), True), - (self.call.battery.GetCharging(), False)): + with self.assertCalls((self.call.device.RunShellCommand( + mock.ANY, + shell=True, + check_return=True, + as_root=True, + large_output=True), []), (self.call.battery.GetCharging(), True), + (self.call.battery.GetCharging(), False)): self.battery._HardwareSetCharging(False) class BatteryUtilsSetBatteryMeasurementTest(BatteryUtilsTest): - @mock.patch('time.sleep', mock.Mock()) def testBatteryMeasurementWifi(self): - with self.patch_call(self.call.device.build_version_sdk, - return_value=22): + with self.patch_call(self.call.device.build_version_sdk, return_value=22): with self.assertCalls( (self.call.battery._ClearPowerData(), True), (self.call.device.RunShellCommand( ['dumpsys', 'battery', 'set', 'ac', '0'], check_return=True), []), (self.call.device.RunShellCommand( ['dumpsys', 'battery', 'set', 'usb', '0'], check_return=True), - []), - (self.call.battery.GetCharging(), False), - (self.call.device.RunShellCommand( - ['dumpsys', 'battery', 'reset'], check_return=True), []), + []), (self.call.battery.GetCharging(), False), + (self.call.device.RunShellCommand(['dumpsys', 'battery', 'reset'], + check_return=True), []), (self.call.battery.GetCharging(), False), (self.call.device.RunShellCommand( ['dumpsys', 'battery'], check_return=True), ['UPDATES STOPPED']), (self.call.battery.GetCharging(), False), - (self.call.device.RunShellCommand( - ['dumpsys', 'battery'], check_return=True), [])): + (self.call.device.RunShellCommand(['dumpsys', 'battery'], + check_return=True), [])): with self.battery.BatteryMeasurement(): pass @mock.patch('time.sleep', mock.Mock()) def testBatteryMeasurementUsb(self): - with self.patch_call(self.call.device.build_version_sdk, - return_value=22): + with self.patch_call(self.call.device.build_version_sdk, return_value=22): with self.assertCalls( (self.call.battery._ClearPowerData(), True), (self.call.device.RunShellCommand( ['dumpsys', 'battery', 'set', 'ac', '0'], check_return=True), []), (self.call.device.RunShellCommand( ['dumpsys', 'battery', 'set', 'usb', '0'], check_return=True), - []), - (self.call.battery.GetCharging(), False), - (self.call.device.RunShellCommand( - ['dumpsys', 'battery', 'reset'], check_return=True), []), + []), (self.call.battery.GetCharging(), False), + (self.call.device.RunShellCommand(['dumpsys', 'battery', 'reset'], + check_return=True), []), (self.call.battery.GetCharging(), False), (self.call.device.RunShellCommand( ['dumpsys', 'battery'], check_return=True), ['UPDATES STOPPED']), @@ -184,86 +197,102 @@ class BatteryUtilsSetBatteryMeasurementTest(BatteryUtilsTest): class BatteryUtilsGetPowerData(BatteryUtilsTest): - def testGetPowerData(self): with self.assertCalls( - (self.call.device.RunShellCommand( - ['dumpsys', 'batterystats', '-c'], - check_return=True, large_output=True), - _DUMPSYS_OUTPUT)): + (self.call.device.RunShellCommand(['dumpsys', 'batterystats', '-c'], + check_return=True, + large_output=True), _DUMPSYS_OUTPUT)): data = self.battery.GetPowerData() check = { - 'system_total': 2000.0, - 'per_package': { - 'test_package1': {'uid': '1000', 'data': [1.0]}, - 'test_package2': {'uid': '1001', 'data': [2.0]} - } + 'system_total': 2000.0, + 'per_package': { + 'test_package1': { + 'uid': '1000', + 'data': [1.0] + }, + 'test_package2': { + 'uid': '1001', + 'data': [2.0] + } + } } self.assertEqual(data, check) def testGetPowerData_packageCollisionSame(self): self.battery._cache['uids'] = {'test_package1': '1000'} with self.assertCall( - self.call.device.RunShellCommand( - ['dumpsys', 'batterystats', '-c'], - check_return=True, large_output=True), - _DUMPSYS_OUTPUT): + self.call.device.RunShellCommand(['dumpsys', 'batterystats', '-c'], + check_return=True, + large_output=True), _DUMPSYS_OUTPUT): data = self.battery.GetPowerData() check = { - 'system_total': 2000.0, - 'per_package': { - 'test_package1': {'uid': '1000', 'data': [1.0]}, - 'test_package2': {'uid': '1001', 'data': [2.0]} - } + 'system_total': 2000.0, + 'per_package': { + 'test_package1': { + 'uid': '1000', + 'data': [1.0] + }, + 'test_package2': { + 'uid': '1001', + 'data': [2.0] + } + } } self.assertEqual(data, check) def testGetPowerData_packageCollisionDifferent(self): self.battery._cache['uids'] = {'test_package1': '1'} with self.assertCall( - self.call.device.RunShellCommand( - ['dumpsys', 'batterystats', '-c'], - check_return=True, large_output=True), - _DUMPSYS_OUTPUT): + self.call.device.RunShellCommand(['dumpsys', 'batterystats', '-c'], + check_return=True, + large_output=True), _DUMPSYS_OUTPUT): with self.assertRaises(device_errors.CommandFailedError): self.battery.GetPowerData() def testGetPowerData_cacheCleared(self): with self.assertCalls( - (self.call.device.RunShellCommand( - ['dumpsys', 'batterystats', '-c'], - check_return=True, large_output=True), - _DUMPSYS_OUTPUT)): + (self.call.device.RunShellCommand(['dumpsys', 'batterystats', '-c'], + check_return=True, + large_output=True), _DUMPSYS_OUTPUT)): self.battery._cache.clear() data = self.battery.GetPowerData() check = { - 'system_total': 2000.0, - 'per_package': { - 'test_package1': {'uid': '1000', 'data': [1.0]}, - 'test_package2': {'uid': '1001', 'data': [2.0]} - } + 'system_total': 2000.0, + 'per_package': { + 'test_package1': { + 'uid': '1000', + 'data': [1.0] + }, + 'test_package2': { + 'uid': '1001', + 'data': [2.0] + } + } } self.assertEqual(data, check) class BatteryUtilsChargeDevice(BatteryUtilsTest): - @mock.patch('time.sleep', mock.Mock()) def testChargeDeviceToLevel_pass(self): - with self.assertCalls( - (self.call.battery.SetCharging(True)), - (self.call.battery.GetBatteryInfo(), {'level': '50'}), - (self.call.battery.GetBatteryInfo(), {'level': '100'})): + with self.assertCalls((self.call.battery.SetCharging(True)), + (self.call.battery.GetBatteryInfo(), { + 'level': '50' + }), (self.call.battery.GetBatteryInfo(), { + 'level': '100' + })): self.battery.ChargeDeviceToLevel(95) @mock.patch('time.sleep', mock.Mock()) def testChargeDeviceToLevel_failureSame(self): - with self.assertCalls( - (self.call.battery.SetCharging(True)), - (self.call.battery.GetBatteryInfo(), {'level': '50'}), - (self.call.battery.GetBatteryInfo(), {'level': '50'}), - - (self.call.battery.GetBatteryInfo(), {'level': '50'})): + with self.assertCalls((self.call.battery.SetCharging(True)), + (self.call.battery.GetBatteryInfo(), { + 'level': '50' + }), (self.call.battery.GetBatteryInfo(), { + 'level': '50' + }), (self.call.battery.GetBatteryInfo(), { + 'level': '50' + })): with self.assertRaises(device_errors.DeviceChargingError): old_max = battery_utils._MAX_CHARGE_ERROR try: @@ -274,11 +303,14 @@ class BatteryUtilsChargeDevice(BatteryUtilsTest): @mock.patch('time.sleep', mock.Mock()) def testChargeDeviceToLevel_failureDischarge(self): - with self.assertCalls( - (self.call.battery.SetCharging(True)), - (self.call.battery.GetBatteryInfo(), {'level': '50'}), - (self.call.battery.GetBatteryInfo(), {'level': '49'}), - (self.call.battery.GetBatteryInfo(), {'level': '48'})): + with self.assertCalls((self.call.battery.SetCharging(True)), + (self.call.battery.GetBatteryInfo(), { + 'level': '50' + }), (self.call.battery.GetBatteryInfo(), { + 'level': '49' + }), (self.call.battery.GetBatteryInfo(), { + 'level': '48' + })): with self.assertRaises(device_errors.DeviceChargingError): old_max = battery_utils._MAX_CHARGE_ERROR try: @@ -289,158 +321,167 @@ class BatteryUtilsChargeDevice(BatteryUtilsTest): class BatteryUtilsDischargeDevice(BatteryUtilsTest): - @mock.patch('time.sleep', mock.Mock()) def testDischargeDevice_exact(self): - with self.assertCalls( - (self.call.battery.GetBatteryInfo(), {'level': '100'}), - (self.call.battery._HardwareSetCharging(False)), - (self.call.battery._HardwareSetCharging(True)), - (self.call.battery.GetBatteryInfo(), {'level': '99'})): + with self.assertCalls((self.call.battery.GetBatteryInfo(), { + 'level': '100' + }), (self.call.battery._HardwareSetCharging(False)), + (self.call.battery._HardwareSetCharging(True)), + (self.call.battery.GetBatteryInfo(), { + 'level': '99' + })): self.battery._DischargeDevice(1) @mock.patch('time.sleep', mock.Mock()) def testDischargeDevice_over(self): - with self.assertCalls( - (self.call.battery.GetBatteryInfo(), {'level': '100'}), - (self.call.battery._HardwareSetCharging(False)), - (self.call.battery._HardwareSetCharging(True)), - (self.call.battery.GetBatteryInfo(), {'level': '50'})): + with self.assertCalls((self.call.battery.GetBatteryInfo(), { + 'level': '100' + }), (self.call.battery._HardwareSetCharging(False)), + (self.call.battery._HardwareSetCharging(True)), + (self.call.battery.GetBatteryInfo(), { + 'level': '50' + })): self.battery._DischargeDevice(1) @mock.patch('time.sleep', mock.Mock()) def testDischargeDevice_takeslong(self): - with self.assertCalls( - (self.call.battery.GetBatteryInfo(), {'level': '100'}), - (self.call.battery._HardwareSetCharging(False)), - (self.call.battery._HardwareSetCharging(True)), - (self.call.battery.GetBatteryInfo(), {'level': '100'}), - (self.call.battery._HardwareSetCharging(False)), - (self.call.battery._HardwareSetCharging(True)), - (self.call.battery.GetBatteryInfo(), {'level': '99'}), - (self.call.battery._HardwareSetCharging(False)), - (self.call.battery._HardwareSetCharging(True)), - (self.call.battery.GetBatteryInfo(), {'level': '98'}), - (self.call.battery._HardwareSetCharging(False)), - (self.call.battery._HardwareSetCharging(True)), - (self.call.battery.GetBatteryInfo(), {'level': '97'})): + with self.assertCalls((self.call.battery.GetBatteryInfo(), { + 'level': '100' + }), (self.call.battery._HardwareSetCharging(False)), + (self.call.battery._HardwareSetCharging(True)), + (self.call.battery.GetBatteryInfo(), { + 'level': '100' + }), (self.call.battery._HardwareSetCharging(False)), + (self.call.battery._HardwareSetCharging(True)), + (self.call.battery.GetBatteryInfo(), { + 'level': '99' + }), (self.call.battery._HardwareSetCharging(False)), + (self.call.battery._HardwareSetCharging(True)), + (self.call.battery.GetBatteryInfo(), { + 'level': '98' + }), (self.call.battery._HardwareSetCharging(False)), + (self.call.battery._HardwareSetCharging(True)), + (self.call.battery.GetBatteryInfo(), { + 'level': '97' + })): self.battery._DischargeDevice(3) @mock.patch('time.sleep', mock.Mock()) def testDischargeDevice_dischargeTooClose(self): - with self.assertCalls( - (self.call.battery.GetBatteryInfo(), {'level': '100'})): + with self.assertCalls((self.call.battery.GetBatteryInfo(), { + 'level': '100' + })): self.battery._DischargeDevice(99) @mock.patch('time.sleep', mock.Mock()) def testDischargeDevice_percentageOutOfBounds(self): - with self.assertCalls( - (self.call.battery.GetBatteryInfo(), {'level': '100'})): + with self.assertCalls((self.call.battery.GetBatteryInfo(), { + 'level': '100' + })): with self.assertRaises(ValueError): self.battery._DischargeDevice(100) - with self.assertCalls( - (self.call.battery.GetBatteryInfo(), {'level': '100'})): + with self.assertCalls((self.call.battery.GetBatteryInfo(), { + 'level': '100' + })): with self.assertRaises(ValueError): self.battery._DischargeDevice(0) class BatteryUtilsGetBatteryInfoTest(BatteryUtilsTest): - def testGetBatteryInfo_normal(self): with self.assertCalls( - (self.call.device.RunShellCommand( - ['dumpsys', 'battery'], check_return=True), - [ - 'Current Battery Service state:', - ' AC powered: false', - ' USB powered: true', - ' level: 100', - ' temperature: 321', - ])): - self.assertEquals( - { - 'AC powered': 'false', - 'USB powered': 'true', - 'level': '100', - 'temperature': '321', - }, - self.battery.GetBatteryInfo()) + (self.call.device.RunShellCommand(['dumpsys', 'battery'], + check_return=True), [ + 'Current Battery Service state:', + ' AC powered: false', + ' USB powered: true', + ' level: 100', + ' temperature: 321', + ])): + self.assertEquals({ + 'AC powered': 'false', + 'USB powered': 'true', + 'level': '100', + 'temperature': '321', + }, self.battery.GetBatteryInfo()) def testGetBatteryInfo_nothing(self): with self.assertCalls( - (self.call.device.RunShellCommand( - ['dumpsys', 'battery'], check_return=True), [])): + (self.call.device.RunShellCommand(['dumpsys', 'battery'], + check_return=True), [])): self.assertEquals({}, self.battery.GetBatteryInfo()) class BatteryUtilsGetChargingTest(BatteryUtilsTest): - def testGetCharging_usb(self): - with self.assertCall( - self.call.battery.GetBatteryInfo(), {'USB powered': 'true'}): + with self.assertCall(self.call.battery.GetBatteryInfo(), + {'USB powered': 'true'}): self.assertTrue(self.battery.GetCharging()) def testGetCharging_usbFalse(self): - with self.assertCall( - self.call.battery.GetBatteryInfo(), {'USB powered': 'false'}): + with self.assertCall(self.call.battery.GetBatteryInfo(), + {'USB powered': 'false'}): self.assertFalse(self.battery.GetCharging()) def testGetCharging_ac(self): - with self.assertCall( - self.call.battery.GetBatteryInfo(), {'AC powered': 'true'}): + with self.assertCall(self.call.battery.GetBatteryInfo(), + {'AC powered': 'true'}): self.assertTrue(self.battery.GetCharging()) def testGetCharging_wireless(self): - with self.assertCall( - self.call.battery.GetBatteryInfo(), {'Wireless powered': 'true'}): + with self.assertCall(self.call.battery.GetBatteryInfo(), + {'Wireless powered': 'true'}): self.assertTrue(self.battery.GetCharging()) def testGetCharging_unknown(self): - with self.assertCall( - self.call.battery.GetBatteryInfo(), {'level': '42'}): + with self.assertCall(self.call.battery.GetBatteryInfo(), {'level': '42'}): self.assertFalse(self.battery.GetCharging()) class BatteryUtilsLetBatteryCoolToTemperatureTest(BatteryUtilsTest): - @mock.patch('time.sleep', mock.Mock()) def testLetBatteryCoolToTemperature_startUnder(self): self.battery._cache['profile'] = self._NEXUS_6 - with self.assertCalls( - (self.call.battery.EnableBatteryUpdates(), []), - (self.call.battery.GetBatteryInfo(), {'temperature': '500'})): + with self.assertCalls((self.call.battery.EnableBatteryUpdates(), []), + (self.call.battery.GetBatteryInfo(), { + 'temperature': '500' + })): self.battery.LetBatteryCoolToTemperature(600) @mock.patch('time.sleep', mock.Mock()) def testLetBatteryCoolToTemperature_startOver(self): self.battery._cache['profile'] = self._NEXUS_6 - with self.assertCalls( - (self.call.battery.EnableBatteryUpdates(), []), - (self.call.battery.GetBatteryInfo(), {'temperature': '500'}), - (self.call.battery.GetBatteryInfo(), {'temperature': '400'})): + with self.assertCalls((self.call.battery.EnableBatteryUpdates(), []), + (self.call.battery.GetBatteryInfo(), { + 'temperature': '500' + }), (self.call.battery.GetBatteryInfo(), { + 'temperature': '400' + })): self.battery.LetBatteryCoolToTemperature(400) @mock.patch('time.sleep', mock.Mock()) def testLetBatteryCoolToTemperature_nexus5Hot(self): self.battery._cache['profile'] = self._NEXUS_5 - with self.assertCalls( - (self.call.battery.EnableBatteryUpdates(), []), - (self.call.battery.GetBatteryInfo(), {'temperature': '500'}), - (self.call.battery._DischargeDevice(1), []), - (self.call.battery.GetBatteryInfo(), {'temperature': '400'})): + with self.assertCalls((self.call.battery.EnableBatteryUpdates(), []), + (self.call.battery.GetBatteryInfo(), { + 'temperature': '500' + }), (self.call.battery._DischargeDevice(1), []), + (self.call.battery.GetBatteryInfo(), { + 'temperature': '400' + })): self.battery.LetBatteryCoolToTemperature(400) @mock.patch('time.sleep', mock.Mock()) def testLetBatteryCoolToTemperature_nexus5Cool(self): self.battery._cache['profile'] = self._NEXUS_5 - with self.assertCalls( - (self.call.battery.EnableBatteryUpdates(), []), - (self.call.battery.GetBatteryInfo(), {'temperature': '400'})): + with self.assertCalls((self.call.battery.EnableBatteryUpdates(), []), + (self.call.battery.GetBatteryInfo(), { + 'temperature': '400' + })): self.battery.LetBatteryCoolToTemperature(400) class BatteryUtilsSupportsFuelGaugeTest(BatteryUtilsTest): - def testSupportsFuelGauge_false(self): self.battery._cache['profile'] = self._NEXUS_5 self.assertFalse(self.battery.SupportsFuelGauge()) @@ -459,7 +500,6 @@ class BatteryUtilsSupportsFuelGaugeTest(BatteryUtilsTest): class BatteryUtilsGetFuelGaugeChargeCounterTest(BatteryUtilsTest): - def testGetFuelGaugeChargeCounter_noFuelGauge(self): self.battery._cache['profile'] = self._NEXUS_5 with self.assertRaises(device_errors.CommandFailedError): @@ -467,21 +507,19 @@ class BatteryUtilsGetFuelGaugeChargeCounterTest(BatteryUtilsTest): def testGetFuelGaugeChargeCounter_fuelGaugePresent(self): self.battery._cache['profile'] = self._NEXUS_6 - with self.assertCalls( - (self.call.battery.SupportsFuelGauge(), True), - (self.call.device.ReadFile(mock.ANY), '123')): + with self.assertCalls((self.call.battery.SupportsFuelGauge(), True), + (self.call.device.ReadFile(mock.ANY), '123')): self.assertEqual(self.battery.GetFuelGaugeChargeCounter(), 123) class BatteryUtilsSetCharging(BatteryUtilsTest): - @mock.patch('time.sleep', mock.Mock()) def testSetCharging_softwareSetTrue(self): self.battery._cache['profile'] = self._NEXUS_6 with self.assertCalls( (self.call.battery.GetCharging(), False), - (self.call.device.RunShellCommand( - ['dumpsys', 'battery', 'reset'], check_return=True), []), + (self.call.device.RunShellCommand(['dumpsys', 'battery', 'reset'], + check_return=True), []), (self.call.battery.GetCharging(), False), (self.call.device.RunShellCommand( ['dumpsys', 'battery'], check_return=True), ['UPDATES STOPPED']), @@ -504,18 +542,16 @@ class BatteryUtilsSetCharging(BatteryUtilsTest): @mock.patch('time.sleep', mock.Mock()) def testSetCharging_hardwareSetTrue(self): self.battery._cache['profile'] = self._NEXUS_5 - with self.assertCalls( - (self.call.battery.GetCharging(), False), - (self.call.battery._HardwareSetCharging(True))): + with self.assertCalls((self.call.battery.GetCharging(), False), + (self.call.battery._HardwareSetCharging(True))): self.battery.SetCharging(True) @mock.patch('time.sleep', mock.Mock()) def testSetCharging_hardwareSetFalse(self): self.battery._cache['profile'] = self._NEXUS_5 - with self.assertCalls( - (self.call.battery.GetCharging(), True), - (self.call.battery._ClearPowerData(), True), - (self.call.battery._HardwareSetCharging(False))): + with self.assertCalls((self.call.battery.GetCharging(), True), + (self.call.battery._ClearPowerData(), True), + (self.call.battery._HardwareSetCharging(False))): self.battery.SetCharging(False) def testSetCharging_expectedStateAlreadyTrue(self): @@ -528,15 +564,13 @@ class BatteryUtilsSetCharging(BatteryUtilsTest): class BatteryUtilsPowerMeasurement(BatteryUtilsTest): - def testPowerMeasurement_hardware(self): self.battery._cache['profile'] = self._NEXUS_5 - with self.assertCalls( - (self.call.battery.GetCharging(), True), - (self.call.battery._ClearPowerData(), True), - (self.call.battery._HardwareSetCharging(False)), - (self.call.battery.GetCharging(), False), - (self.call.battery._HardwareSetCharging(True))): + with self.assertCalls((self.call.battery.GetCharging(), True), + (self.call.battery._ClearPowerData(), True), + (self.call.battery._HardwareSetCharging(False)), + (self.call.battery.GetCharging(), False), + (self.call.battery._HardwareSetCharging(True))): with self.battery.PowerMeasurement(): pass @@ -552,8 +586,8 @@ class BatteryUtilsPowerMeasurement(BatteryUtilsTest): ['dumpsys', 'battery', 'set', 'usb', '0'], check_return=True), []), (self.call.battery.GetCharging(), False), (self.call.battery.GetCharging(), False), - (self.call.device.RunShellCommand( - ['dumpsys', 'battery', 'reset'], check_return=True), []), + (self.call.device.RunShellCommand(['dumpsys', 'battery', 'reset'], + check_return=True), []), (self.call.battery.GetCharging(), False), (self.call.device.RunShellCommand( ['dumpsys', 'battery'], check_return=True), ['UPDATES STOPPED']), @@ -563,30 +597,25 @@ class BatteryUtilsPowerMeasurement(BatteryUtilsTest): class BatteryUtilsDiscoverDeviceProfile(BatteryUtilsTest): - def testDiscoverDeviceProfile_known(self): - with self.patch_call(self.call.device.product_model, - return_value='Nexus 4'): + with self.patch_call( + self.call.device.product_model, return_value='Nexus 4'): self.battery._DiscoverDeviceProfile() self.assertListEqual(self.battery._cache['profile']['name'], ["Nexus 4"]) def testDiscoverDeviceProfile_unknown(self): - with self.patch_call(self.call.device.product_model, - return_value='Other'): + with self.patch_call(self.call.device.product_model, return_value='Other'): self.battery._DiscoverDeviceProfile() self.assertListEqual(self.battery._cache['profile']['name'], []) class BatteryUtilsClearPowerData(BatteryUtilsTest): - def testClearPowerData_preL(self): - with self.patch_call(self.call.device.build_version_sdk, - return_value=20): + with self.patch_call(self.call.device.build_version_sdk, return_value=20): self.assertFalse(self.battery._ClearPowerData()) def testClearPowerData_clearedL(self): - with self.patch_call(self.call.device.build_version_sdk, - return_value=22): + with self.patch_call(self.call.device.build_version_sdk, return_value=22): with self.assertCalls( (self.call.device.RunShellCommand( ['dumpsys', 'battery', 'set', 'usb', '1'], check_return=True), @@ -597,15 +626,14 @@ class BatteryUtilsClearPowerData(BatteryUtilsTest): ['dumpsys', 'batterystats', '--reset'], check_return=True), []), (self.call.device.RunShellCommand( ['dumpsys', 'batterystats', '--charged', '-c'], - check_return=True, large_output=True), []), - (self.call.device.RunShellCommand( - ['dumpsys', 'battery', 'reset'], check_return=True), [])): + check_return=True, + large_output=True), []), (self.call.device.RunShellCommand( + ['dumpsys', 'battery', 'reset'], check_return=True), [])): self.assertTrue(self.battery._ClearPowerData()) @mock.patch('time.sleep', mock.Mock()) def testClearPowerData_notClearedL(self): - with self.patch_call(self.call.device.build_version_sdk, - return_value=22): + with self.patch_call(self.call.device.build_version_sdk, return_value=22): with self.assertCalls( (self.call.device.RunShellCommand( ['dumpsys', 'battery', 'set', 'usb', '1'], check_return=True), @@ -616,28 +644,28 @@ class BatteryUtilsClearPowerData(BatteryUtilsTest): ['dumpsys', 'batterystats', '--reset'], check_return=True), []), (self.call.device.RunShellCommand( ['dumpsys', 'batterystats', '--charged', '-c'], - check_return=True, large_output=True), - ['9,1000,l,pwi,uid,0.0327']), + check_return=True, + large_output=True), ['9,1000,l,pwi,uid,0.0327']), (self.call.device.RunShellCommand( ['dumpsys', 'batterystats', '--reset'], check_return=True), []), (self.call.device.RunShellCommand( ['dumpsys', 'batterystats', '--charged', '-c'], - check_return=True, large_output=True), - ['9,1000,l,pwi,uid,0.0327']), + check_return=True, + large_output=True), ['9,1000,l,pwi,uid,0.0327']), (self.call.device.RunShellCommand( ['dumpsys', 'batterystats', '--reset'], check_return=True), []), (self.call.device.RunShellCommand( ['dumpsys', 'batterystats', '--charged', '-c'], - check_return=True, large_output=True), - ['9,1000,l,pwi,uid,0.0327']), + check_return=True, + large_output=True), ['9,1000,l,pwi,uid,0.0327']), (self.call.device.RunShellCommand( ['dumpsys', 'batterystats', '--reset'], check_return=True), []), (self.call.device.RunShellCommand( ['dumpsys', 'batterystats', '--charged', '-c'], - check_return=True, large_output=True), - ['9,1000,l,pwi,uid,0.0']), - (self.call.device.RunShellCommand( - ['dumpsys', 'battery', 'reset'], check_return=True), [])): + check_return=True, + large_output=True), ['9,1000,l,pwi,uid,0.0']), + (self.call.device.RunShellCommand(['dumpsys', 'battery', 'reset'], + check_return=True), [])): self.battery._ClearPowerData() |