[AutoRelease] t2-cosmosdb-2022-08-11-82446(Do not merge) (Azure#25664)

* code and test

* Update CHANGELOG.md

Co-authored-by: PythonSdkPipelines <PythonSdkPipelines>
Co-authored-by: Yuchao Yan <yuchaoyan@microsoft.com>

Author: azure-sdk

sdk/cosmos/azure-mgmt-cosmosdb/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Release History
 
+## 8.0.0b2 (2022-08-11)
+
+### Breaking Changes
+
+  - Renamed model `ComponentsM9L909SchemasCassandraclusterpublicstatusPropertiesDatacentersItemsPropertiesNodesItems` to `CassandraClusterDataCenterNodeItem`
+  - Renamed model `Components1Jq1T4ISchemasManagedserviceidentityPropertiesUserassignedidentitiesAdditionalproperties` to `ManagedServiceIdentityUserAssignedIdentity`
+
 ## 8.0.0b1 (2022-08-03)
 
 **Features**

sdk/cosmos/azure-mgmt-cosmosdb/_meta.json

@@ -1,11 +1,11 @@
 {
-  "autorest": "3.7.2",
+  "autorest": "3.8.4",
   "use": [
-    "@autorest/python@5.16.0",
-    "@autorest/modelerfour@4.19.3"
+    "@autorest/python@6.0.1",
+    "@autorest/modelerfour@4.23.5"
   ],
-  "commit": "1dd79c416cdccde274113ec03b92b75069fad464",
+  "commit": "4b5383a3c9fb089d361d2e11f65d0d04c76e94ac",
   "repository_url": "https://github.com/Azure/azure-rest-api-specs",
-  "autorest_command": "autorest specification/cosmos-db/resource-manager/readme.md --python --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --python3-only --use=@autorest/python@5.16.0 --use=@autorest/modelerfour@4.19.3 --version=3.7.2",
+  "autorest_command": "autorest specification/cosmos-db/resource-manager/readme.md --models-mode=msrest --python --python-sdks-folder=/home/vsts/work/1/azure-sdk-for-python/sdk --use=@autorest/python@6.0.1 --use=@autorest/modelerfour@4.23.5 --version=3.8.4 --version-tolerant=False",
   "readme": "specification/cosmos-db/resource-manager/readme.md"
 }

sdk/cosmos/azure-mgmt-cosmosdb/azure/mgmt/cosmosdb/__init__.py

@@ -17,7 +17,8 @@
 except ImportError:
     _patch_all = []
 from ._patch import patch_sdk as _patch_sdk
-__all__ = ['CosmosDBManagementClient']
+
+__all__ = ["CosmosDBManagementClient"]
 __all__.extend([p for p in _patch_all if p not in __all__])
 
 _patch_sdk()

sdk/cosmos/azure-mgmt-cosmosdb/azure/mgmt/cosmosdb/_configuration.py

@@ -25,23 +25,18 @@ class CosmosDBManagementClientConfiguration(Configuration):  # pylint: disable=t
     Note that all parameters used to create this instance are saved as instance
     attributes.
 
-    :param credential: Credential needed for the client to connect to Azure.
+    :param credential: Credential needed for the client to connect to Azure. Required.
     :type credential: ~azure.core.credentials.TokenCredential
-    :param subscription_id: The ID of the target subscription.
+    :param subscription_id: The ID of the target subscription. Required.
     :type subscription_id: str
     :keyword api_version: Api Version. Default value is "2022-05-15-preview". Note that overriding
      this default value may result in unsupported behavior.
     :paramtype api_version: str
     """
 
-    def __init__(
-        self,
-        credential: "TokenCredential",
-        subscription_id: str,
-        **kwargs: Any
-    ) -> None:
+    def __init__(self, credential: "TokenCredential", subscription_id: str, **kwargs: Any) -> None:
         super(CosmosDBManagementClientConfiguration, self).__init__(**kwargs)
-        api_version = kwargs.pop('api_version', "2022-05-15-preview")  # type: str
+        api_version = kwargs.pop("api_version", "2022-05-15-preview")  # type: str
 
         if credential is None:
             raise ValueError("Parameter 'credential' must not be None.")
@@ -51,23 +46,24 @@ def __init__(
         self.credential = credential
         self.subscription_id = subscription_id
         self.api_version = api_version
-        self.credential_scopes = kwargs.pop('credential_scopes', ['https://management.azure.com/.default'])
-        kwargs.setdefault('sdk_moniker', 'mgmt-cosmosdb/{}'.format(VERSION))
+        self.credential_scopes = kwargs.pop("credential_scopes", ["https://management.azure.com/.default"])
+        kwargs.setdefault("sdk_moniker", "mgmt-cosmosdb/{}".format(VERSION))
         self._configure(**kwargs)
 
     def _configure(
-        self,
-        **kwargs  # type: Any
+        self, **kwargs  # type: Any
     ):
         # type: (...) -> None
-        self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs)
-        self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs)
-        self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs)
-        self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs)
-        self.http_logging_policy = kwargs.get('http_logging_policy') or ARMHttpLoggingPolicy(**kwargs)
-        self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs)
-        self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs)
-        self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs)
-        self.authentication_policy = kwargs.get('authentication_policy')
+        self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs)
+        self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs)
+        self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs)
+        self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs)
+        self.http_logging_policy = kwargs.get("http_logging_policy") or ARMHttpLoggingPolicy(**kwargs)
+        self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs)
+        self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs)
+        self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs)
+        self.authentication_policy = kwargs.get("authentication_policy")
         if self.credential and not self.authentication_policy:
-            self.authentication_policy = ARMChallengeAuthenticationPolicy(self.credential, *self.credential_scopes, **kwargs)
+            self.authentication_policy = ARMChallengeAuthenticationPolicy(
+                self.credential, *self.credential_scopes, **kwargs
+            )

sdk/cosmos/azure-mgmt-cosmosdb/azure/mgmt/cosmosdb/_cosmos_db_management_client.py

@@ -9,20 +9,60 @@
 from copy import deepcopy
 from typing import Any, TYPE_CHECKING
 
-from msrest import Deserializer, Serializer
-
 from azure.core.rest import HttpRequest, HttpResponse
 from azure.mgmt.core import ARMPipelineClient
 
 from . import models
 from ._configuration import CosmosDBManagementClientConfiguration
-from .operations import CassandraClustersOperations, CassandraDataCentersOperations, CassandraResourcesOperations, CollectionOperations, CollectionPartitionOperations, CollectionPartitionRegionOperations, CollectionRegionOperations, DataTransferJobsOperations, DatabaseAccountRegionOperations, DatabaseAccountsOperations, DatabaseOperations, GraphResourcesOperations, GremlinResourcesOperations, LocationsOperations, MongoDBResourcesOperations, NotebookWorkspacesOperations, Operations, PartitionKeyRangeIdOperations, PartitionKeyRangeIdRegionOperations, PercentileOperations, PercentileSourceTargetOperations, PercentileTargetOperations, PrivateEndpointConnectionsOperations, PrivateLinkResourcesOperations, RestorableDatabaseAccountsOperations, RestorableGremlinDatabasesOperations, RestorableGremlinGraphsOperations, RestorableGremlinResourcesOperations, RestorableMongodbCollectionsOperations, RestorableMongodbDatabasesOperations, RestorableMongodbResourcesOperations, RestorableSqlContainersOperations, RestorableSqlDatabasesOperations, RestorableSqlResourcesOperations, RestorableTableResourcesOperations, RestorableTablesOperations, ServiceOperations, SqlResourcesOperations, TableResourcesOperations
+from ._serialization import Deserializer, Serializer
+from .operations import (
+    CassandraClustersOperations,
+    CassandraDataCentersOperations,
+    CassandraResourcesOperations,
+    CollectionOperations,
+    CollectionPartitionOperations,
+    CollectionPartitionRegionOperations,
+    CollectionRegionOperations,
+    DataTransferJobsOperations,
+    DatabaseAccountRegionOperations,
+    DatabaseAccountsOperations,
+    DatabaseOperations,
+    GraphResourcesOperations,
+    GremlinResourcesOperations,
+    LocationsOperations,
+    MongoDBResourcesOperations,
+    NotebookWorkspacesOperations,
+    Operations,
+    PartitionKeyRangeIdOperations,
+    PartitionKeyRangeIdRegionOperations,
+    PercentileOperations,
+    PercentileSourceTargetOperations,
+    PercentileTargetOperations,
+    PrivateEndpointConnectionsOperations,
+    PrivateLinkResourcesOperations,
+    RestorableDatabaseAccountsOperations,
+    RestorableGremlinDatabasesOperations,
+    RestorableGremlinGraphsOperations,
+    RestorableGremlinResourcesOperations,
+    RestorableMongodbCollectionsOperations,
+    RestorableMongodbDatabasesOperations,
+    RestorableMongodbResourcesOperations,
+    RestorableSqlContainersOperations,
+    RestorableSqlDatabasesOperations,
+    RestorableSqlResourcesOperations,
+    RestorableTableResourcesOperations,
+    RestorableTablesOperations,
+    ServiceOperations,
+    SqlResourcesOperations,
+    TableResourcesOperations,
+)
 
 if TYPE_CHECKING:
     # pylint: disable=unused-import,ungrouped-imports
     from azure.core.credentials import TokenCredential
 
-class CosmosDBManagementClient:    # pylint: disable=too-many-instance-attributes
+
+class CosmosDBManagementClient:  # pylint: disable=client-accepts-api-version-keyword,too-many-instance-attributes
     """Azure Cosmos DB Database Service Resource Provider REST API.
 
     :ivar database_accounts: DatabaseAccountsOperations operations
@@ -119,9 +159,9 @@ class CosmosDBManagementClient:    # pylint: disable=too-many-instance-attribute
      azure.mgmt.cosmosdb.operations.RestorableTableResourcesOperations
     :ivar service: ServiceOperations operations
     :vartype service: azure.mgmt.cosmosdb.operations.ServiceOperations
-    :param credential: Credential needed for the client to connect to Azure.
+    :param credential: Credential needed for the client to connect to Azure. Required.
     :type credential: ~azure.core.credentials.TokenCredential
-    :param subscription_id: The ID of the target subscription.
+    :param subscription_id: The ID of the target subscription. Required.
     :type subscription_id: str
     :param base_url: Service URL. Default value is "https://management.azure.com".
     :type base_url: str
@@ -139,7 +179,9 @@ def __init__(
         base_url: str = "https://management.azure.com",
         **kwargs: Any
     ) -> None:
-        self._config = CosmosDBManagementClientConfiguration(credential=credential, subscription_id=subscription_id, **kwargs)
+        self._config = CosmosDBManagementClientConfiguration(
+            credential=credential, subscription_id=subscription_id, **kwargs
+        )
         self._client = ARMPipelineClient(base_url=base_url, config=self._config, **kwargs)
 
         client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
@@ -149,15 +191,9 @@ def __init__(
         self.database_accounts = DatabaseAccountsOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
-        self.operations = Operations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
-        self.database = DatabaseOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
-        self.collection = CollectionOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
+        self.operations = Operations(self._client, self._config, self._serialize, self._deserialize)
+        self.database = DatabaseOperations(self._client, self._config, self._serialize, self._deserialize)
+        self.collection = CollectionOperations(self._client, self._config, self._serialize, self._deserialize)
         self.collection_region = CollectionRegionOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
@@ -170,9 +206,7 @@ def __init__(
         self.percentile_target = PercentileTargetOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
-        self.percentile = PercentileOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
+        self.percentile = PercentileOperations(self._client, self._config, self._serialize, self._deserialize)
         self.collection_partition_region = CollectionPartitionRegionOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
@@ -185,27 +219,19 @@ def __init__(
         self.partition_key_range_id_region = PartitionKeyRangeIdRegionOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
-        self.graph_resources = GraphResourcesOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
-        self.sql_resources = SqlResourcesOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
+        self.graph_resources = GraphResourcesOperations(self._client, self._config, self._serialize, self._deserialize)
+        self.sql_resources = SqlResourcesOperations(self._client, self._config, self._serialize, self._deserialize)
         self.mongo_db_resources = MongoDBResourcesOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
-        self.table_resources = TableResourcesOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
+        self.table_resources = TableResourcesOperations(self._client, self._config, self._serialize, self._deserialize)
         self.cassandra_resources = CassandraResourcesOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
         self.gremlin_resources = GremlinResourcesOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
-        self.locations = LocationsOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
+        self.locations = LocationsOperations(self._client, self._config, self._serialize, self._deserialize)
         self.data_transfer_jobs = DataTransferJobsOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
@@ -260,16 +286,9 @@ def __init__(
         self.restorable_table_resources = RestorableTableResourcesOperations(
             self._client, self._config, self._serialize, self._deserialize
         )
-        self.service = ServiceOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
+        self.service = ServiceOperations(self._client, self._config, self._serialize, self._deserialize)
 
-
-    def _send_request(
-        self,
-        request: HttpRequest,
-        **kwargs: Any
-    ) -> HttpResponse:
+    def _send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse:
         """Runs the network request through the client's chained policies.
 
         >>> from azure.core.rest import HttpRequest
@@ -278,7 +297,7 @@ def _send_request(
         >>> response = client._send_request(request)
         <HttpResponse: 200 OK>
 
-        For more information on this code flow, see https://aka.ms/azsdk/python/protocol/quickstart
+        For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request
 
         :param request: The network request you want to make. Required.
         :type request: ~azure.core.rest.HttpRequest

sdk/cosmos/azure-mgmt-cosmosdb/azure/mgmt/cosmosdb/_patch.py

@@ -28,4 +28,4 @@
 # This file is used for handwritten extensions to the generated code. Example:
 # https://github.com/Azure/azure-sdk-for-python/blob/main/doc/dev/customize_code/how-to-patch-sdk-code.md
 def patch_sdk():
-    pass
+    pass