From 1917700a576c6bc9c3f94fd1042694fc95065da2 Mon Sep 17 00:00:00 2001 From: jesus Date: Fri, 2 Oct 2026 03:01:46 +0000 Subject: [PATCH] test(keys): mock writer_db team lookup used by check_db_only path Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../management_endpoints/test_key_management_endpoints.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/proxy/management_endpoints/test_key_management_endpoints.py b/tests/unit/proxy/management_endpoints/test_key_management_endpoints.py index a18a2833f36..8f2060cc8a8 100644 --- a/tests/unit/proxy/management_endpoints/test_key_management_endpoints.py +++ b/tests/unit/proxy/management_endpoints/test_key_management_endpoints.py @@ -20491,9 +20491,9 @@ async def test_project_detachment_uses_effective_project_for_validation(project_ def _project_assignment_prisma() -> MagicMock: database = MagicMock() - database.db.litellm_teamtable.find_unique = AsyncMock( - return_value=LiteLLM_TeamTable(team_id="team-lit-5823", members=[]) - ) + team_row: Final = LiteLLM_TeamTable(team_id="team-lit-5823", members=[]) + database.db.litellm_teamtable.find_unique = AsyncMock(return_value=team_row) + database.writer_db.litellm_teamtable.find_unique = AsyncMock(return_value=team_row) return database