diff --git a/tests/proxy_admin_ui_tests/test_sso_sign_in.py b/tests/proxy_admin_ui_tests/test_sso_sign_in.py index 7ecee7879..94d10b4be 100644 --- a/tests/proxy_admin_ui_tests/test_sso_sign_in.py +++ b/tests/proxy_admin_ui_tests/test_sso_sign_in.py @@ -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: