forked from phoenix/litellm-mirror
fix(management/utils.py): fix add_member to team when adding user_email
Fixes https://github.com/BerriAI/litellm/issues/5112
This commit is contained in:
parent
575afa8029
commit
a0a1feb7da
2 changed files with 52 additions and 3 deletions
|
@ -414,6 +414,42 @@ async def test_team_update_sc_2():
|
|||
assert new_team_data["data"][k] == team_data[k]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_team_member_add_email():
|
||||
from test_users import get_user_info
|
||||
|
||||
async with aiohttp.ClientSession() as session:
|
||||
## Create admin
|
||||
admin_user = f"{uuid.uuid4()}"
|
||||
await new_user(session=session, i=0, user_id=admin_user)
|
||||
## Create team with 1 admin and 1 user
|
||||
member_list = [
|
||||
{"role": "admin", "user_id": admin_user},
|
||||
]
|
||||
team_data = await new_team(session=session, i=0, member_list=member_list)
|
||||
## Add 1 user via email
|
||||
user_email = "krrish{}@berri.ai".format(uuid.uuid4())
|
||||
new_user_info = await new_user(session=session, i=0, user_email=user_email)
|
||||
new_member = {"role": "user", "user_email": user_email}
|
||||
await add_member(
|
||||
session=session, i=0, team_id=team_data["team_id"], members=[new_member]
|
||||
)
|
||||
|
||||
## check user info to confirm user is in team
|
||||
updated_user_info = await get_user_info(
|
||||
session=session, get_user=new_user_info["user_id"], call_user="sk-1234"
|
||||
)
|
||||
|
||||
print(updated_user_info)
|
||||
|
||||
## check if team in user table
|
||||
is_team_in_list: bool = False
|
||||
for team in updated_user_info["teams"]:
|
||||
if team_data["team_id"] == team["team_id"]:
|
||||
is_team_in_list = True
|
||||
assert is_team_in_list
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_team_delete():
|
||||
"""
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue