[Core] Add runtime_checkable to protocol definitions (#25187)

mccoyp · web-flow · commit 388efaaa2b8f · 2022-08-19T17:25:02.000-07:00
diff --git a/sdk/core/azure-core/azure/core/credentials.py b/sdk/core/azure-core/azure/core/credentials.py
@@ -5,7 +5,7 @@
 # -------------------------------------------------------------------------
 from collections import namedtuple
 from typing import Any, NamedTuple, Optional
-from typing_extensions import Protocol
+from typing_extensions import Protocol, runtime_checkable
 import six
 
 
@@ -19,6 +19,7 @@ class AccessToken(NamedTuple):
 AccessToken.expires_on.__doc__ = """The token's expiration time in Unix time."""
 
 
+@runtime_checkable
 class TokenCredential(Protocol):
     """Protocol for classes able to provide OAuth tokens."""
 
diff --git a/sdk/core/azure-core/azure/core/credentials_async.py b/sdk/core/azure-core/azure/core/credentials_async.py
@@ -3,9 +3,10 @@
 # Licensed under the MIT License.
 # ------------------------------------
 from typing import Any, Optional
-from typing_extensions import Protocol
+from typing_extensions import Protocol, runtime_checkable
 from .credentials import AccessToken as _AccessToken
 
+@runtime_checkable
 class AsyncTokenCredential(Protocol):
     """Protocol for classes able to provide OAuth tokens."""
 
