Added AAD support

sdk/communication/azure-communication-sms/azure/communication/sms/_shared/user_credential.py

@@ -11,7 +11,7 @@
 )
 import six
 from msrest.serialization import TZ_UTC
-from .utils import create_access_token
+from .utils import create_access_token, get_authentication_policy
 
 class CommunicationUserCredential(object):
     """Credential type used for authenticating to an Azure Communication service.

sdk/communication/azure-communication-sms/azure/communication/sms/_shared/utils.py

@@ -68,3 +68,31 @@ def create_access_token(token):
         return AccessToken(token, datetime.fromtimestamp(payload['exp']).replace(tzinfo=TZ_UTC))
     except ValueError:
         raise ValueError(token_parse_err_msg)
+
+    def get_authentication_policy(
+            endpoint, # type: str
+            credential # type: TokenCredential or str
+    ):
+        # type: (...) -> BearerTokenCredentialPolicy or HMACCredentialPolicy
+        """Returns the correct authentication policy based
+        on which credential is being passed.
+        :param endpoint: The endpoint to which we are authenticating to.
+        :type endpoint: str
+        :param credential: The credential we use to authenticate to the service
+        :type credential: TokenCredential or str
+        :rtype: ~azure.core.pipeline.policies.BearerTokenCredentialPolicy
+        ~HMACCredentialsPolicy
+        """
+
+        if credential is None:
+            raise ValueError("Parameter 'credential' must not be None.")
+        if hasattr(credential, "get_token"):
+            from azure.core.pipeline.policies import BearerTokenCredentialPolicy
+            return BearerTokenCredentialPolicy(
+                credential, "https://communication.azure.com//.default")
+        if isinstance(credential, str):
+            from .._shared.policy import HMACCredentialsPolicy
+            return HMACCredentialsPolicy(endpoint, credential)
+
+        raise TypeError("Unsupported credential: {}. Use an access token string to use HMACCredentialsPolicy"
+                        "or a token credential from azure.identity".format(type(credential)))

sdk/communication/azure-communication-sms/azure/communication/sms/_sms_client.py

@@ -9,8 +9,7 @@
 from azure.communication.sms._generated.models import SendSmsResponse
 
 from ._generated._azure_communication_sms_service import AzureCommunicationSMSService
-from ._shared.policy import HMACCredentialsPolicy
-from ._shared.utils import parse_connection_str
+from ._shared.utils import parse_connection_str, get_authentication_policy
 from ._version import SDK_MONIKER
 
 class SmsClient(object):
@@ -41,8 +40,7 @@ def __init__(
                 "invalid credential from connection string.")
 
         self._endpoint = endpoint
-        self._authentication_policy = HMACCredentialsPolicy(endpoint, credential)
-
+        self._authentication_policy = get_authentication_policy(endpoint, credential)
         self._sms_service_client = AzureCommunicationSMSService(
             self._endpoint,
             authentication_policy=self._authentication_policy,

sdk/communication/azure-communication-sms/azure/communication/sms/aio/_sms_client_async.py

@@ -9,8 +9,7 @@
 from azure.communication.sms._generated.models import SendSmsResponse
 
 from .._generated.aio._azure_communication_sms_service import AzureCommunicationSMSService
-from .._shared.policy import HMACCredentialsPolicy
-from .._shared.utils import parse_connection_str
+from .._shared.utils import parse_connection_str, get_authentication_policy
 from .._version import SDK_MONIKER
 
 class SmsClient(object):
@@ -41,7 +40,7 @@ def __init__(
                 "invalid credential from connection string.")
 
         self._endpoint = endpoint
-        self._authentication_policy = HMACCredentialsPolicy(endpoint, credential)
+        self._authentication_policy = get_authentication_policy(endpoint, credential)
 
         self._sms_service_client = AzureCommunicationSMSService(
             self._endpoint,