From 0b15d8ed0744c16d84e2adf7687ab02e2fe188fb Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Thu, 16 Jul 2026 14:34:30 +0000 Subject: [PATCH] test(secret_managers): cover azure.identity credential fallback path Co-Authored-By: Ishaan Jaffer <155045088+ishaan-berri@users.noreply.github.com> --- .../test_get_azure_ad_token_provider.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tests/test_litellm/secret_managers/test_get_azure_ad_token_provider.py b/tests/test_litellm/secret_managers/test_get_azure_ad_token_provider.py index c178c92598a..48f9aabb028 100644 --- a/tests/test_litellm/secret_managers/test_get_azure_ad_token_provider.py +++ b/tests/test_litellm/secret_managers/test_get_azure_ad_token_provider.py @@ -376,3 +376,18 @@ class TestGetAzureCredential: assert credential is mock_credential_instance mock_default_azure_credential.assert_called_once_with() + + @patch.dict(os.environ, {"AZURE_CREDENTIAL": "AzureCliCredential"}, clear=True) + @patch("azure.identity.AzureCliCredential") + def test_unknown_credential_name_falls_back_to_azure_identity_class( + self, mock_azure_cli_credential + ): + """An AZURE_CREDENTIAL value that isn't in AzureCredentialType is resolved + as a class name on azure.identity.""" + mock_credential_instance = MagicMock() + mock_azure_cli_credential.return_value = mock_credential_instance + + credential = get_azure_credential() + + assert credential is mock_credential_instance + mock_azure_cli_credential.assert_called_once_with()