diff --git a/backend/tests/test_ai_routing_governance_service.py b/backend/tests/test_ai_routing_governance_service.py index 722af9bf..78ebb7ff 100644 --- a/backend/tests/test_ai_routing_governance_service.py +++ b/backend/tests/test_ai_routing_governance_service.py @@ -1416,8 +1416,12 @@ async def test_ai_routing_api_returns_enveloped_payloads(monkeypatch) -> None: assert slot_publish.json()["data"]["active_version"] == "slot-v1" assert slot_rollback.json()["data"]["active_version"] == "slot-v2" assert diagnostics.json()["data"]["summary"]["ok"] == 1 - assert len(operation_log_calls) == 1 - log_kwargs = operation_log_calls[0]["kwargs"] + assert len(operation_log_calls) >= 1 + log_kwargs = next( + item["kwargs"] + for item in operation_log_calls + if item["kwargs"].get("entity_type") == "ai_routing_policy" + ) assert log_kwargs["tenant_id"] == 9 assert log_kwargs["user_id"] == 101 assert log_kwargs["entity_type"] == "ai_routing_policy"