diff --git a/tests/test_litellm/secret_managers/test_secret_managers_main.py b/tests/test_litellm/secret_managers/test_secret_managers_main.py index 159e41546d..3b9c6f2713 100644 --- a/tests/test_litellm/secret_managers/test_secret_managers_main.py +++ b/tests/test_litellm/secret_managers/test_secret_managers_main.py @@ -147,7 +147,12 @@ def test_oidc_azure_file_success(mock_env, tmp_path): @patch("litellm.secret_managers.main.get_azure_ad_token_provider") +@patch.dict(os.environ, {}, clear=False) # Ensure AZURE_FEDERATED_TOKEN_FILE is not set def test_oidc_azure_ad_token_success(mock_get_azure_ad_token_provider): + # Ensure the env var is not set so it falls through to Azure AD token provider + if "AZURE_FEDERATED_TOKEN_FILE" in os.environ: + del os.environ["AZURE_FEDERATED_TOKEN_FILE"] + mock_token_provider = Mock(return_value="azure_ad_token") mock_get_azure_ad_token_provider.return_value = mock_token_provider secret_name = "oidc/azure/api://azure-audience"