mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-15 23:31:29 +00:00
test: isolate org mcp permission regressions
This commit is contained in:
parent
5a6b81d4ed
commit
b11e46fffe
2 changed files with 16 additions and 8 deletions
|
|
@ -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}"
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue