From 8c38cfff9d2de6bf9f94bab993ef08f3ef4ea051 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 28 Sep 2024 19:11:28 -0700 Subject: [PATCH] fix sso sign in tests --- tests/proxy_admin_ui_tests/test_sso_sign_in.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) 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: