scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(tests): set default_team_member_models=None on team mocks to match new field

Browse source
This commit is contained in:
Ishaan Jaffer 2026-04-06 17:25:45 -07:00
parent cf9fdff72e
commit 9e0fb6bd17
No known key found for this signature in database
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1033,6 +1033,7 @@ async def test_process_team_members_single_member():
mock_prisma_client = MagicMock()
mock_team = MagicMock(spec=LiteLLM_TeamTable)
mock_team.metadata = {"team_member_budget_id": "budget-123"}
mock_team.default_team_member_models = None
# Mock user and membership objects
mock_user = MagicMock(spec=LiteLLM_UserTable)
@ -1074,6 +1075,7 @@ async def test_process_team_members_single_member():
litellm_proxy_admin_name="admin",
team_id="test-team-123",
default_team_budget_id="budget-123",
allowed_models=None,
)
@ -1089,6 +1091,7 @@ async def test_process_team_members_multiple_members():
mock_prisma_client = MagicMock()
mock_team = MagicMock(spec=LiteLLM_TeamTable)
mock_team.metadata = None
mock_team.default_team_member_models = None
# Create multiple members as dictionaries (they will be converted to Member objects)
members = [
@ -2840,6 +2843,7 @@ async def test_new_team_max_budget_within_user_limit():
mock_created_team.max_budget = 50.0
mock_created_team.members_with_roles = []
mock_created_team.metadata = None
mock_created_team.default_team_member_models = None
mock_created_team.model_dump.return_value = {
"team_id": "team-within-budget-789",
"team_alias": "within-budget-team",
@ -2968,6 +2972,7 @@ async def test_new_team_org_scoped_budget_bypasses_user_limit():
mock_created_team.organization_id = "test-org-123"
mock_created_team.members_with_roles = []
mock_created_team.metadata = None
mock_created_team.default_team_member_models = None
mock_created_team.model_dump.return_value = {
"team_id": "team-org-scoped-789",
"team_alias": "org-scoped-team",
@ -3099,6 +3104,7 @@ async def test_new_team_org_scoped_models_bypasses_user_limit():
mock_created_team.models = ["gpt-4"]
mock_created_team.members_with_roles = []
mock_created_team.metadata = None
mock_created_team.default_team_member_models = None
mock_created_team.model_dump.return_value = {
"team_id": "team-org-scoped-models-789",
"team_alias": "org-scoped-models-team",
@ -5293,6 +5299,7 @@ async def test_new_team_soft_budget_validation(
mock_created_team.max_budget = expected_max_budget
mock_created_team.members_with_roles = []
mock_created_team.metadata = None
mock_created_team.default_team_member_models = None
mock_created_team.model_dump.return_value = {
"team_id": "test-team-123",
"team_alias": "test-soft-budget-team",