diff --git a/test/unit/webdriver/device/lock_test.py b/test/unit/webdriver/device/lock_test.py index 603efffb..0569405d 100644 --- a/test/unit/webdriver/device/lock_test.py +++ b/test/unit/webdriver/device/lock_test.py @@ -48,12 +48,20 @@ def test_islocked_true(self): assert driver.is_locked() is True @httpretty.activate - def test_unlock(self): + def test_unlock_when_locked(self): driver = android_w3c_driver() httpretty.register_uri(httpretty.POST, appium_command('/session/1234567890/execute/sync'), body='{"value": true}') assert isinstance(driver.unlock(), WebDriver) + assert get_httpretty_request_body(httpretty.latest_requests()[-4])['script'] == 'mobile: isLocked' assert get_httpretty_request_body(httpretty.last_request())['script'] == 'mobile: unlock' + @httpretty.activate + def test_unlock_when_unlocked(self): + driver = android_w3c_driver() + httpretty.register_uri(httpretty.POST, appium_command('/session/1234567890/execute/sync'), body='{"value": false}') + assert isinstance(driver.unlock(), WebDriver) + assert get_httpretty_request_body(httpretty.last_request())['script'] == 'mobile: isLocked' + class TestWebDriverLockIOS: @httpretty.activate @@ -85,12 +93,20 @@ def test_islocked_true(self): assert driver.is_locked() is True @httpretty.activate - def test_unlock(self): + def test_unlock_when_locked(self): driver = ios_w3c_driver() httpretty.register_uri(httpretty.POST, appium_command('/session/1234567890/execute/sync'), body='{"value": true}') assert isinstance(driver.unlock(), WebDriver) + assert get_httpretty_request_body(httpretty.latest_requests()[-4])['script'] == 'mobile: isLocked' assert get_httpretty_request_body(httpretty.last_request())['script'] == 'mobile: unlock' + @httpretty.activate + def test_unlock_when_unlocked(self): + driver = ios_w3c_driver() + httpretty.register_uri(httpretty.POST, appium_command('/session/1234567890/execute/sync'), body='{"value": false}') + assert isinstance(driver.unlock(), WebDriver) + assert get_httpretty_request_body(httpretty.last_request())['script'] == 'mobile: isLocked' + @httpretty.activate def test_touch_id(self): driver = ios_w3c_driver()