67 lines
2.0 KiB
Python
67 lines
2.0 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
|
|
|
|
|
|
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
|