68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
"""gaokao-data-trace CLI tests (T3.4)."""
|
||
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import subprocess
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
import pytest
|
||
|
||
from data.crowd_db.cli import main as cli_main
|
||
|
||
PROJECT_ROOT = Path(__file__).resolve().parents[3]
|
||
SCRIPT_PATH = PROJECT_ROOT / "scripts" / "gaokao-data-trace"
|
||
|
||
|
||
def _run_cli(*args: str) -> subprocess.CompletedProcess[str]:
|
||
return subprocess.run(
|
||
[sys.executable, str(SCRIPT_PATH), *args],
|
||
cwd=PROJECT_ROOT,
|
||
capture_output=True,
|
||
text=True,
|
||
)
|
||
|
||
|
||
def test_trace_cli_json_output_contains_matches() -> None:
|
||
result = _run_cli("长沙理工大学")
|
||
assert result.returncode == 0, result.stderr
|
||
|
||
payload = json.loads(result.stdout)
|
||
assert payload["query"] == "长沙理工大学"
|
||
assert payload["match_count"] >= 1
|
||
assert any(match["school"] == "长沙理工大学" for match in payload["matches"])
|
||
|
||
hunan_match = next(
|
||
match for match in payload["matches"] if match["province"] == "湖南"
|
||
)
|
||
assert hunan_match["data_year"] == 2025
|
||
assert hunan_match["source_url"].startswith("https://")
|
||
assert 0 <= hunan_match["confidence"] <= 1
|
||
assert hunan_match["source_type"] == "report"
|
||
assert hunan_match["raw_source_type"] == "manual_summary"
|
||
|
||
|
||
def test_trace_cli_human_output_contains_required_lines(
|
||
capsys: pytest.CaptureFixture[str],
|
||
) -> None:
|
||
exit_code = cli_main(["--human", "长沙理工大学"])
|
||
captured = capsys.readouterr()
|
||
|
||
assert exit_code == 0
|
||
assert "query: 长沙理工大学" in captured.out
|
||
assert "湖南 / 2025年数据 / 长沙理工大学 / 会计学" in captured.out
|
||
assert "source_url: https://" in captured.out
|
||
assert "confidence: 0.85" in captured.out
|
||
assert "quality_level: high (A级（高置信）)" in captured.out
|
||
|
||
|
||
def test_trace_cli_missing_school_returns_nonzero(
|
||
capsys: pytest.CaptureFixture[str],
|
||
) -> None:
|
||
exit_code = cli_main(["不存在的测试院校XYZ"])
|
||
captured = capsys.readouterr()
|
||
|
||
assert exit_code == 1
|
||
assert "不存在的测试院校XYZ" in captured.err
|