gaokao-volunteer-system/tests/test_t5_e2e_workflows.py

"""T5.1 端到端业务场景测试。

覆盖 5 条主链路：
1. 咨询 -> 方案生成
2. 审核 -> 报告
3. 订单 -> 交付
4. 升级流程
5. 数据溯源展示
"""

from __future__ import annotations

import importlib
import importlib.util
import json
import os
import subprocess
import sys
from pathlib import Path
from typing import Any, cast

import pytest

PROJECT_ROOT = Path(__file__).resolve().parents[1]
if str(PROJECT_ROOT) not in sys.path:
    sys.path.insert(0, str(PROJECT_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)
_TRACE_CLI = importlib.import_module("data.crowd_db.cli")

SAMPLE_PLAN = (
    PROJECT_ROOT
    / "skills"
    / "gaokao-audit"
    / "tests"
    / "fixtures"
    / "sample_xianyu.txt"
)
QUICK_SCRIPT = PROJECT_ROOT / "scripts" / "gaokao-quick-3min.py"
ORDER_SCRIPT = PROJECT_ROOT / "scripts" / "gaokao-order-manager"
TRACE_SCRIPT = PROJECT_ROOT / "scripts" / "gaokao-data-trace"

os.environ.setdefault("GAOKAO_ORDERS_FERNET_KEY", "test-secret-for-t5-e2e")


class _CaptureReportGenerator(_ReportGeneratorBase):  # type: ignore[valid-type, misc]
    last_html: str = ""

    def __init__(self, **kwargs: object) -> None:
        super().__init__(
            now_text=lambda: "2026-06-13 10:00",
            report_id_factory=lambda: "AUDIT-T5-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\nt5 fake pdf\n")
        return str(target)


@pytest.fixture(scope="module")
def quick_module():
    spec = importlib.util.spec_from_file_location("gaokao_quick_3min", QUICK_SCRIPT)
    assert spec is not None and spec.loader is not None
    module = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(module)
    return module


@pytest.fixture
def tmp_orders_db(tmp_path: Path) -> Path:
    return tmp_path / "orders.db"


@pytest.fixture
def order_env() -> dict[str, str]:
    return {"GAOKAO_ORDERS_FERNET_KEY": os.environ["GAOKAO_ORDERS_FERNET_KEY"]}


def _run_script(
    script: Path, *args: str, env: dict[str, str] | None = None
) -> subprocess.CompletedProcess[str]:
    merged_env = os.environ.copy()
    if env:
        merged_env.update(env)
    return subprocess.run(
        [sys.executable, str(script), *args],
        cwd=PROJECT_ROOT,
        capture_output=True,
        text=True,
        env=merged_env,
    )


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_consultation_to_plan_generation_flow(quick_module) -> None:
    reply = """1. 李明
2. 浙江
3. 612
4. 15230
5. R
6. 物理、数学
7. C
8. ③
9. ①
10. ②
"""

    info = quick_module.parse_quick_response(reply)
    summary = quick_module.generate_quick_summary(info)
    recommendation = quick_module.generate_quick_recommendation(info)

    assert info["basic"]["name"] == "李明"
    assert info["basic"]["province"] == "浙江"
    assert info["exam"]["score"] == 612
    assert info["exam"]["rank"] == 15230
    assert info["profile"]["type_code"] == "R"
    assert "✅ 核心信息完整！可以开始推荐" in summary
    assert "📊 高考：612分" in summary
    assert "📊 位次：15230名" in summary
    assert "计算机科学与技术" in recommendation
    assert "物理数学强 → 计算机、电子信息、自动化" in recommendation


def test_audit_to_report_flow(
    tmp_path: Path,
    monkeypatch: pytest.MonkeyPatch,
    capsys: pytest.CaptureFixture[str],
) -> None:
    output_path = tmp_path / "audit-report.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)

    assert exit_code == 0
    assert output_path.exists()
    assert output_path.read_bytes().startswith(b"%PDF-1.4")
    assert payload["province"] == "湖南"
    assert payload["candidate_score"] == 578
    assert payload["crowd_risks"]
    assert "AUDIT-T5-E2E-001" in _CaptureReportGenerator.last_html
    assert "免责声明" in _CaptureReportGenerator.last_html


def test_order_to_delivery_flow_records_artifacts(
    tmp_orders_db: Path,
    tmp_path: Path,
    order_env: dict[str, str],
) -> None:
    plan_path = tmp_path / "plan.md"
    report_path = tmp_path / "audit.json"
    pdf_path = tmp_path / "report.pdf"
    plan_path.write_text("咨询后生成的志愿方案", encoding="utf-8")
    report_path.write_text('{"overall_score": 88}', encoding="utf-8")
    pdf_path.write_bytes(b"%PDF-1.4\nreport\n")

    created = _run_script(
        ORDER_SCRIPT,
        "--db",
        str(tmp_orders_db),
        "create",
        "--source",
        "xianyu",
        "--service-version",
        "audit",
        "--amount-cents",
        "4900",
        "--customer-name",
        "王家长",
        "--customer-phone",
        "13800001234",
        "--candidate-name",
        "李明",
        "--candidate-province",
        "湖南",
        "--candidate-score",
        "578",
        "--candidate-rank",
        "26800",
        env=order_env,
    )
    assert created.returncode == 0, created.stderr
    order_id = json.loads(created.stdout)["order"]["id"]

    updated = _run_script(
        ORDER_SCRIPT,
        "--db",
        str(tmp_orders_db),
        "update",
        order_id,
        "--assigned-consultant",
        "long-teacher",
        "--plan-file",
        str(plan_path),
        "--audit-report",
        str(report_path),
        "--pdf-path",
        str(pdf_path),
        "--note",
        "方案与审核报告已归档",
        env=order_env,
    )
    assert updated.returncode == 0, updated.stderr
    updated_payload = json.loads(updated.stdout)
    assert updated_payload["order"]["plan_file"] == str(plan_path)
    assert updated_payload["order"]["audit_report"] == str(report_path)
    assert updated_payload["order"]["pdf_path"] == str(pdf_path)

    paid = _run_script(
        ORDER_SCRIPT,
        "--db",
        str(tmp_orders_db),
        "pay",
        order_id,
        "--reason",
        "xianyu-paid",
        env=order_env,
    )
    assert paid.returncode == 0, paid.stderr

    delivered = _run_script(
        ORDER_SCRIPT,
        "--db",
        str(tmp_orders_db),
        "deliver",
        order_id,
        "--reason",
        "pdf-delivered",
        env=order_env,
    )
    assert delivered.returncode == 0, delivered.stderr
    delivered_payload = json.loads(delivered.stdout)
    assert delivered_payload["order"]["status"] == "delivered"
    assert delivered_payload["order"]["delivered_at"] is not None
    assert delivered_payload["order"]["plan_file"] == str(plan_path)
    assert delivered_payload["order"]["pdf_path"] == str(pdf_path)


def test_upgrade_flow_creates_delta_order(
    tmp_orders_db: Path,
    order_env: dict[str, str],
) -> None:
    created = _run_script(
        ORDER_SCRIPT,
        "--db",
        str(tmp_orders_db),
        "create",
        "--source",
        "wechat",
        "--service-version",
        "audit",
        "--amount-cents",
        "4900",
        "--customer-name",
        "王家长",
        "--customer-phone",
        "13900001234",
        env=order_env,
    )
    assert created.returncode == 0, created.stderr
    source_order_id = json.loads(created.stdout)["order"]["id"]

    upgraded = _run_script(
        ORDER_SCRIPT,
        "--db",
        str(tmp_orders_db),
        "upgrade",
        source_order_id,
        "--service-version",
        "standard",
        "--target-amount-cents",
        "9900",
        "--reason",
        "upgrade_to_standard",
        env=order_env,
    )
    assert upgraded.returncode == 0, upgraded.stderr
    payload = json.loads(upgraded.stdout)

    assert payload["order"]["upgrade_from"] == source_order_id
    assert payload["order"]["service_version"] == "standard"
    assert payload["order"]["amount_cents"] == 5000
    assert payload["source_order"]["id"] == source_order_id
    assert "upgraded" in payload["source_order"]["tags"]


def test_traceability_display_flow(capsys: pytest.CaptureFixture[str]) -> None:
    exit_code = _TRACE_CLI.main(["--human", "长沙理工大学"])
    captured = capsys.readouterr()

    assert exit_code == 0
    assert "query: 长沙理工大学" in captured.out
    assert "湖南 / 2026年数据 / 长沙理工大学 / 会计学" in captured.out
    assert "source_type: report (⚠️报告)" in captured.out
    assert "source_url: https://" in captured.out
    assert "confidence: 0.85" in captured.out


def test_traceability_json_entrypoint_matches_cli_contract() -> None:
    result = _run_script(TRACE_SCRIPT, "长沙理工大学")
    assert result.returncode == 0, result.stderr
    payload = json.loads(result.stdout)
    assert payload["query"] == "长沙理工大学"
    assert payload["match_count"] >= 1
    assert any(match["province"] == "湖南" for match in payload["matches"])