phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix sso sign in tests

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-28 19:11:28 -07:00
parent 67d2989d1a
commit 8c38cfff9d
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/proxy_admin_ui_tests/test_sso_sign_in.py
View file

@ -65,6 +65,7 @@ async def test_auth_callback_new_user(mock_google_sso, mock_env_vars, prisma_cli
# Generate a unique user ID
unique_user_id = str(uuid.uuid4())
unique_user_email = f"newuser{unique_user_id}@example.com"
try:
# Set up the prisma client
@ -76,7 +77,7 @@ async def test_auth_callback_new_user(mock_google_sso, mock_env_vars, prisma_cli
# Mock the GoogleSSO verify_and_process method
mock_sso_result = MagicMock()
mock_sso_result.email = "newuser@example.com"
mock_sso_result.email = unique_user_email
mock_sso_result.id = unique_user_id
mock_sso_result.provider = "google"
mock_google_sso.return_value.verify_and_process = AsyncMock(
@ -109,7 +110,7 @@ async def test_auth_callback_new_user(mock_google_sso, mock_env_vars, prisma_cli
)
print("inserted user from SSO", user)
assert user is not None
assert user.user_email == "newuser@example.com"
assert user.user_email == unique_user_email
assert user.user_role == LitellmUserRoles.INTERNAL_USER_VIEW_ONLY
assert user.metadata == {"auth_provider": "google"}
@ -134,6 +135,7 @@ async def test_auth_callback_new_user_with_sso_default(
# Generate a unique user ID
unique_user_id = str(uuid.uuid4())
unique_user_email = f"newuser{unique_user_id}@example.com"
try:
# Set up the prisma client
@ -148,7 +150,7 @@ async def test_auth_callback_new_user_with_sso_default(
# Mock the GoogleSSO verify_and_process method
mock_sso_result = MagicMock()
mock_sso_result.email = "newuser@example.com"
mock_sso_result.email = unique_user_email
mock_sso_result.id = unique_user_id
mock_sso_result.provider = "google"
mock_google_sso.return_value.verify_and_process = AsyncMock(
@ -181,7 +183,7 @@ async def test_auth_callback_new_user_with_sso_default(
)
print("inserted user from SSO", user)
assert user is not None
assert user.user_email == "newuser@example.com"
assert user.user_email == unique_user_email
assert user.user_role == LitellmUserRoles.INTERNAL_USER
finally: