Added testing for async

Author: vladvildanov

tests/entraid_utils.py

@@ -18,7 +18,8 @@
     ManagedIdentityType,
     ServicePrincipalIdentityProviderConfig,
     _create_provider_from_managed_identity,
-    _create_provider_from_service_principal, DefaultAzureCredentialIdentityProviderConfig,
+    _create_provider_from_service_principal,
+    DefaultAzureCredentialIdentityProviderConfig,
     _create_provider_from_default_azure_credential,
 )
 from tests.conftest import mock_identity_provider
@@ -51,10 +52,12 @@ def identity_provider(request) -> IdentityProviderInterface:
     return _create_provider_from_service_principal(config)
 
 
-def get_identity_provider_config(request) -> Union[
+def get_identity_provider_config(
+    request,
+) -> Union[
     ManagedIdentityProviderConfig,
     ServicePrincipalIdentityProviderConfig,
-    DefaultAzureCredentialIdentityProviderConfig
+    DefaultAzureCredentialIdentityProviderConfig,
 ]:
     if hasattr(request, "param"):
         kwargs = request.param.get("idp_kwargs", {})
@@ -123,7 +126,10 @@ def _get_service_principal_provider_config(
         app_kwargs=kwargs,
     )
 
-def _get_default_azure_credential_provider_config(request) -> DefaultAzureCredentialIdentityProviderConfig:
+
+def _get_default_azure_credential_provider_config(
+    request,
+) -> DefaultAzureCredentialIdentityProviderConfig:
     scopes = os.getenv("AZURE_REDIS_SCOPES", ())
 
     if hasattr(request, "param"):
@@ -134,12 +140,10 @@ def _get_default_azure_credential_provider_config(request) -> DefaultAzureCreden
         token_kwargs = {}
 
     if isinstance(scopes, str):
-        scopes = scopes.split(',')
+        scopes = scopes.split(",")
 
     return DefaultAzureCredentialIdentityProviderConfig(
-        scopes=scopes,
-        app_kwargs=kwargs,
-        token_kwargs=token_kwargs
+        scopes=scopes, app_kwargs=kwargs, token_kwargs=token_kwargs
     )
 
 

tests/test_asyncio/conftest.py

@@ -18,11 +18,6 @@
 from .compat import mock
 
 
-class AuthType(Enum):
-    MANAGED_IDENTITY = "managed_identity"
-    SERVICE_PRINCIPAL = "service_principal"
-
-
 async def _get_info(redis_url):
     client = redis.Redis.from_url(redis_url)
     info = await client.info()

tests/test_asyncio/test_credentials.py

@@ -18,6 +18,7 @@
 from redis.exceptions import ConnectionError
 from redis.utils import str_if_bytes
 from tests.conftest import get_endpoint, skip_if_redis_enterprise
+from tests.entraid_utils import AuthType
 from tests.test_asyncio.conftest import get_credential_provider
 
 try:
@@ -616,8 +617,12 @@ class TestEntraIdCredentialsProvider:
                 "cred_provider_class": EntraIdCredentialsProvider,
                 "cred_provider_kwargs": {"block_for_initial": True},
             },
+            {
+                "cred_provider_class": EntraIdCredentialsProvider,
+                "idp_kwargs": {"auth_type": AuthType.DEFAULT_AZURE_CREDENTIAL},
+            },
         ],
-        ids=["blocked", "non-blocked"],
+        ids=["blocked", "non-blocked", "DefaultAzureCredential"],
         indirect=True,
     )
     @pytest.mark.asyncio
@@ -692,8 +697,12 @@ class TestClusterEntraIdCredentialsProvider:
                 "cred_provider_class": EntraIdCredentialsProvider,
                 "cred_provider_kwargs": {"block_for_initial": True},
             },
+            {
+                "cred_provider_class": EntraIdCredentialsProvider,
+                "idp_kwargs": {"auth_type": AuthType.DEFAULT_AZURE_CREDENTIAL},
+            },
         ],
-        ids=["blocked", "non-blocked"],
+        ids=["blocked", "non-blocked", "DefaultAzureCredential"],
         indirect=True,
     )
     @pytest.mark.asyncio