From 6ee2572897d81acdf8a915f6350ac4088296265e Mon Sep 17 00:00:00 2001 From: a2a-cloud Date: Mon, 13 Jul 2026 02:52:12 +0000 Subject: [PATCH] a2a-source-edit: write tests/test_agent.py --- tests/test_agent.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_agent.py b/tests/test_agent.py index 6e5dec1..dd20948 100644 --- a/tests/test_agent.py +++ b/tests/test_agent.py @@ -163,13 +163,13 @@ async def test_end_to_end_deterministic_crashing_image_scenario(ctx, workspace): plan = await agent.plan_remediation(ctx, "checkout-001", "rollback the deployment to the previous image") assert any(step["action_type"] == "kubernetes.rollback_deployment" for step in plan["steps"]) - denied = await agent.execute_remediation(ctx, "checkout-001", "outputs/incidents/checkout-001/remediation-plan.json", [plan.steps[0].id], "NO") + denied = await agent.execute_remediation(ctx, "checkout-001", "outputs/incidents/checkout-001/remediation-plan.json", [plan["steps"][0]["id"]], "NO") assert denied.status == "denied" - dry = await agent.execute_remediation(ctx, "checkout-001", "outputs/incidents/checkout-001/remediation-plan.json", [plan.steps[0].id], APPROVAL_ACK) + dry = await agent.execute_remediation(ctx, "checkout-001", "outputs/incidents/checkout-001/remediation-plan.json", [plan["steps"][0]["id"]], APPROVAL_ACK) assert dry.status == "dry_run" - executed = await agent.execute_remediation(ctx, "checkout-001", "outputs/incidents/checkout-001/remediation-plan.json", [plan.steps[0].id], APPROVAL_ACK, dry_run=False) + executed = await agent.execute_remediation(ctx, "checkout-001", "outputs/incidents/checkout-001/remediation-plan.json", [plan["steps"][0]["id"]], APPROVAL_ACK, dry_run=False) assert executed.status == "executed" verification = await agent.verify_recovery(ctx, "checkout-001", observation_seconds=0)