diff --git a/tests/integration/test_invoke_hitl.py b/tests/integration/test_invoke_hitl.py new file mode 100644 index 0000000..bb43413 --- /dev/null +++ b/tests/integration/test_invoke_hitl.py @@ -0,0 +1,41 @@ +"""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("/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"/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("/agents/incident_analyzer/invoke", json={"input": payload}) + trace_id = r.json()["trace_id"] + r2 = integration_client.post( + f"/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"] diff --git a/tests/integration/test_invoke_pii_block.py b/tests/integration/test_invoke_pii_block.py new file mode 100644 index 0000000..263fc14 --- /dev/null +++ b/tests/integration/test_invoke_pii_block.py @@ -0,0 +1,13 @@ +"""Integration: input con PII (NIF español + email) bloqueado por guardrails de entrada.""" + +import pytest + + +@pytest.mark.integration +def test_pii_es_nif_bloquea(integration_client) -> None: # type: ignore[no-untyped-def] + pii_input = "El cliente con NIF 12345678Z reporta caída de servicio. Email juan@example.com" + r = integration_client.post("/agents/incident_analyzer/invoke", json={"input": pii_input}) + assert r.status_code == 200 + body = r.json() + assert body["status"] == "blocked_by_guardrail" + assert any(v["validator"] == "DetectPII" and v["blocked"] for v in body["violations"])