From bdc00ac85498554ac70b6507d09eeaf1309224a8 Mon Sep 17 00:00:00 2001 From: mrinal Date: Sun, 4 Oct 2026 09:29:51 +0000 Subject: [PATCH] test(auth): return typed permission rows in request flow vector store tests Co-authored-by: mrinal Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../proxy/auth/test_user_api_key_auth_request_flow.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tests/unit/proxy/auth/test_user_api_key_auth_request_flow.py b/tests/unit/proxy/auth/test_user_api_key_auth_request_flow.py index 76f7fb2aae6..df31595f7ec 100644 --- a/tests/unit/proxy/auth/test_user_api_key_auth_request_flow.py +++ b/tests/unit/proxy/auth/test_user_api_key_auth_request_flow.py @@ -28,6 +28,7 @@ from litellm.proxy._types import ( LiteLLM_JWTAuth, LiteLLM_BudgetTable, LiteLLM_EndUserTable, + LiteLLM_ObjectPermissionTable, LiteLLM_OrganizationTable, LiteLLM_TeamTableCachedObj, LiteLLM_UserTable, @@ -5328,13 +5329,16 @@ async def test_team_key_vector_store_access_when_team_cannot_be_resolved( request._url = URL(url="/v1/rag/query") database: Final = MagicMock() database.db.litellm_objectpermissiontable.find_unique = AsyncMock( - side_effect=lambda where: SimpleNamespace(vector_stores=["KBSTOREA"]) + side_effect=lambda where: LiteLLM_ObjectPermissionTable( + object_permission_id=where["object_permission_id"], vector_stores=["KBSTOREA"] + ) if where["object_permission_id"] == "key-permission" else None ) attrs: Final = { **_proxy_attrs_for_centralized_checks(), "prisma_client": database, + "proxy_logging_obj": MagicMock(service_logging_obj=MagicMock(async_service_success_hook=AsyncMock())), "general_settings": {"vector_store_deny_by_default": deny_by_default}, } for name, value in attrs.items(): @@ -5398,11 +5402,14 @@ async def test_keyless_team_member_vector_store_access_uses_only_the_resolved_te } database: Final = MagicMock() database.db.litellm_objectpermissiontable.find_unique = AsyncMock( - side_effect=lambda where: SimpleNamespace(vector_stores=grants[where["object_permission_id"]]) + side_effect=lambda where: LiteLLM_ObjectPermissionTable( + object_permission_id=where["object_permission_id"], vector_stores=grants[where["object_permission_id"]] + ) ) attrs: Final = { **_proxy_attrs_for_centralized_checks(), "prisma_client": database, + "proxy_logging_obj": MagicMock(service_logging_obj=MagicMock(async_service_success_hook=AsyncMock())), "general_settings": {"vector_store_deny_by_default": True}, } for name, value in attrs.items():