mirror of
https://git.zavage.net/Zavage-Software/wabot.git
synced 2026-07-21 13:06:08 -06:00
100 lines
2.7 KiB
Python
100 lines
2.7 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
from selenium.common.exceptions import WebDriverException
|
|
|
|
from wabot.browser import Browser
|
|
from wabot.pacing import HumanPacing, NoPacing
|
|
from wabot.page import Page
|
|
|
|
|
|
class Login(Page):
|
|
elements = {"username": ("el", ("id", "username"))}
|
|
|
|
def do_login(self):
|
|
return "logged-in"
|
|
|
|
def explode(self):
|
|
raise WebDriverException("browser gone")
|
|
|
|
|
|
class Unverifiable(Page):
|
|
def verify(self):
|
|
return False
|
|
|
|
|
|
@pytest.fixture
|
|
def bot():
|
|
return Browser(MagicMock(name="driver"), pacing=NoPacing())
|
|
|
|
|
|
class TestConstruction:
|
|
def test_good_flag_starts_true(self, bot):
|
|
assert bot.good is True
|
|
|
|
def test_default_pacing_is_human(self):
|
|
assert isinstance(Browser(MagicMock()).pacing, HumanPacing)
|
|
|
|
|
|
class TestSetPage:
|
|
def test_sets_and_returns_true(self, bot):
|
|
assert bot.set_page(Login) is True
|
|
assert isinstance(bot.page, Login)
|
|
|
|
def test_failed_verify_refuses_switch(self, bot):
|
|
bot.set_page(Login)
|
|
assert bot.set_page(Unverifiable) is False
|
|
assert isinstance(bot.page, Login) # unchanged
|
|
|
|
|
|
class TestDelegation:
|
|
def test_getattr_delegates_to_page(self, bot):
|
|
bot.set_page(Login)
|
|
assert bot.do_login() == "logged-in"
|
|
|
|
def test_getitem_delegates_to_page(self, bot):
|
|
bot.set_page(Login)
|
|
el = bot["username"]
|
|
assert el.name == "username"
|
|
|
|
def test_getattr_without_page_raises(self, bot):
|
|
with pytest.raises(AttributeError):
|
|
_ = bot.do_login
|
|
|
|
|
|
class TestPerform:
|
|
def test_returns_method_result(self, bot):
|
|
bot.set_page(Login)
|
|
assert bot.perform("do_login") == "logged-in"
|
|
|
|
def test_missing_method_returns_none(self, bot):
|
|
bot.set_page(Login)
|
|
assert bot.perform("nope") is None
|
|
|
|
def test_webdriver_exception_flips_good(self, bot):
|
|
bot.set_page(Login)
|
|
assert bot.perform("explode") is None
|
|
assert bot.good is False
|
|
|
|
def test_broken_state_refuses_actions_until_reset(self, bot):
|
|
bot.set_page(Login)
|
|
bot.perform("explode")
|
|
assert bot.perform("do_login") is None # refused
|
|
assert bot.do_login() is None # __getattr__ path also refused
|
|
bot.reset()
|
|
assert bot.perform("do_login") == "logged-in"
|
|
|
|
|
|
class TestQuit:
|
|
def test_quit_quits_driver_and_removes_session(self):
|
|
driver, store = MagicMock(), MagicMock()
|
|
bot = Browser(driver, session_name="s1", store=store)
|
|
bot.quit()
|
|
driver.quit.assert_called_once_with()
|
|
store.remove.assert_called_once_with("s1")
|
|
|
|
def test_quit_without_session_only_quits(self):
|
|
driver = MagicMock()
|
|
Browser(driver).quit()
|
|
driver.quit.assert_called_once_with()
|