Support set_power_ac

ki4070ma · ki4070ma · commit e8f53c7151be · 2019-04-29T00:48:01.000+09:00
diff --git a/appium/webdriver/extensions/power.py b/appium/webdriver/extensions/power.py
@@ -29,6 +29,20 @@ def set_power_capacity(self, percent):
         """
         self.execute(Command.SET_POWER_CAPACITY, {'percent': percent})
 
+    def set_power_ac(self, ac_state):
+        """Emulate power state change on the connected emulator.
+
+        :Args:
+         - state: The power ac state to be set
+
+        :Usage:
+            self.driver.set_power_state('on')
+            self.driver.set_power_state('off')
+        """
+        self.execute(Command.SET_POWER_AC, {'state': ac_state})
+
     def _addCommands(self):
         self.command_executor._commands[Command.SET_POWER_CAPACITY] = \
             ('POST', '/session/$sessionId/appium/device/power_capacity')
+        self.command_executor._commands[Command.SET_POWER_AC] = \
+            ('POST', '/session/$sessionId/appium/device/power_ac')
diff --git a/appium/webdriver/mobilecommand.py b/appium/webdriver/mobilecommand.py
@@ -78,3 +78,4 @@ class MobileCommand(object):
     FINGER_PRINT = 'fingerPrint'
     SEND_SMS = 'sendSms'
     SET_POWER_CAPACITY = 'setPowerCapacity'
+    SET_POWER_AC = 'setPowerAc'
