test: isolate org mcp permission regressions

This commit is contained in:
Genmin 2026-04-30 10:11:37 -07:00
parent 5a6b81d4ed
commit b11e46fffe
2 changed files with 16 additions and 8 deletions

View file

@ -511,12 +511,19 @@ async def test_e2e_jwt_team_mcp_key_intersection(monkeypatch):
) as mock_access_groups:
mock_access_groups.return_value = []
allowed_servers = await MCPRequestHandler.get_allowed_mcp_servers(
user_api_key_auth
)
with patch.object(
MCPRequestHandler, "_get_allowed_mcp_servers_for_org"
) as mock_org_servers:
mock_org_servers.return_value = []
# Should be intersection: only server-2 is in both
expected = ["server-2"]
assert sorted(allowed_servers) == sorted(
expected
), f"Expected intersection {expected}, got {allowed_servers}"
allowed_servers = (
await MCPRequestHandler.get_allowed_mcp_servers(
user_api_key_auth
)
)
# Should be intersection: only server-2 is in both
expected = ["server-2"]
assert sorted(allowed_servers) == sorted(
expected
), f"Expected intersection {expected}, got {allowed_servers}"

View file

@ -5477,6 +5477,7 @@ async def test_update_team_guardrails_with_org_id():
mock_prisma.db.litellm_teamtable.find_unique = AsyncMock(
return_value=mock_existing_team
)
mock_cache.async_get_cache = AsyncMock(return_value=None)
mock_cache.async_set_cache = AsyncMock()
# Mock organization fetch - this is where the bug occurred