code and test (#34594)

Co-authored-by: azure-sdk <PythonSdkPipelines>

Author: azure-sdk

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

@@ -1,5 +1,20 @@
 # Release History
 
+## 10.0.0b3 (2024-03-18)
+
+### Features Added
+
+  - Added operation DataTransferJobsOperations.complete
+  - Model DatabaseAccountCreateUpdateParameters has a new parameter enable_per_region_per_partition_autoscale
+  - Model DatabaseAccountGetResults has a new parameter enable_per_region_per_partition_autoscale
+  - Model DatabaseAccountUpdateParameters has a new parameter enable_per_region_per_partition_autoscale
+  - Model PrivateEndpointConnection has a new parameter system_data
+  - Model ProxyResource has a new parameter system_data
+  - Model Resource has a new parameter system_data
+  - Model ResourceRestoreParameters has a new parameter restore_with_ttl_disabled
+  - Model RestoreParameters has a new parameter restore_with_ttl_disabled
+  - Model RestoreParametersBase has a new parameter restore_with_ttl_disabled
+
 ## 10.0.0b2 (2024-01-26)
 
 ### Features Added

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

@@ -1,7 +1,7 @@
 # Microsoft Azure SDK for Python
 
 This is the Microsoft Azure Cosmos DB Management Client Library.
-This package has been tested with Python 3.7+.
+This package has been tested with Python 3.8+.
 For a more complete view of Azure libraries, see the [azure sdk python release](https://aka.ms/azsdk/python/all).
 
 ## _Disclaimer_
@@ -12,7 +12,7 @@ _Azure SDK Python packages support for Python 2.7 has ended 01 January 2022. For
 
 ### Prerequisites
 
-- Python 3.7+ is required to use this package.
+- Python 3.8+ is required to use this package.
 - [Azure subscription](https://azure.microsoft.com/free/)
 
 ### Install the package

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

@@ -1,5 +1,5 @@
 {
-  "commit": "8e674dd2a88ae73868c6fa7593a0ba4371e45991",
+  "commit": "01a71545e82bb98b8137d3038150c436d46a59ed",
   "repository_url": "https://github.com/Azure/azure-rest-api-specs",
   "autorest": "3.9.7",
   "use": [

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

@@ -29,14 +29,14 @@ class CosmosDBManagementClientConfiguration(Configuration):  # pylint: disable=t
     :type credential: ~azure.core.credentials.TokenCredential
     :param subscription_id: The ID of the target subscription. Required.
     :type subscription_id: str
-    :keyword api_version: Api Version. Default value is "2023-11-15-preview". Note that overriding
+    :keyword api_version: Api Version. Default value is "2024-02-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:
         super(CosmosDBManagementClientConfiguration, self).__init__(**kwargs)
-        api_version: str = kwargs.pop("api_version", "2023-11-15-preview")
+        api_version: str = kwargs.pop("api_version", "2024-02-15-preview")
 
         if credential is None:
             raise ValueError("Parameter 'credential' must not be None.")

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

@@ -182,7 +182,7 @@ class CosmosDBManagementClient:  # pylint: disable=client-accepts-api-version-ke
     :type subscription_id: str
     :param base_url: Service URL. Default value is "https://management.azure.com".
     :type base_url: str
-    :keyword api_version: Api Version. Default value is "2023-11-15-preview". Note that overriding
+    :keyword api_version: Api Version. Default value is "2024-02-15-preview". Note that overriding
      this default value may result in unsupported behavior.
     :paramtype api_version: str
     :keyword int polling_interval: Default waiting time between two polls for LRO operations if no

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

@@ -6,4 +6,4 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "10.0.0b2"
+VERSION = "10.0.0b3"

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

@@ -29,14 +29,14 @@ class CosmosDBManagementClientConfiguration(Configuration):  # pylint: disable=t
     :type credential: ~azure.core.credentials_async.AsyncTokenCredential
     :param subscription_id: The ID of the target subscription. Required.
     :type subscription_id: str
-    :keyword api_version: Api Version. Default value is "2023-11-15-preview". Note that overriding
+    :keyword api_version: Api Version. Default value is "2024-02-15-preview". Note that overriding
      this default value may result in unsupported behavior.
     :paramtype api_version: str
     """
 
     def __init__(self, credential: "AsyncTokenCredential", subscription_id: str, **kwargs: Any) -> None:
         super(CosmosDBManagementClientConfiguration, self).__init__(**kwargs)
-        api_version: str = kwargs.pop("api_version", "2023-11-15-preview")
+        api_version: str = kwargs.pop("api_version", "2024-02-15-preview")
 
         if credential is None:
             raise ValueError("Parameter 'credential' must not be None.")

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

@@ -185,7 +185,7 @@ class CosmosDBManagementClient:  # pylint: disable=client-accepts-api-version-ke
     :type subscription_id: str
     :param base_url: Service URL. Default value is "https://management.azure.com".
     :type base_url: str
-    :keyword api_version: Api Version. Default value is "2023-11-15-preview". Note that overriding
+    :keyword api_version: Api Version. Default value is "2024-02-15-preview". Note that overriding
      this default value may result in unsupported behavior.
     :paramtype api_version: str
     :keyword int polling_interval: Default waiting time between two polls for LRO operations if no

sdk/cosmos/azure-mgmt-cosmosdb/azure/mgmt/cosmosdb/aio/operations/_cassandra_clusters_operations.py

@@ -1608,7 +1608,7 @@ async def get_backup(
     }
 
     async def _deallocate_initial(  # pylint: disable=inconsistent-return-statements
-        self, resource_group_name: str, cluster_name: str, x_ms_force_deallocate: Optional[bool] = None, **kwargs: Any
+        self, resource_group_name: str, cluster_name: str, x_ms_force_deallocate: Optional[str] = None, **kwargs: Any
     ) -> None:
         error_map = {
             401: ClientAuthenticationError,
@@ -1657,7 +1657,7 @@ async def _deallocate_initial(  # pylint: disable=inconsistent-return-statements
 
     @distributed_trace_async
     async def begin_deallocate(
-        self, resource_group_name: str, cluster_name: str, x_ms_force_deallocate: Optional[bool] = None, **kwargs: Any
+        self, resource_group_name: str, cluster_name: str, x_ms_force_deallocate: Optional[str] = None, **kwargs: Any
     ) -> AsyncLROPoller[None]:
         """Deallocate the Managed Cassandra Cluster and Associated Data Centers. Deallocation will
         deallocate the host virtual machine of this cluster, and reserved the data disk. This won't do
@@ -1671,7 +1671,7 @@ async def begin_deallocate(
         :param x_ms_force_deallocate: Force to deallocate a cluster of Cluster Type Production. Force
          to deallocate a cluster of Cluster Type Production might cause data loss. Default value is
          None.
-        :type x_ms_force_deallocate: bool
+        :type x_ms_force_deallocate: str
         :keyword callable cls: A custom type or function that will be passed the direct response
         :keyword str continuation_token: A continuation token to restart a poller from a saved state.
         :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for

sdk/cosmos/azure-mgmt-cosmosdb/azure/mgmt/cosmosdb/aio/operations/_data_transfer_jobs_operations.py

@@ -31,6 +31,7 @@
 from ..._vendor import _convert_request
 from ...operations._data_transfer_jobs_operations import (
     build_cancel_request,
+    build_complete_request,
     build_create_request,
     build_get_request,
     build_list_by_database_account_request,
@@ -480,6 +481,73 @@ async def cancel(
         "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/dataTransferJobs/{jobName}/cancel"
     }
 
+    @distributed_trace_async
+    async def complete(
+        self, resource_group_name: str, account_name: str, job_name: str, **kwargs: Any
+    ) -> _models.DataTransferJobGetResults:
+        """Completes a Data Transfer Online Job.
+
+        :param resource_group_name: The name of the resource group. The name is case insensitive.
+         Required.
+        :type resource_group_name: str
+        :param account_name: Cosmos DB database account name. Required.
+        :type account_name: str
+        :param job_name: Name of the Data Transfer Job. Required.
+        :type job_name: str
+        :keyword callable cls: A custom type or function that will be passed the direct response
+        :return: DataTransferJobGetResults or the result of cls(response)
+        :rtype: ~azure.mgmt.cosmosdb.models.DataTransferJobGetResults
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        error_map = {
+            401: ClientAuthenticationError,
+            404: ResourceNotFoundError,
+            409: ResourceExistsError,
+            304: ResourceNotModifiedError,
+        }
+        error_map.update(kwargs.pop("error_map", {}) or {})
+
+        _headers = kwargs.pop("headers", {}) or {}
+        _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
+
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._config.api_version))
+        cls: ClsType[_models.DataTransferJobGetResults] = kwargs.pop("cls", None)
+
+        request = build_complete_request(
+            resource_group_name=resource_group_name,
+            account_name=account_name,
+            job_name=job_name,
+            subscription_id=self._config.subscription_id,
+            api_version=api_version,
+            template_url=self.complete.metadata["url"],
+            headers=_headers,
+            params=_params,
+        )
+        request = _convert_request(request)
+        request.url = self._client.format_url(request.url)
+
+        _stream = False
+        pipeline_response: PipelineResponse = await self._client._pipeline.run(  # pylint: disable=protected-access
+            request, stream=_stream, **kwargs
+        )
+
+        response = pipeline_response.http_response
+
+        if response.status_code not in [200]:
+            map_error(status_code=response.status_code, response=response, error_map=error_map)
+            raise HttpResponseError(response=response, error_format=ARMErrorFormat)
+
+        deserialized = self._deserialize("DataTransferJobGetResults", pipeline_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})
+
+        return deserialized
+
+    complete.metadata = {
+        "url": "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/{accountName}/dataTransferJobs/{jobName}/complete"
+    }
+
     @distributed_trace
     def list_by_database_account(
         self, resource_group_name: str, account_name: str, **kwargs: Any