Replace 'on' with AC_ON

ki4070ma · ki4070ma · commit c17df314f419 · 2019-04-29T02:35:06.000+09:00
diff --git a/test/unit/webdriver/device/power_test.py b/test/unit/webdriver/device/power_test.py
@@ -21,6 +21,7 @@
 import httpretty
 
 from appium.webdriver.webdriver import WebDriver
+from appium.webdriver.extensions.power import Power
 
 
 class TestWebDriverPower(object):
@@ -44,7 +45,7 @@ def test_set_power_ac(self):
             httpretty.POST,
             appium_command('/session/1234567890/appium/device/power_ac'),
         )
-        assert isinstance(driver.set_power_ac('on'), WebDriver) is True
+        assert isinstance(driver.set_power_ac(Power.AC_ON), WebDriver) is True
 
         d = get_httpretty_request_body(httpretty.last_request())
-        assert d['state'] == 'on'
+        assert d['state'] == Power.AC_ON
