From a163bf9f5112eb6c6e22a5c06c376c8ea8dd80f0 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 11 Oct 2025 08:55:23 -0700 Subject: [PATCH] test fix: MicrosoftSSOHandler --- .../proxy/management_endpoints/test_ui_sso.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py b/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py index bfd5358dba6..941517db120 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py +++ b/tests/test_litellm/proxy/management_endpoints/test_ui_sso.py @@ -2,7 +2,6 @@ import asyncio import json import os import sys -from litellm._uuid import uuid from typing import Optional, cast from unittest.mock import AsyncMock, MagicMock, patch @@ -10,6 +9,8 @@ import pytest from fastapi import Request from fastapi.testclient import TestClient +from litellm._uuid import uuid + sys.path.insert( 0, os.path.abspath("../../../") ) # Adds the parent directory to the system path @@ -46,7 +47,7 @@ def test_microsoft_sso_handler_openid_from_response_user_principal_name(): # Act # Call the method being tested result = MicrosoftSSOHandler.openid_from_response( - response=mock_response, team_ids=expected_team_ids + response=mock_response, team_ids=expected_team_ids, user_role=None ) # Assert @@ -77,7 +78,7 @@ def test_microsoft_sso_handler_openid_from_response(): # Act # Call the method being tested result = MicrosoftSSOHandler.openid_from_response( - response=mock_response, team_ids=expected_team_ids + response=mock_response, team_ids=expected_team_ids, user_role=None ) # Assert @@ -98,7 +99,7 @@ def test_microsoft_sso_handler_with_empty_response(): # Test with None response # Act - result = MicrosoftSSOHandler.openid_from_response(response=None, team_ids=[]) + result = MicrosoftSSOHandler.openid_from_response(response=None, team_ids=[], user_role=None) # Assert assert isinstance(result, CustomOpenID)