Support set_power_capacity

ki4070ma · ki4070ma · commit fb0aa7cbdec1 · 2019-04-29T00:18:55.000+09:00
diff --git a/appium/webdriver/extensions/power.py b/appium/webdriver/extensions/power.py
@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from selenium import webdriver
+from ..mobilecommand import MobileCommand as Command
+
+
+class Power(webdriver.Remote):
+
+    def set_power_capacity(self, percent):
+        """Emulate power capacity change on the connected emulator.
+
+        :Args:
+         - percent: The power capacity to be set
+
+        :Usage:
+            self.driver.set_power_capacity(50)
+        """
+        self.execute(Command.SET_POWER_CAPACITY, {'percent': percent})
+
+    def _addCommands(self):
+        self.command_executor._commands[Command.SET_POWER_CAPACITY] = \
+            ('POST', '/session/$sessionId/appium/device/power_capacity')
diff --git a/appium/webdriver/mobilecommand.py b/appium/webdriver/mobilecommand.py
@@ -77,3 +77,4 @@ class MobileCommand(object):
     COMPARE_IMAGES = 'compareImages'
     FINGER_PRINT = 'fingerPrint'
     SEND_SMS = 'sendSms'
+    SET_POWER_CAPACITY = 'setPowerCapacity'
diff --git a/appium/webdriver/webdriver.py b/appium/webdriver/webdriver.py
@@ -38,6 +38,7 @@
 from .extensions.hw_actions import HardwareActions
 from .extensions.location import Location
 from .extensions.network import Network
+from .extensions.power import Power
 from .extensions.remote_fs import RemoteFS
 from .extensions.screen_record import ScreenRecord
 from .extensions.search_context import AppiumSearchContext
@@ -117,6 +118,7 @@ class WebDriver(
     Keyboard,
     Location,
     Network,
+    Power,
     RemoteFS,
     ScreenRecord,
     Settings,
