mirror of
https://git.zavage.net/Zavage-Software/wabot.git
synced 2026-07-21 13:06:08 -06:00
test: flagship cross-process browser pickup + destroy cleanup
This commit is contained in:
parent
8fd9bd1fa0
commit
2dd4d8c13c
71
tests/integration/test_persistence.py
Normal file
71
tests/integration/test_persistence.py
Normal file
@ -0,0 +1,71 @@
|
|||||||
|
"""The flagship feature: a second Python PROCESS picks up a saved browser."""
|
||||||
|
|
||||||
|
import json
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
import textwrap
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
import wabot
|
||||||
|
|
||||||
|
from .conftest import BROWSERS
|
||||||
|
|
||||||
|
REATTACH_SCRIPT = textwrap.dedent(
|
||||||
|
"""
|
||||||
|
import json, sys
|
||||||
|
import wabot
|
||||||
|
from wabot.sessions import SessionStore
|
||||||
|
|
||||||
|
store = SessionStore(path=sys.argv[1])
|
||||||
|
bot = wabot.browser(session=sys.argv[2], pacing=wabot.NoPacing(), store=store)
|
||||||
|
value = bot.driver.find_element("id", "username").get_attribute("value")
|
||||||
|
print(json.dumps({"url": bot.driver.current_url, "username_value": value}))
|
||||||
|
"""
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize("browser_name", BROWSERS)
|
||||||
|
def test_second_process_picks_up_saved_browser(browser_name, site_url, store, tmp_path):
|
||||||
|
bot = wabot.browser(
|
||||||
|
session="flagship", browser=browser_name, headless=True,
|
||||||
|
pacing=wabot.NoPacing(), store=store,
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
bot.driver.get(f"{site_url}/login.html")
|
||||||
|
bot.driver.find_element("id", "username").send_keys("persisted-value")
|
||||||
|
|
||||||
|
# simulate "the process exits": drop our handle without quitting
|
||||||
|
record = store.get("flagship")
|
||||||
|
assert record is not None and record.service_pid is not None
|
||||||
|
|
||||||
|
result = subprocess.run(
|
||||||
|
[sys.executable, "-c", REATTACH_SCRIPT, str(store.path), "flagship"],
|
||||||
|
capture_output=True, text=True, timeout=120, check=True,
|
||||||
|
)
|
||||||
|
payload = json.loads(result.stdout.strip().splitlines()[-1])
|
||||||
|
assert payload["url"] == f"{site_url}/login.html"
|
||||||
|
assert payload["username_value"] == "persisted-value"
|
||||||
|
finally:
|
||||||
|
assert wabot.destroy("flagship", store=store) is True
|
||||||
|
|
||||||
|
|
||||||
|
def test_destroyed_session_is_really_gone(site_url, store):
|
||||||
|
from wabot.hosts import service_alive
|
||||||
|
|
||||||
|
bot = wabot.browser(
|
||||||
|
session="doomed", browser="chromium", headless=True,
|
||||||
|
pacing=wabot.NoPacing(), store=store,
|
||||||
|
)
|
||||||
|
bot.driver.get(f"{site_url}/login.html")
|
||||||
|
record = store.get("doomed")
|
||||||
|
assert wabot.destroy("doomed", store=store) is True
|
||||||
|
assert store.get("doomed") is None
|
||||||
|
assert wabot.destroy("doomed", store=store) is False
|
||||||
|
# the managed driver service was SIGTERMed
|
||||||
|
import time
|
||||||
|
|
||||||
|
deadline = time.monotonic() + 10
|
||||||
|
while time.monotonic() < deadline and service_alive(record.executor_url):
|
||||||
|
time.sleep(0.2)
|
||||||
|
assert not service_alive(record.executor_url)
|
||||||
Loading…
Reference in New Issue
Block a user