65 lines
2.7 KiB
Python
65 lines
2.7 KiB
Python
import argparse
|
|
import json
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from scripts.import_agent_investigation import convert_plan, run
|
|
|
|
|
|
@pytest.fixture
|
|
def anyio_backend():
|
|
return "asyncio"
|
|
|
|
|
|
def snapshot():
|
|
return {"session_id": "ses_test", "revision": 1, "updated_at": "2026-09-23T00:00:00Z",
|
|
"title": "合成调查", "nodes": [
|
|
{"id": "STAGE_JUDGMENT", "title": "阶段判断", "owner": "agent",
|
|
"status": "completed", "detail": "合成证据不足以确定原因", "evidence": ["合成输入"]},
|
|
{"id": "SAMPLE", "title": "采样", "owner": "human", "status": "waiting",
|
|
"detail": "等待现场采样反馈", "evidence": []},
|
|
], "edges": [{"from": "STAGE_JUDGMENT", "to": "SAMPLE"}]}
|
|
|
|
|
|
def test_legacy_conversion_preserves_waiting_and_explicit_synthetic_boundary():
|
|
plan, evidence, report = convert_plan(snapshot())
|
|
assert plan["nodes"][1]["status"] == "waiting"
|
|
assert all(item["kind"] == "synthetic" for item in evidence)
|
|
assert report["pending_items"] == ["等待现场采样反馈"]
|
|
assert report["current_conclusion"] == "合成证据不足以确定原因"
|
|
assert "未重新查询当前数据" in report["conclusion_boundary"]
|
|
assert report["evidence_refs"] == [evidence[0]["evidence_id"]]
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_dry_run_validates_historical_schema_before_any_network_access(tmp_path, monkeypatch, capsys):
|
|
metadata_dir = tmp_path / "session-metadata"
|
|
states_dir = tmp_path / "session-ui-states"
|
|
metadata_dir.mkdir()
|
|
states_dir.mkdir()
|
|
(metadata_dir / "ses_test.json").write_text(json.dumps({
|
|
"triggerWorkflow": "water_quality_complaint", "triggerSource": "peripheral_sample",
|
|
"projectId": "10000000-0000-4000-8000-000000000001",
|
|
}))
|
|
state = snapshot()
|
|
state_path = states_dir / "ses_test.json"
|
|
state_path.write_text(json.dumps({"executionPlans": [state]}))
|
|
|
|
def no_network(*args, **kwargs):
|
|
raise AssertionError("dry run must not open a network client")
|
|
|
|
monkeypatch.setattr("scripts.import_agent_investigation.httpx.AsyncClient", no_network)
|
|
args = argparse.Namespace(session_id="ses_test", agent_data=tmp_path, dry_run=True)
|
|
await run(args)
|
|
assert "2 nodes and 1 synthetic evidence records; no writes" in capsys.readouterr().out
|
|
args.dry_run = False
|
|
with pytest.raises(ValidationError):
|
|
await run(args)
|
|
args.dry_run = True
|
|
state["nodes"][1]["status"] = "invented_status"
|
|
state_path.write_text(json.dumps({"executionPlans": [state]}))
|
|
with pytest.raises(ValidationError):
|
|
await run(args)
|
|
assert "Validated" not in capsys.readouterr().out
|