ilia
c91cd221d1
CI / Lint + tests (push) Has been cancelled
Ship a shared markdown/prompt core used by a Brave/Chrome MV3 extension and a Python automation API, with pytest coverage, CI, and packaging smoke.
72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
"""Unit-style tests for the shared prompt.js formatter (via page.evaluate)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from context_extractor.session import _build_prompt_script, _read_js
|
|
|
|
CORE = Path(__file__).resolve().parents[2] / "extension" / "core"
|
|
|
|
|
|
def test_shared_js_files_exist_and_match_package():
|
|
assert (CORE / "dom.js").is_file()
|
|
assert (CORE / "prompt.js").is_file()
|
|
# package-side copies/symlinks must resolve to the same source
|
|
assert "function extractMarkdown" in _read_js("dom.js")
|
|
assert "function buildAIPrompt" in _read_js("prompt.js")
|
|
|
|
|
|
def test_build_ai_prompt_script_formats_errors_and_failures(page):
|
|
page.goto("about:blank")
|
|
meta = {
|
|
"url": "https://example.test/page",
|
|
"title": "T",
|
|
"ts": 1_700_000_000_000,
|
|
"selector": "body",
|
|
}
|
|
store = {
|
|
"console": [{"level": "warn", "ts": 1, "msg": "careful"}],
|
|
"errors": [{
|
|
"type": "error", "ts": 2, "msg": "boom",
|
|
"source": "a.js", "line": 3, "col": 4, "stack": "Error: boom\n at x",
|
|
}],
|
|
"network": [{
|
|
"type": "fetch", "method": "GET", "url": "https://example.test/api",
|
|
"status": 500, "ts": 3, "duration": 12, "error": None,
|
|
}],
|
|
}
|
|
script = _build_prompt_script(meta, "# Hello", store)
|
|
out = page.evaluate(script)
|
|
assert "# Page Context" in out
|
|
assert "https://example.test/page" in out
|
|
assert "## Page Content" in out
|
|
assert "# Hello" in out
|
|
assert "## JavaScript Errors" in out
|
|
assert "boom" in out
|
|
assert "## Console Errors & Warnings" in out
|
|
assert "[WARN] careful" in out
|
|
assert "## Failed Requests" in out
|
|
assert "GET 500 https://example.test/api" in out
|
|
|
|
|
|
def test_build_ai_prompt_truncates_huge_markdown_by_default(page):
|
|
"""Regression test: a JS-heavy SPA (e.g. LinkedIn) can extract 700k+ chars
|
|
of body content. The prompt must not blindly include all of it."""
|
|
page.goto("about:blank")
|
|
meta = {"url": "https://example.test/huge", "title": "T", "ts": 1, "selector": "body"}
|
|
huge_markdown = "x" * 50_000
|
|
script = _build_prompt_script(meta, huge_markdown, {"console": [], "errors": [], "network": []})
|
|
out = page.evaluate(script)
|
|
assert "truncated to 20,000 of 50,000 chars" in out
|
|
assert len(out) < 21_000 # header/footer overhead + capped body, nowhere near 50k
|
|
|
|
|
|
def test_build_ai_prompt_respects_custom_max_chars(page):
|
|
page.goto("about:blank")
|
|
meta = {"url": "https://example.test/huge", "title": "T", "ts": 1, "selector": "body"}
|
|
markdown = "y" * 1000
|
|
script = _build_prompt_script(meta, markdown, {"console": [], "errors": [], "network": []}, max_chars=200)
|
|
out = page.evaluate(script)
|
|
assert "truncated to 200 of 1,000 chars" in out
|