97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
"""gaokao-audit 端到端集成测试。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import importlib
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Any, cast
|
|
|
|
import pytest
|
|
|
|
_REPO_ROOT = Path(__file__).resolve().parents[1]
|
|
if str(_REPO_ROOT) not in sys.path:
|
|
sys.path.insert(0, str(_REPO_ROOT))
|
|
|
|
_AUDIT_CLI = importlib.import_module("skills.gaokao-audit.scripts.audit_cli")
|
|
_REPORT_GENERATOR = importlib.import_module(
|
|
"skills.gaokao-audit.scripts.report_generator"
|
|
)
|
|
_ReportGeneratorBase = cast(type[Any], _REPORT_GENERATOR.ReportGenerator)
|
|
|
|
SAMPLE_PLAN = (
|
|
_REPO_ROOT / "skills" / "gaokao-audit" / "tests" / "fixtures" / "sample_xianyu.txt"
|
|
)
|
|
|
|
|
|
class _CaptureReportGenerator(_ReportGeneratorBase): # type: ignore[valid-type, misc]
|
|
last_html: str = ""
|
|
|
|
def __init__(self, **kwargs: object) -> None:
|
|
super().__init__(
|
|
now_text=lambda: "2026-06-12 23:40",
|
|
report_id_factory=lambda: "AUDIT-E2E-001",
|
|
**kwargs,
|
|
)
|
|
|
|
def generate_pdf(self, result, output_path: str, **kwargs: object) -> str:
|
|
html = self.render_html(result, **kwargs)
|
|
type(self).last_html = html
|
|
target = Path(output_path)
|
|
target.parent.mkdir(parents=True, exist_ok=True)
|
|
target.write_bytes(b"%PDF-1.4\ne2e fake pdf\n")
|
|
return str(target)
|
|
|
|
|
|
def _extract_json(stdout: str) -> dict:
|
|
lines = stdout.splitlines()
|
|
for index, line in enumerate(lines):
|
|
if line.startswith("{"):
|
|
return json.loads("\n".join(lines[index:]))
|
|
raise AssertionError(f"stdout 中未找到 JSON 输出: {stdout}")
|
|
|
|
|
|
def test_audit_cli_end_to_end_generates_pdf_and_report_content(
|
|
tmp_path: Path,
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
output_path = tmp_path / "audit-e2e.pdf"
|
|
monkeypatch.setattr(_AUDIT_CLI, "ReportGenerator", _CaptureReportGenerator)
|
|
|
|
exit_code = _AUDIT_CLI.main([
|
|
str(SAMPLE_PLAN),
|
|
"--output",
|
|
str(output_path),
|
|
"--json",
|
|
])
|
|
captured = capsys.readouterr()
|
|
payload = _extract_json(captured.out)
|
|
rendered_html = _CaptureReportGenerator.last_html
|
|
|
|
assert exit_code == 0
|
|
assert output_path.exists()
|
|
assert output_path.read_bytes().startswith(b"%PDF-1.4")
|
|
|
|
assert "输入文件" in captured.out
|
|
assert str(output_path) in captured.out
|
|
assert "综合评分" in captured.out
|
|
|
|
assert payload["province"] == "湖南"
|
|
assert payload["candidate_score"] == 578
|
|
assert payload["source"] == "百度AI"
|
|
assert len(payload["volunteers"]) == 6
|
|
assert payload["policy_errors"] == []
|
|
assert payload["policy_serious_warnings"]
|
|
assert payload["crowd_risks"]
|
|
risk_schools = {item["school"] for item in payload["crowd_risks"]}
|
|
assert "湖南师范大学" in risk_schools
|
|
assert payload["overall_score"] < 100
|
|
|
|
assert "AUDIT-E2E-001" in rendered_html
|
|
assert "湖南 578分 物理+化学+生物" in rendered_html
|
|
assert any(school in rendered_html for school in risk_schools)
|
|
assert "免责声明" in rendered_html
|
|
assert "本审核仅供建议，最终填报以官方公布为准" in rendered_html
|