mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
test(test_get_azure_ad_token_provider.py): add unit test to ensure default azure credentials used in the right context
This commit is contained in:
parent
84a7329dba
commit
d0732f55b3
1 changed files with 29 additions and 0 deletions
|
|
@ -214,3 +214,32 @@ class TestGetAzureAdTokenProvider:
|
|||
# Test that the returned callable works
|
||||
token = result()
|
||||
assert token == "mock-certificate-token"
|
||||
|
||||
@patch.dict(os.environ, {}, clear=True) # Clear all environment variables
|
||||
@patch("azure.identity.get_bearer_token_provider")
|
||||
@patch("azure.identity.DefaultAzureCredential")
|
||||
def test_get_azure_ad_token_provider_defaults_to_default_azure_credential(
|
||||
self, mock_default_azure_credential, mock_get_bearer_token_provider
|
||||
):
|
||||
"""Test get_azure_ad_token_provider defaults to DefaultAzureCredential when no credentials are present."""
|
||||
# Mock the Azure identity credential instance
|
||||
mock_credential_instance = MagicMock()
|
||||
mock_default_azure_credential.return_value = mock_credential_instance
|
||||
|
||||
# Mock the bearer token provider
|
||||
mock_token_provider = MagicMock(return_value="mock-default-token")
|
||||
mock_get_bearer_token_provider.return_value = mock_token_provider
|
||||
|
||||
# Call the function
|
||||
result = get_azure_ad_token_provider()
|
||||
|
||||
# Assertions
|
||||
assert callable(result)
|
||||
mock_default_azure_credential.assert_called_once_with()
|
||||
mock_get_bearer_token_provider.assert_called_once_with(
|
||||
mock_credential_instance, "https://cognitiveservices.azure.com/.default"
|
||||
)
|
||||
|
||||
# Test that the returned callable works
|
||||
token = result()
|
||||
assert token == "mock-default-token"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue