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

test_team_endpoints.py

Browse source
This commit is contained in:
Ishaan Jaff 2025-06-21 15:57:58 -07:00
parent a3bfd29104
commit 6cd2fe4879
1 changed files with 3 additions and 1 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

@ -271,7 +271,7 @@ async def test_new_team_with_object_permission(mock_db_client, mock_admin_auth):
return_value=MagicMock(id="model123")
)
# 3. Capture team table creation
# 3. Capture team table creation and count
team_create_result = MagicMock(
team_id="team-456",
object_permission_id="objperm123",
@ -281,8 +281,10 @@ async def test_new_team_with_object_permission(mock_db_client, mock_admin_auth):
"object_permission_id": "objperm123",
}
mock_team_create = AsyncMock(return_value=team_create_result)
mock_team_count = AsyncMock(return_value=0) # Mock count to return 0 (no existing teams)
mock_db_client.db.litellm_teamtable = MagicMock()
mock_db_client.db.litellm_teamtable.create = mock_team_create
mock_db_client.db.litellm_teamtable.count = mock_team_count
# 4. Mock user table update behaviour (called for each member)
mock_db_client.db.litellm_usertable = MagicMock()