42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
"""Integration HITL: escenario SIP (risk>=4) -> awaiting_approval -> approve/reject."""
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
EXAMPLES = (
|
|
Path(__file__).parent.parent.parent / "agents" / "incident_analyzer" / "examples"
|
|
)
|
|
|
|
|
|
@pytest.mark.integration
|
|
def test_hitl_approve_completa(integration_client) -> None: # type: ignore[no-untyped-def]
|
|
payload = (EXAMPLES / "01_sip_registration_drop.txt").read_text(encoding="utf-8")
|
|
r = integration_client.post("/api/agents/incident_analyzer/invoke", json={"input": payload})
|
|
body = r.json()
|
|
assert body["status"] == "awaiting_approval"
|
|
assert body["needs_human_for"]
|
|
trace_id = body["trace_id"]
|
|
pending = [a["id"] for a in body["needs_human_for"]]
|
|
|
|
r2 = integration_client.post(
|
|
f"/api/executions/{trace_id}/approve",
|
|
json={"approved_action_ids": pending, "comment": "OK rollback"},
|
|
)
|
|
assert r2.status_code == 200
|
|
assert r2.json()["status"] == "completed"
|
|
assert r2.json()["final_output"]["approved_actions"]
|
|
|
|
|
|
@pytest.mark.integration
|
|
def test_hitl_reject_marca_failed(integration_client) -> None: # type: ignore[no-untyped-def]
|
|
payload = (EXAMPLES / "01_sip_registration_drop.txt").read_text(encoding="utf-8")
|
|
r = integration_client.post("/api/agents/incident_analyzer/invoke", json={"input": payload})
|
|
trace_id = r.json()["trace_id"]
|
|
r2 = integration_client.post(
|
|
f"/api/executions/{trace_id}/reject", json={"reason": "rollback no procede"}
|
|
)
|
|
assert r2.status_code == 200
|
|
assert r2.json()["status"] == "failed"
|
|
assert "rejected_by_human" in r2.json()["error"]
|