scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-08-28 05:25:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Fix failing OIDC secret manager tests

Browse source 
Fixed two test failures in test_secret_managers_main.py:

1. test_oidc_azure_ad_token_success: Corrected the patch path for get_bearer_token_provider from 'litellm.secret_managers.get_azure_ad_token_provider.get_bearer_token_provider' to 'azure.identity.get_bearer_token_provider' since the function is imported from azure.identity.

2. test_oidc_google_success: Added @patch('httpx.Client') decorator to prevent any real HTTP connections during test execution, resolving httpx.ConnectError issues.

Both tests now pass successfully.
This commit is contained in:
Alexsander Hamir 2026-01-26 17:33:27 -08:00
parent 831b186bd1
commit 262b63a4ea
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_litellm/secret_managers/test_secret_managers_main.py
View file

@ -48,7 +48,8 @@ def mock_env():
@patch("litellm.secret_managers.main.oidc_cache")
@patch("litellm.secret_managers.main._get_oidc_http_handler")
def test_oidc_google_success(mock_get_http_handler, mock_oidc_cache):
@patch("httpx.Client") # Prevent any real HTTP connections
def test_oidc_google_success(mock_httpx_client, mock_get_http_handler, mock_oidc_cache):
mock_oidc_cache.get_cache.return_value = None
mock_handler = MockHTTPHandler(timeout=600.0)
mock_get_http_handler.return_value = mock_handler
@ -161,7 +162,7 @@ def test_oidc_azure_ad_token_success(mock_get_azure_ad_token_provider):
mock_get_azure_ad_token_provider.return_value = mock_token_provider
# Also mock the Azure Identity SDK to prevent any real Azure calls
with patch("litellm.secret_managers.get_azure_ad_token_provider.get_bearer_token_provider") as mock_bearer:
with patch("azure.identity.get_bearer_token_provider") as mock_bearer:
mock_bearer.return_value = mock_token_provider
secret_name = "oidc/azure/api://azure-audience"