diff --git a/tests/test_agent.py b/tests/test_agent.py index 299a0f7..a758bfc 100644 --- a/tests/test_agent.py +++ b/tests/test_agent.py @@ -133,8 +133,8 @@ async def test_allowlisted_operation_enforcement(ctx, workspace): ) workspace.write_bytes("outputs/incidents/inc-003/remediation-plan.json", plan.model_dump_json().encode()) result = await agent.execute_remediation(ctx, "inc-003", "outputs/incidents/inc-003/remediation-plan.json", ["step-unsafe"], APPROVAL_ACK, dry_run=False) - assert result.status == "failed" - assert result.step_results[0].status == "denied" + assert result["status"] == "failed" + assert result["step_results"][0]["status"] == "denied" @pytest.mark.asyncio