pylint and better comments on sample

Azure · Mar 22, 2022 · 37a27e0 · 37a27e0
1 parent edb3f9c
commit 37a27e0
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 11 deletions.
diff --git a/sdk/cosmos/azure-cosmos/azure/cosmos/auth.py b/sdk/cosmos/azure-cosmos/azure/cosmos/auth.py
@@ -26,15 +26,12 @@
 from hashlib import sha256
 import hmac
 import urllib.parse
-
-from . import http_constants
-from typing import Any, Dict, Optional, ClassVar
 import time
-
-from azure.core.pipeline.policies._base import HTTPPolicy
+from typing import Any, Dict, Optional, ClassVar
 from azure.core.exceptions import ServiceRequestError
 from azure.core.credentials import AccessToken
 from azure.core.pipeline import PipelineRequest, PipelineResponse
+from . import http_constants
 
 TokenCredential = ClassVar
 
@@ -206,7 +203,7 @@ def _need_new_token(self):
         return not self._token or self._token.expires_on - time.time() < 300
 
 
-class CosmosBearerTokenCredentialPolicy(_CosmosBearerTokenCredentialPolicyBase, HTTPPolicy):
+class CosmosBearerTokenCredentialPolicy(_CosmosBearerTokenCredentialPolicyBase):
     """Adds a Cosmos bearer token Authorization header to requests.
 
     :param credential: The credential.

diff --git a/sdk/cosmos/azure-cosmos/azure/cosmos/cosmos_client.py b/sdk/cosmos/azure-cosmos/azure/cosmos/cosmos_client.py
@@ -25,7 +25,6 @@
 from typing import Any, Dict, Optional, Union, cast, Iterable, List  # pylint: disable=unused-import
 
 from azure.core.tracing.decorator import distributed_trace  # type: ignore
-from azure.identity import ClientSecretCredential
 
 from ._cosmos_client_connection import CosmosClientConnection
 from ._base import build_options

diff --git a/sdk/cosmos/azure-cosmos/samples/access_cosmos_with_aad.py b/sdk/cosmos/azure-cosmos/samples/access_cosmos_with_aad.py
@@ -74,10 +74,12 @@ def run_sample():
     db = aad_client.get_database_client(DATABASE_ID)
     container = db.get_container_client(CONTAINER_ID)
 
-    print(container.read())
-    container.create_item(get_test_item(0))
-    print(container.read_item(item='Item_0', partition_key='Item_0'))
-    print(list(container.query_items(query='select * from c', partition_key='Item_0')))
+    print("Container info: " + str(container.read()))
+    container.create_item(get_test_item(879))
+    print("Point read result: " + str(container.read_item(item='Item_0', partition_key='Item_0')))
+    query_results = list(container.query_items(query='select * from c', partition_key='Item_0'))
+    assert len(query_results) == 1
+    print("Query result: " + str(query_results[0]))
     container.delete_item(item='Item_0', partition_key='Item_0')
 
     # Attempting to do management operations will return a 403 Forbidden exception.