scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

test team endpoints

Browse source
This commit is contained in:
Ishaan Jaff 2025-08-13 18:43:50 -07:00
parent ce4210a17a
commit cd893134b7
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
View file

@ -1055,7 +1055,7 @@ async def test_update_team_team_member_budget_not_passed_to_db():
), patch(
"litellm.proxy.auth.auth_checks._cache_team_object"
) as mock_cache_team, patch(
"litellm.proxy.management_endpoints.team_endpoints._upsert_team_member_budget_table"
"litellm.proxy.management_endpoints.team_endpoints.TeamMemberBudgetHandler.upsert_team_member_budget_table"
) as mock_upsert_budget:
# Setup mock prisma client
@ -1082,7 +1082,7 @@ async def test_update_team_team_member_budget_not_passed_to_db():
# Mock budget upsert to return updated_kv without team_member_budget
def mock_upsert_side_effect(
team_table, updated_kv, team_member_budget, user_api_key_dict
team_table, user_api_key_dict, updated_kv, team_member_budget=None, team_member_rpm_limit=None, team_member_tpm_limit=None
):
# Remove team_member_budget from updated_kv as the real function does
result_kv = updated_kv.copy()