tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-13 14:23:04 +00:00
Code Issues Packages Projects Releases Wiki 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
tests/proxy_admin_ui_tests/test_sso_sign_in.py
+6 -4
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: