From 93da4cf603bac1473e8b4a56642a5d8e7ff8c06c Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 3 Jul 2026 01:29:12 +0000 Subject: [PATCH] fix(test): mock execute_raw in team endpoint tests These tests mock prisma_client as MagicMock, so db.execute_raw needs to be explicitly set to AsyncMock for the new atomic team append call Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../proxy/management_endpoints/test_team_endpoints.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py index 5f3974b46fb..0bc1132be9e 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py @@ -4025,6 +4025,7 @@ async def test_new_team_max_budget_within_user_limit(): mock_prisma.db.litellm_usertable = MagicMock() mock_prisma.db.litellm_usertable.upsert = AsyncMock(return_value=mock_user) mock_prisma.db.litellm_usertable.update = AsyncMock(return_value=mock_user) + mock_prisma.db.execute_raw = AsyncMock() # Mock team membership table mock_membership = MagicMock() @@ -4166,6 +4167,7 @@ async def test_new_team_org_scoped_budget_bypasses_user_limit(): mock_prisma.db.litellm_usertable = MagicMock() mock_prisma.db.litellm_usertable.upsert = AsyncMock(return_value=mock_user) mock_prisma.db.litellm_usertable.update = AsyncMock(return_value=mock_user) + mock_prisma.db.execute_raw = AsyncMock() # Mock team membership table mock_membership = MagicMock() @@ -4312,6 +4314,7 @@ async def test_new_team_org_scoped_models_bypasses_user_limit(): mock_prisma.db.litellm_usertable = MagicMock() mock_prisma.db.litellm_usertable.upsert = AsyncMock(return_value=mock_user) mock_prisma.db.litellm_usertable.update = AsyncMock(return_value=mock_user) + mock_prisma.db.execute_raw = AsyncMock() # Mock team membership table mock_membership = MagicMock() @@ -7164,6 +7167,7 @@ async def test_new_team_soft_budget_validation( mock_prisma.db.litellm_usertable = MagicMock() mock_prisma.db.litellm_usertable.upsert = AsyncMock(return_value=mock_user) mock_prisma.db.litellm_usertable.update = AsyncMock(return_value=mock_user) + mock_prisma.db.execute_raw = AsyncMock() # Mock team membership table mock_membership = MagicMock()