From 640b0e5fa9678fb4f018db0a6850f5ebeac62d55 Mon Sep 17 00:00:00 2001 From: Joshua Valluru <326636767+joshua-berri@users.noreply.github.com> Date: Thu, 17 Sep 2026 17:41:57 -0700 Subject: [PATCH] test(mcp): discover concrete tools through a catalog key --- tests/integration/observability/test_guardrail_effects.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/integration/observability/test_guardrail_effects.py b/tests/integration/observability/test_guardrail_effects.py index cd44c06cd82..25ecee96c4e 100644 --- a/tests/integration/observability/test_guardrail_effects.py +++ b/tests/integration/observability/test_guardrail_effects.py @@ -179,7 +179,8 @@ def test_request_selected_mcp_guardrail_blocks_direct_and_virtual_calls(gateway: key_selected = scenario.key(object_permission=permission, guardrails=[guardrail]) team = scenario.team(guardrails=[guardrail]) team_selected = scenario.key(team_id=team, object_permission=permission) - names = tool_names(candidate, key, identity) + catalog_key = scenario.key(object_permission={"mcp_servers": [identity]}) + names = tool_names(candidate, catalog_key, identity) assert set(names) == {"add", "multiply", "fail"} for virtual in (False, True): for caller, selected, tool, expected in (