124 lines
4.7 KiB
Python
124 lines
4.7 KiB
Python
"""Seam 1 page check for the AI desktop page: the no-AI dashboard.html with the plain-HTML
|
|
AI sections (rpa-dashboard-ai/templates/desktop/ai_sections.html) pasted in before </body>.
|
|
|
|
Page under test: set PAGE_CHECK_AI_DESKTOP_DIR to a generated dashboard folder (the one with
|
|
dashboard.html and server_stub.py). Without it, the no-AI template is filled with the no-AI
|
|
check's TEST_FILL and the AI sections are pasted in, the way the skill does it.
|
|
|
|
The page sees a fake Qt bridge, as in the no-AI check: each slot call lands in window.__calls.
|
|
The AI slot, ai_request(id, method, path, body), goes to the review backend's dispatch() in
|
|
this process, and the reply comes back as an `ai_response` envelope, exactly what
|
|
ai_bridge.py's slot does. So the approve gate under test is the Python one. Tests set run and
|
|
review state on the backend directly (the `stub` fixture). Light, dark and narrow for every case.
|
|
"""
|
|
import importlib.util
|
|
import json
|
|
import os
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
NO_AI = ROOT / "rpa-dashboard-no-ai" / "templates" / "dashboard.html"
|
|
AI = ROOT / "rpa-dashboard-ai" / "templates"
|
|
TEST_FILL = {
|
|
"APP_NAME": "Test Extractor",
|
|
"APP_SUBTITLE": "Page check",
|
|
"LOGO_SRC": "",
|
|
"API_BASE": "",
|
|
"STEPS_JSON": json.dumps(["Initializing", "Login", "Process items", "Completed"]),
|
|
"REPORT_COLUMNS_JSON": json.dumps([{"key": "item", "label": "Item"}, {"key": "status", "label": "Status"}]),
|
|
}
|
|
FAKE_BRIDGE = """
|
|
window.__calls = [];
|
|
window.qt = {webChannelTransport: {}};
|
|
window.QWebChannel = function (_transport, ready) {
|
|
const pyBridge = new Proxy({}, {get: (_o, slot) => (...args) => {
|
|
window.__calls.push([slot, ...args]);
|
|
if (slot === 'ai_request') window.__aiRequest(...args).then(window.__pyMsg);
|
|
}});
|
|
ready({objects: {pyBridge}});
|
|
};
|
|
"""
|
|
MODES = {"light": (1280, 820, False), "dark": (1280, 820, True), "narrow": (800, 900, False)}
|
|
WIRED = os.environ.get("PAGE_CHECK_WIRED") == "1"
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_data():
|
|
if WIRED:
|
|
pytest.skip("page is wired to a bot, so it has no sample data")
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def site(tmp_path_factory):
|
|
"""The dashboard folder under test: dashboard.html, server_stub.py and bands.py."""
|
|
if os.environ.get("PAGE_CHECK_AI_DESKTOP_DIR"):
|
|
return Path(os.environ["PAGE_CHECK_AI_DESKTOP_DIR"]).resolve()
|
|
html = re.sub(r"\{\{(\w+)\}\}", lambda m: TEST_FILL[m[1]], NO_AI.read_text(encoding="utf-8"))
|
|
if WIRED:
|
|
html = html.replace("const SAMPLE_DATA = true", "const SAMPLE_DATA = false")
|
|
sections = (AI / "desktop" / "ai_sections.html").read_text(encoding="utf-8")
|
|
out = tmp_path_factory.mktemp("desktop")
|
|
(out / "dashboard.html").write_text(html.replace("</body>", sections + "\n</body>"), encoding="utf-8")
|
|
for f in ["server_stub.py", "bands.py"]:
|
|
(out / f).write_bytes((AI / "web" / f).read_bytes())
|
|
return out
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def page_url(site):
|
|
return (site / "dashboard.html").as_uri()
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def stub_module(site):
|
|
sys.path.insert(0, str(site)) # the stub imports bands.py from its own folder
|
|
spec = importlib.util.spec_from_file_location("desktop_server_stub", site / "server_stub.py")
|
|
mod = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(mod)
|
|
return mod
|
|
|
|
|
|
@pytest.fixture
|
|
def stub(stub_module):
|
|
"""The review backend the bridge talks to, reset to its sample data."""
|
|
stub_module.reset()
|
|
return stub_module
|
|
|
|
|
|
def ai_request(stub, req_id, method, path, body):
|
|
"""What ai_bridge.py's ai_request slot does: dispatch, then answer with one envelope."""
|
|
status, payload = stub.dispatch(method, path, json.loads(body or "{}"))
|
|
return json.dumps({"event": "ai_response", "data": {"id": req_id, "status": status, "body": payload}})
|
|
|
|
|
|
@pytest.fixture(params=list(MODES))
|
|
def page(request, browser, page_url, stub):
|
|
width, height, dark = MODES[request.param]
|
|
ctx = browser.new_context(viewport={"width": width, "height": height})
|
|
ctx.expose_function("__aiRequest", lambda i, method, path, body: ai_request(stub, i, method, path, body))
|
|
ctx.add_init_script(FAKE_BRIDGE)
|
|
pg = ctx.new_page()
|
|
pg.set_default_timeout(5000)
|
|
errors = []
|
|
pg.on("pageerror", lambda e: errors.append(e))
|
|
pg.goto(page_url)
|
|
assert not errors, f"page script errors: {errors}"
|
|
if dark:
|
|
push_to(pg, "apply_dark", {"on": True})
|
|
yield pg
|
|
assert not errors, f"page script errors: {errors}"
|
|
ctx.close()
|
|
|
|
|
|
def push_to(pg, event, data):
|
|
pg.evaluate("([event, data]) => window.__pyMsg(JSON.stringify({event, data}))", [event, data])
|
|
|
|
|
|
@pytest.fixture
|
|
def push(page):
|
|
return lambda event, data: push_to(page, event, data)
|