From d383149ab008587172a4b75ddda788f51f0760b4 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@berri.ai>
Date: Fri, 3 Jul 2026 01:17:38 +0000
Subject: [PATCH] fix(test): mock execute_raw in
 test_create_team_member_add_team_admin

The test needs to mock the new atomic execute_raw call introduced by the
race condition fix, since it runs without a connected Prisma client

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 tests/proxy_unit_tests/test_proxy_server.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/proxy_unit_tests/test_proxy_server.py b/tests/proxy_unit_tests/test_proxy_server.py
index 212f7772cad..82177d52f05 100644
--- a/tests/proxy_unit_tests/test_proxy_server.py
+++ b/tests/proxy_unit_tests/test_proxy_server.py
@@ -1420,6 +1420,10 @@ async def test_create_team_member_add_team_admin(
             "litellm.proxy.proxy_server.prisma_client.get_data",
             new=AsyncMock(return_value=[]),
         ) as mock_get_data,
+        patch(
+            "litellm.proxy.proxy_server.prisma_client.db.execute_raw",
+            new_callable=AsyncMock,
+        ),
     ):
         mock_client = AsyncMock(
             return_value=LiteLLM_UserTable(