Add return value

ki4070ma · ki4070ma · commit 05199c65e3a1 · 2019-04-29T01:51:47.000+09:00
diff --git a/appium/webdriver/extensions/power.py b/appium/webdriver/extensions/power.py
@@ -28,6 +28,7 @@ def set_power_capacity(self, percent):
             self.driver.set_power_capacity(50)
         """
         self.execute(Command.SET_POWER_CAPACITY, {'percent': percent})
+        return self
 
     def set_power_ac(self, ac_state):
         """Emulate power state change on the connected emulator.
@@ -40,6 +41,7 @@ def set_power_ac(self, ac_state):
             self.driver.set_power_ac('on')
         """
         self.execute(Command.SET_POWER_AC, {'state': ac_state})
+        return self
 
     def _addCommands(self):
         self.command_executor._commands[Command.SET_POWER_CAPACITY] = \
diff --git a/appium/webdriver/extensions/sms.py b/appium/webdriver/extensions/sms.py
@@ -29,6 +29,7 @@ def send_sms(self, phone_number, message):
             self.driver.send_sms('555-123-4567', 'Hey lol')
         """
         self.execute(Command.SEND_SMS, {'phoneNumber': phone_number, 'message': message})
+        return self
 
     def _addCommands(self):
         self.command_executor._commands[Command.SEND_SMS] = \
diff --git a/test/unit/webdriver/device/power_test.py b/test/unit/webdriver/device/power_test.py
@@ -20,6 +20,8 @@
 
 import httpretty
 
+from appium.webdriver.webdriver import WebDriver
+
 
 class TestWebDriverPower(object):
 
@@ -30,7 +32,7 @@ def test_set_power_capacity(self):
             httpretty.POST,
             appium_command('/session/1234567890/appium/device/power_capacity'),
         )
-        assert driver.set_power_capacity(50) is None
+        assert isinstance(driver.set_power_capacity(50), WebDriver) is True
 
         d = get_httpretty_request_body(httpretty.last_request())
         assert d['percent'] == 50
@@ -42,7 +44,7 @@ def test_set_power_ac(self):
             httpretty.POST,
             appium_command('/session/1234567890/appium/device/power_ac'),
         )
-        assert driver.set_power_ac('on') is None
+        assert isinstance(driver.set_power_ac('on'), WebDriver) is True
 
         d = get_httpretty_request_body(httpretty.last_request())
         assert d['state'] == 'on'
diff --git a/test/unit/webdriver/device/sms_test.py b/test/unit/webdriver/device/sms_test.py
@@ -20,6 +20,8 @@
 
 import httpretty
 
+from appium.webdriver.webdriver import WebDriver
+
 
 class TestWebDriverSms(object):
 
@@ -30,7 +32,7 @@ def test_send_sms(self):
             httpretty.POST,
             appium_command('/session/1234567890/appium/device/send_sms'),
         )
-        assert driver.send_sms('555-123-4567', 'Hey lol') is None
+        assert isinstance(driver.send_sms('555-123-4567', 'Hey lol'), WebDriver) is True
 
         d = get_httpretty_request_body(httpretty.last_request())
         assert d['phoneNumber'] == '555-123-4567'
