126 lines
3.7 KiB
Python
126 lines
3.7 KiB
Python
"""SqliteDiagnosisSignalStore 单元测试。
|
||
|
||
验证逐题诊断信号的 upsert 幂等、断点续跑（done_question_ids）、
|
||
load 往返还原，以及 INFRA 行的可空字段处理。
|
||
"""
|
||
|
||
import sqlite3
|
||
|
||
import pytest
|
||
|
||
from adapters.baseline_diagnosis_store import SqliteDiagnosisSignalStore
|
||
from core.evolution.types import DiagnosisSignalRow
|
||
|
||
|
||
def _store(tmp_path):
|
||
"""构造指向临时 harness.db 的信号存储。"""
|
||
return SqliteDiagnosisSignalStore(str(tmp_path / "h.db"))
|
||
|
||
|
||
def test_upsert_and_get_done_ids(tmp_path):
|
||
s = _store(tmp_path)
|
||
row = DiagnosisSignalRow(
|
||
question_id="q1",
|
||
video_id="v1",
|
||
baseline_run_id="r",
|
||
diag_fingerprint="fp",
|
||
task_type="Counting",
|
||
error_type="search_failure",
|
||
cause_category="defect",
|
||
tier="T2",
|
||
evolution_target="skill",
|
||
degraded=False,
|
||
infra=False,
|
||
session_id="s",
|
||
)
|
||
s.upsert(row)
|
||
assert s.done_question_ids("r", "fp") == {"q1"}
|
||
s.upsert(row) # 同键覆盖，不重复
|
||
assert s.done_question_ids("r", "fp") == {"q1"}
|
||
assert s.done_question_ids("r", "other_fp") == set() # 不同 fingerprint 隔离
|
||
|
||
|
||
def test_load_rows_roundtrip(tmp_path):
|
||
s = _store(tmp_path)
|
||
s.upsert(
|
||
DiagnosisSignalRow(
|
||
"q2",
|
||
"v2",
|
||
"r",
|
||
"fp",
|
||
"OCR Problems",
|
||
"extraction_failure",
|
||
"lapse",
|
||
"T1",
|
||
"tool",
|
||
False,
|
||
False,
|
||
"s",
|
||
)
|
||
)
|
||
rows = s.load("r", "fp")
|
||
assert len(rows) == 1 and rows[0].tier == "T1" and rows[0].evolution_target == "tool"
|
||
|
||
|
||
def test_null_fields_for_infra_row(tmp_path):
|
||
s = _store(tmp_path)
|
||
# T0 INFRA 行：error_type/cause_category/evolution_target 为 None
|
||
s.upsert(
|
||
DiagnosisSignalRow(
|
||
"q3",
|
||
"v3",
|
||
"r",
|
||
"fp",
|
||
"Counting",
|
||
None,
|
||
None,
|
||
"T0",
|
||
None,
|
||
degraded=False,
|
||
infra=True,
|
||
session_id=None,
|
||
)
|
||
)
|
||
rows = s.load("r", "fp")
|
||
assert rows[0].error_type is None and rows[0].evolution_target is None and rows[0].infra is True
|
||
|
||
|
||
def test_done_question_ids_retry_uncertain_excludes(tmp_path):
|
||
"""retry_uncertain=True 时 uncertain 题不算完成（会被重诊），非 uncertain 仍算完成。"""
|
||
s = _store(tmp_path)
|
||
# T2（defect）题：已完成
|
||
s.upsert(
|
||
DiagnosisSignalRow(
|
||
"t2_qid", "v1", "run", "fp", "Counting",
|
||
"search_failure", "defect", "T2", "skill",
|
||
degraded=False, infra=False, session_id="s",
|
||
)
|
||
)
|
||
# uncertain（degraded）题：信号不可信
|
||
s.upsert(
|
||
DiagnosisSignalRow(
|
||
"uncertain_qid", "v2", "run", "fp", "Counting",
|
||
None, None, "uncertain", None,
|
||
degraded=True, infra=False, session_id=None,
|
||
)
|
||
)
|
||
# 默认：两者都算完成
|
||
assert s.done_question_ids("run", "fp") == {"t2_qid", "uncertain_qid"}
|
||
# retry_uncertain：uncertain 被排除，T2 仍算完成
|
||
done = s.done_question_ids("run", "fp", retry_uncertain=True)
|
||
assert "uncertain_qid" not in done
|
||
assert "t2_qid" in done
|
||
|
||
|
||
def test_context_manager_closes_connection(tmp_path):
|
||
# with 块退出后连接关闭，再操作应报 ProgrammingError
|
||
with SqliteDiagnosisSignalStore(str(tmp_path / "h.db")) as s:
|
||
s.upsert(
|
||
DiagnosisSignalRow(
|
||
"q4", "v4", "r", "fp", "Counting", None, None, "T0", None, False, True, None
|
||
)
|
||
)
|
||
assert s.done_question_ids("r", "fp") == {"q4"}
|
||
with pytest.raises(sqlite3.ProgrammingError):
|
||
s.done_question_ids("r", "fp")
|