For integration tests, we still define key. Code uses key if defined.

microsoft · Oct 6, 2024 · 6560ae0 · 6560ae0
1 parent 9691aa3
commit 6560ae0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pctasks/core/pctasks/core/cosmos/settings.py b/pctasks/core/pctasks/core/cosmos/settings.py
@@ -125,7 +125,7 @@ def get_client(self) -> CosmosClient:
             # If the connection string is not set, the credetials are
             # automatically picked up from the environment/managed identity
             assert self.url
-            credential = DefaultAzureCredential()
+            credential = self.key or DefaultAzureCredential()
             return CosmosClient(
                 self.url, credential=credential, connection_verify=connection_verify
             )
@@ -147,7 +147,7 @@ def get_async_client(self) -> AsyncCosmosClient:
             # If the connection string is not set, the credetials are
             # automatically picked up from the environment/managed identity
             assert self.url
-            credential = azure.identity.aio.DefaultAzureCredential()
+            credential = self.key or azure.identity.aio.DefaultAzureCredential()
             return AsyncCosmosClient(
                 self.url, credential=credential, connection_verify=connection_verify
             )