azure-mgmt-storage 21.2.1__py3-none-any.whl → 22.1.0__py3-none-any.whl

azure/mgmt/storage/{v2023_01_01 → v2024_01_01}/aio/operations/_storage_accounts_operations.py

@@ -1,4 +1,4 @@
-# pylint: disable=too-many-lines,too-many-statements
+# pylint: disable=too-many-lines
 # coding=utf-8
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
@@ -8,7 +8,20 @@
 # --------------------------------------------------------------------------
 from io import IOBase
 import sys
-from typing import Any, AsyncIterable, Callable, Dict, IO, Literal, Optional, Type, TypeVar, Union, cast, overload
+from typing import (
+    Any,
+    AsyncIterable,
+    AsyncIterator,
+    Callable,
+    Dict,
+    IO,
+    Literal,
+    Optional,
+    TypeVar,
+    Union,
+    cast,
+    overload,
+)
 import urllib.parse
 
 from azure.core.async_paging import AsyncItemPaged, AsyncList
@@ -18,12 +31,13 @@ from azure.core.exceptions import (
     ResourceExistsError,
     ResourceNotFoundError,
     ResourceNotModifiedError,
+    StreamClosedError,
+    StreamConsumedError,
     map_error,
 )
 from azure.core.pipeline import PipelineResponse
-from azure.core.pipeline.transport import AsyncHttpResponse
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
-from azure.core.rest import HttpRequest
+from azure.core.rest import AsyncHttpResponse, HttpRequest
 from azure.core.tracing.decorator import distributed_trace
 from azure.core.tracing.decorator_async import distributed_trace_async
 from azure.core.utils import case_insensitive_dict
@@ -31,7 +45,6 @@ from azure.mgmt.core.exceptions import ARMErrorFormat
 from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling
 
 from ... import models as _models
-from ..._vendor import _convert_request
 from ...operations._storage_accounts_operations import (
     build_abort_hierarchical_namespace_migration_request,
     build_check_name_availability_request,
@@ -56,7 +69,7 @@ from ...operations._storage_accounts_operations import (
 if sys.version_info >= (3, 9):
     from collections.abc import MutableMapping
 else:
-    from typing import MutableMapping  # type: ignore  # pylint: disable=ungrouped-imports
+    from typing import MutableMapping  # type: ignore
 T = TypeVar("T")
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
@@ -67,7 +80,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         **DO NOT** instantiate this class directly.
 
         Instead, you should access the following operations through
-        :class:`~azure.mgmt.storage.v2023_01_01.aio.StorageManagementClient`'s
+        :class:`~azure.mgmt.storage.v2024_01_01.aio.StorageManagementClient`'s
         :attr:`storage_accounts` attribute.
     """
 
@@ -95,12 +108,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          Storage account names must be between 3 and 24 characters in length and use numbers and
          lower-case letters only. Required.
         :type account_name:
-         ~azure.mgmt.storage.v2023_01_01.models.StorageAccountCheckNameAvailabilityParameters
+         ~azure.mgmt.storage.v2024_01_01.models.StorageAccountCheckNameAvailabilityParameters
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
         :return: CheckNameAvailabilityResult or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.CheckNameAvailabilityResult
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.CheckNameAvailabilityResult
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -118,7 +131,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          Default value is "application/json".
         :paramtype content_type: str
         :return: CheckNameAvailabilityResult or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.CheckNameAvailabilityResult
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.CheckNameAvailabilityResult
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -133,13 +146,13 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          lower-case letters only. Is either a StorageAccountCheckNameAvailabilityParameters type or a
          IO[bytes] type. Required.
         :type account_name:
-         ~azure.mgmt.storage.v2023_01_01.models.StorageAccountCheckNameAvailabilityParameters or
+         ~azure.mgmt.storage.v2024_01_01.models.StorageAccountCheckNameAvailabilityParameters or
          IO[bytes]
         :return: CheckNameAvailabilityResult or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.CheckNameAvailabilityResult
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.CheckNameAvailabilityResult
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -150,7 +163,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
         cls: ClsType[_models.CheckNameAvailabilityResult] = kwargs.pop("cls", None)
 
@@ -171,7 +184,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -185,7 +197,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
-        deserialized = self._deserialize("CheckNameAvailabilityResult", pipeline_response)
+        deserialized = self._deserialize("CheckNameAvailabilityResult", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -198,8 +210,8 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         account_name: str,
         parameters: Union[_models.StorageAccountCreateParameters, IO[bytes]],
         **kwargs: Any
-    ) -> Optional[_models.StorageAccount]:
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+    ) -> AsyncIterator[bytes]:
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -210,9 +222,9 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-        cls: ClsType[Optional[_models.StorageAccount]] = kwargs.pop("cls", None)
+        cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None)
 
         content_type = content_type or "application/json"
         _json = None
@@ -233,10 +245,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
-        _stream = False
+        _decompress = kwargs.pop("decompress", True)
+        _stream = True
         pipeline_response: PipelineResponse = await self._client._pipeline.run(  # pylint: disable=protected-access
             _request, stream=_stream, **kwargs
         )
@@ -244,12 +256,14 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 202]:
+            try:
+                await response.read()  # Load the body in memory and close the socket
+            except (StreamConsumedError, StreamClosedError):
+                pass
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
-        deserialized = None
-        if response.status_code == 200:
-            deserialized = self._deserialize("StorageAccount", pipeline_response)
+        deserialized = response.stream_download(self._client._pipeline, decompress=_decompress)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -279,14 +293,14 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          lower-case letters only. Required.
         :type account_name: str
         :param parameters: The parameters to provide for the created account. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountCreateParameters
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountCreateParameters
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
         :return: An instance of AsyncLROPoller that returns either StorageAccount or the result of
          cls(response)
         :rtype:
-         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2023_01_01.models.StorageAccount]
+         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2024_01_01.models.StorageAccount]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -320,7 +334,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :return: An instance of AsyncLROPoller that returns either StorageAccount or the result of
          cls(response)
         :rtype:
-         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2023_01_01.models.StorageAccount]
+         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2024_01_01.models.StorageAccount]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -346,18 +360,18 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type account_name: str
         :param parameters: The parameters to provide for the created account. Is either a
          StorageAccountCreateParameters type or a IO[bytes] type. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountCreateParameters or
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountCreateParameters or
          IO[bytes]
         :return: An instance of AsyncLROPoller that returns either StorageAccount or the result of
          cls(response)
         :rtype:
-         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2023_01_01.models.StorageAccount]
+         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2024_01_01.models.StorageAccount]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
         cls: ClsType[_models.StorageAccount] = kwargs.pop("cls", None)
         polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True)
@@ -375,10 +389,11 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                 params=_params,
                 **kwargs
             )
+            await raw_result.http_response.read()  # type: ignore
         kwargs.pop("error_map", None)
 
         def get_long_running_output(pipeline_response):
-            deserialized = self._deserialize("StorageAccount", pipeline_response)
+            deserialized = self._deserialize("StorageAccount", pipeline_response.http_response)
             if cls:
                 return cls(pipeline_response, deserialized, {})  # type: ignore
             return deserialized
@@ -401,9 +416,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         )
 
     @distributed_trace_async
-    async def delete(  # pylint: disable=inconsistent-return-statements
-        self, resource_group_name: str, account_name: str, **kwargs: Any
-    ) -> None:
+    async def delete(self, resource_group_name: str, account_name: str, **kwargs: Any) -> None:
         """Deletes a storage account in Microsoft Azure.
 
         :param resource_group_name: The name of the resource group within the user's subscription. The
@@ -417,7 +430,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :rtype: None
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -428,7 +441,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[None] = kwargs.pop("cls", None)
 
         _request = build_delete_request(
@@ -439,7 +452,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -479,12 +491,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          data is not included when fetching properties. Currently we only support geoReplicationStats
          and blobRestoreStatus. Known values are: "geoReplicationStats" and "blobRestoreStatus". Default
          value is None.
-        :type expand: str or ~azure.mgmt.storage.v2023_01_01.models.StorageAccountExpand
+        :type expand: str or ~azure.mgmt.storage.v2024_01_01.models.StorageAccountExpand
         :return: StorageAccount or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.StorageAccount
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.StorageAccount
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -495,7 +507,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[_models.StorageAccount] = kwargs.pop("cls", None)
 
         _request = build_get_properties_request(
@@ -507,7 +519,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -521,7 +532,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
-        deserialized = self._deserialize("StorageAccount", pipeline_response)
+        deserialized = self._deserialize("StorageAccount", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -555,12 +566,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          lower-case letters only. Required.
         :type account_name: str
         :param parameters: The parameters to provide for the updated account. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountUpdateParameters
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountUpdateParameters
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
         :return: StorageAccount or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.StorageAccount
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.StorageAccount
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -596,7 +607,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          Default value is "application/json".
         :paramtype content_type: str
         :return: StorageAccount or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.StorageAccount
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.StorageAccount
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -626,13 +637,13 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type account_name: str
         :param parameters: The parameters to provide for the updated account. Is either a
          StorageAccountUpdateParameters type or a IO[bytes] type. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountUpdateParameters or
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountUpdateParameters or
          IO[bytes]
         :return: StorageAccount or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.StorageAccount
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.StorageAccount
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -643,7 +654,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
         cls: ClsType[_models.StorageAccount] = kwargs.pop("cls", None)
 
@@ -666,7 +677,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -680,7 +690,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
-        deserialized = self._deserialize("StorageAccount", pipeline_response)
+        deserialized = self._deserialize("StorageAccount", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -694,16 +704,16 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
 
         :return: An iterator like instance of either StorageAccount or the result of cls(response)
         :rtype:
-         ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storage.v2023_01_01.models.StorageAccount]
+         ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storage.v2024_01_01.models.StorageAccount]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[_models.StorageAccountListResult] = kwargs.pop("cls", None)
 
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -720,7 +730,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                     headers=_headers,
                     params=_params,
                 )
-                _request = _convert_request(_request)
                 _request.url = self._client.format_url(_request.url)
 
             else:
@@ -736,7 +745,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                 _request = HttpRequest(
                     "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params
                 )
-                _request = _convert_request(_request)
                 _request.url = self._client.format_url(_request.url)
                 _request.method = "GET"
             return _request
@@ -777,16 +785,16 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type resource_group_name: str
         :return: An iterator like instance of either StorageAccount or the result of cls(response)
         :rtype:
-         ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storage.v2023_01_01.models.StorageAccount]
+         ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.storage.v2024_01_01.models.StorageAccount]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[_models.StorageAccountListResult] = kwargs.pop("cls", None)
 
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -804,7 +812,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                     headers=_headers,
                     params=_params,
                 )
-                _request = _convert_request(_request)
                 _request.url = self._client.format_url(_request.url)
 
             else:
@@ -820,7 +827,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                 _request = HttpRequest(
                     "GET", urllib.parse.urljoin(next_link, _parsed_next_link.path), params=_next_request_params
                 )
-                _request = _convert_request(_request)
                 _request.url = self._client.format_url(_request.url)
                 _request.method = "GET"
             return _request
@@ -867,10 +873,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          "kerb" and None. Default value is "kerb".
         :type expand: str
         :return: StorageAccountListKeysResult or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountListKeysResult
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountListKeysResult
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -881,7 +887,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[_models.StorageAccountListKeysResult] = kwargs.pop("cls", None)
 
         _request = build_list_keys_request(
@@ -893,7 +899,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -907,7 +912,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
-        deserialized = self._deserialize("StorageAccountListKeysResult", pipeline_response)
+        deserialized = self._deserialize("StorageAccountListKeysResult", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -936,12 +941,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :param regenerate_key: Specifies name of the key which should be regenerated -- key1, key2,
          kerb1, kerb2. Required.
         :type regenerate_key:
-         ~azure.mgmt.storage.v2023_01_01.models.StorageAccountRegenerateKeyParameters
+         ~azure.mgmt.storage.v2024_01_01.models.StorageAccountRegenerateKeyParameters
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
         :return: StorageAccountListKeysResult or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountListKeysResult
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountListKeysResult
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -971,7 +976,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          Default value is "application/json".
         :paramtype content_type: str
         :return: StorageAccountListKeysResult or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountListKeysResult
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountListKeysResult
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -996,12 +1001,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          kerb1, kerb2. Is either a StorageAccountRegenerateKeyParameters type or a IO[bytes] type.
          Required.
         :type regenerate_key:
-         ~azure.mgmt.storage.v2023_01_01.models.StorageAccountRegenerateKeyParameters or IO[bytes]
+         ~azure.mgmt.storage.v2024_01_01.models.StorageAccountRegenerateKeyParameters or IO[bytes]
         :return: StorageAccountListKeysResult or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountListKeysResult
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountListKeysResult
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1012,7 +1017,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
         cls: ClsType[_models.StorageAccountListKeysResult] = kwargs.pop("cls", None)
 
@@ -1035,7 +1040,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -1049,7 +1053,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
-        deserialized = self._deserialize("StorageAccountListKeysResult", pipeline_response)
+        deserialized = self._deserialize("StorageAccountListKeysResult", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -1077,12 +1081,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type account_name: str
         :param parameters: The parameters to provide to list SAS credentials for the storage account.
          Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.AccountSasParameters
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.AccountSasParameters
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
         :return: ListAccountSasResponse or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.ListAccountSasResponse
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.ListAccountSasResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -1112,7 +1116,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          Default value is "application/json".
         :paramtype content_type: str
         :return: ListAccountSasResponse or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.ListAccountSasResponse
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.ListAccountSasResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -1135,12 +1139,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type account_name: str
         :param parameters: The parameters to provide to list SAS credentials for the storage account.
          Is either a AccountSasParameters type or a IO[bytes] type. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.AccountSasParameters or IO[bytes]
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.AccountSasParameters or IO[bytes]
         :return: ListAccountSasResponse or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.ListAccountSasResponse
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.ListAccountSasResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1151,7 +1155,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
         cls: ClsType[_models.ListAccountSasResponse] = kwargs.pop("cls", None)
 
@@ -1174,7 +1178,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -1188,7 +1191,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
-        deserialized = self._deserialize("ListAccountSasResponse", pipeline_response)
+        deserialized = self._deserialize("ListAccountSasResponse", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -1215,12 +1218,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          lower-case letters only. Required.
         :type account_name: str
         :param parameters: The parameters to provide to list service SAS credentials. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.ServiceSasParameters
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.ServiceSasParameters
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
         :return: ListServiceSasResponse or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.ListServiceSasResponse
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.ListServiceSasResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -1249,7 +1252,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          Default value is "application/json".
         :paramtype content_type: str
         :return: ListServiceSasResponse or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.ListServiceSasResponse
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.ListServiceSasResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -1272,12 +1275,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type account_name: str
         :param parameters: The parameters to provide to list service SAS credentials. Is either a
          ServiceSasParameters type or a IO[bytes] type. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.ServiceSasParameters or IO[bytes]
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.ServiceSasParameters or IO[bytes]
         :return: ListServiceSasResponse or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.ListServiceSasResponse
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.ListServiceSasResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1288,7 +1291,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
         cls: ClsType[_models.ListServiceSasResponse] = kwargs.pop("cls", None)
 
@@ -1311,7 +1314,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -1325,17 +1327,17 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
-        deserialized = self._deserialize("ListServiceSasResponse", pipeline_response)
+        deserialized = self._deserialize("ListServiceSasResponse", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
 
         return deserialized  # type: ignore
 
-    async def _failover_initial(  # pylint: disable=inconsistent-return-statements
+    async def _failover_initial(
         self, resource_group_name: str, account_name: str, failover_type: Literal["Planned"] = "Planned", **kwargs: Any
-    ) -> None:
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+    ) -> AsyncIterator[bytes]:
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1346,8 +1348,8 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
-        cls: ClsType[None] = kwargs.pop("cls", None)
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
+        cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None)
 
         _request = build_failover_request(
             resource_group_name=resource_group_name,
@@ -1358,10 +1360,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
-        _stream = False
+        _decompress = kwargs.pop("decompress", True)
+        _stream = True
         pipeline_response: PipelineResponse = await self._client._pipeline.run(  # pylint: disable=protected-access
             _request, stream=_stream, **kwargs
         )
@@ -1369,11 +1371,19 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 202]:
+            try:
+                await response.read()  # Load the body in memory and close the socket
+            except (StreamConsumedError, StreamClosedError):
+                pass
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
+        deserialized = response.stream_download(self._client._pipeline, decompress=_decompress)
+
         if cls:
-            return cls(pipeline_response, None, {})  # type: ignore
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore
 
     @distributed_trace_async
     async def begin_failover(
@@ -1408,13 +1418,13 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[None] = kwargs.pop("cls", None)
         polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True)
         lro_delay = kwargs.pop("polling_interval", self._config.polling_interval)
         cont_token: Optional[str] = kwargs.pop("continuation_token", None)
         if cont_token is None:
-            raw_result = await self._failover_initial(  # type: ignore
+            raw_result = await self._failover_initial(
                 resource_group_name=resource_group_name,
                 account_name=account_name,
                 failover_type=failover_type,
@@ -1424,6 +1434,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                 params=_params,
                 **kwargs
             )
+            await raw_result.http_response.read()  # type: ignore
         kwargs.pop("error_map", None)
 
         def get_long_running_output(pipeline_response):  # pylint: disable=inconsistent-return-statements
@@ -1447,10 +1458,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             )
         return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method)  # type: ignore
 
-    async def _hierarchical_namespace_migration_initial(  # pylint: disable=inconsistent-return-statements,name-too-long
+    async def _hierarchical_namespace_migration_initial(  # pylint: disable=name-too-long
         self, resource_group_name: str, account_name: str, request_type: str, **kwargs: Any
-    ) -> None:
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+    ) -> AsyncIterator[bytes]:
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1461,8 +1472,8 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
-        cls: ClsType[None] = kwargs.pop("cls", None)
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
+        cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None)
 
         _request = build_hierarchical_namespace_migration_request(
             resource_group_name=resource_group_name,
@@ -1473,10 +1484,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
-        _stream = False
+        _decompress = kwargs.pop("decompress", True)
+        _stream = True
         pipeline_response: PipelineResponse = await self._client._pipeline.run(  # pylint: disable=protected-access
             _request, stream=_stream, **kwargs
         )
@@ -1484,12 +1495,20 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 202]:
+            try:
+                await response.read()  # Load the body in memory and close the socket
+            except (StreamConsumedError, StreamClosedError):
+                pass
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response)
             raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
 
+        deserialized = response.stream_download(self._client._pipeline, decompress=_decompress)
+
         if cls:
-            return cls(pipeline_response, None, {})  # type: ignore
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore
 
     @distributed_trace_async
     async def begin_hierarchical_namespace_migration(
@@ -1516,13 +1535,13 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[None] = kwargs.pop("cls", None)
         polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True)
         lro_delay = kwargs.pop("polling_interval", self._config.polling_interval)
         cont_token: Optional[str] = kwargs.pop("continuation_token", None)
         if cont_token is None:
-            raw_result = await self._hierarchical_namespace_migration_initial(  # type: ignore
+            raw_result = await self._hierarchical_namespace_migration_initial(
                 resource_group_name=resource_group_name,
                 account_name=account_name,
                 request_type=request_type,
@@ -1532,6 +1551,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                 params=_params,
                 **kwargs
             )
+            await raw_result.http_response.read()  # type: ignore
         kwargs.pop("error_map", None)
 
         def get_long_running_output(pipeline_response):  # pylint: disable=inconsistent-return-statements
@@ -1555,10 +1575,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             )
         return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method)  # type: ignore
 
-    async def _abort_hierarchical_namespace_migration_initial(  # pylint: disable=inconsistent-return-statements,name-too-long
+    async def _abort_hierarchical_namespace_migration_initial(  # pylint: disable=name-too-long
         self, resource_group_name: str, account_name: str, **kwargs: Any
-    ) -> None:
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+    ) -> AsyncIterator[bytes]:
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1569,8 +1589,8 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
-        cls: ClsType[None] = kwargs.pop("cls", None)
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
+        cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None)
 
         _request = build_abort_hierarchical_namespace_migration_request(
             resource_group_name=resource_group_name,
@@ -1580,10 +1600,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
-        _stream = False
+        _decompress = kwargs.pop("decompress", True)
+        _stream = True
         pipeline_response: PipelineResponse = await self._client._pipeline.run(  # pylint: disable=protected-access
             _request, stream=_stream, **kwargs
         )
@@ -1591,12 +1611,20 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 202]:
+            try:
+                await response.read()  # Load the body in memory and close the socket
+            except (StreamConsumedError, StreamClosedError):
+                pass
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response)
             raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
 
+        deserialized = response.stream_download(self._client._pipeline, decompress=_decompress)
+
         if cls:
-            return cls(pipeline_response, None, {})  # type: ignore
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore
 
     @distributed_trace_async
     async def begin_abort_hierarchical_namespace_migration(  # pylint: disable=name-too-long
@@ -1618,13 +1646,13 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[None] = kwargs.pop("cls", None)
         polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True)
         lro_delay = kwargs.pop("polling_interval", self._config.polling_interval)
         cont_token: Optional[str] = kwargs.pop("continuation_token", None)
         if cont_token is None:
-            raw_result = await self._abort_hierarchical_namespace_migration_initial(  # type: ignore
+            raw_result = await self._abort_hierarchical_namespace_migration_initial(
                 resource_group_name=resource_group_name,
                 account_name=account_name,
                 api_version=api_version,
@@ -1633,6 +1661,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                 params=_params,
                 **kwargs
             )
+            await raw_result.http_response.read()  # type: ignore
         kwargs.pop("error_map", None)
 
         def get_long_running_output(pipeline_response):  # pylint: disable=inconsistent-return-statements
@@ -1656,14 +1685,14 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             )
         return AsyncLROPoller[None](self._client, raw_result, get_long_running_output, polling_method)  # type: ignore
 
-    async def _customer_initiated_migration_initial(  # pylint: disable=inconsistent-return-statements
+    async def _customer_initiated_migration_initial(
         self,
         resource_group_name: str,
         account_name: str,
         parameters: Union[_models.StorageAccountMigration, IO[bytes]],
         **kwargs: Any
-    ) -> None:
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+    ) -> AsyncIterator[bytes]:
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1674,9 +1703,9 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-        cls: ClsType[None] = kwargs.pop("cls", None)
+        cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None)
 
         content_type = content_type or "application/json"
         _json = None
@@ -1697,10 +1726,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
-        _stream = False
+        _decompress = kwargs.pop("decompress", True)
+        _stream = True
         pipeline_response: PipelineResponse = await self._client._pipeline.run(  # pylint: disable=protected-access
             _request, stream=_stream, **kwargs
         )
@@ -1708,6 +1737,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 202]:
+            try:
+                await response.read()  # Load the body in memory and close the socket
+            except (StreamConsumedError, StreamClosedError):
+                pass
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             error = self._deserialize.failsafe_deserialize(_models.ErrorResponseAutoGenerated, pipeline_response)
             raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
@@ -1716,8 +1749,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         if response.status_code == 202:
             response_headers["Location"] = self._deserialize("str", response.headers.get("Location"))
 
+        deserialized = response.stream_download(self._client._pipeline, decompress=_decompress)
+
         if cls:
-            return cls(pipeline_response, None, response_headers)  # type: ignore
+            return cls(pipeline_response, deserialized, response_headers)  # type: ignore
+
+        return deserialized  # type: ignore
 
     @overload
     async def begin_customer_initiated_migration(
@@ -1744,7 +1781,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type account_name: str
         :param parameters: The request parameters required to perform storage account migration.
          Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountMigration
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountMigration
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -1810,7 +1847,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type account_name: str
         :param parameters: The request parameters required to perform storage account migration. Is
          either a StorageAccountMigration type or a IO[bytes] type. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountMigration or IO[bytes]
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountMigration or IO[bytes]
         :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
         :rtype: ~azure.core.polling.AsyncLROPoller[None]
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -1818,14 +1855,14 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
         cls: ClsType[None] = kwargs.pop("cls", None)
         polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True)
         lro_delay = kwargs.pop("polling_interval", self._config.polling_interval)
         cont_token: Optional[str] = kwargs.pop("continuation_token", None)
         if cont_token is None:
-            raw_result = await self._customer_initiated_migration_initial(  # type: ignore
+            raw_result = await self._customer_initiated_migration_initial(
                 resource_group_name=resource_group_name,
                 account_name=account_name,
                 parameters=parameters,
@@ -1836,6 +1873,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                 params=_params,
                 **kwargs
             )
+            await raw_result.http_response.read()  # type: ignore
         kwargs.pop("error_map", None)
 
         def get_long_running_output(pipeline_response):  # pylint: disable=inconsistent-return-statements
@@ -1878,12 +1916,12 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type account_name: str
         :param migration_name: The name of the Storage Account Migration. It should always be
          'default'. "default" Required.
-        :type migration_name: str or ~azure.mgmt.storage.v2023_01_01.models.MigrationName
+        :type migration_name: str or ~azure.mgmt.storage.v2024_01_01.models.MigrationName
         :return: StorageAccountMigration or the result of cls(response)
-        :rtype: ~azure.mgmt.storage.v2023_01_01.models.StorageAccountMigration
+        :rtype: ~azure.mgmt.storage.v2024_01_01.models.StorageAccountMigration
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1894,7 +1932,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[_models.StorageAccountMigration] = kwargs.pop("cls", None)
 
         _request = build_get_customer_initiated_migration_request(
@@ -1906,7 +1944,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False
@@ -1921,7 +1958,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             error = self._deserialize.failsafe_deserialize(_models.ErrorResponseAutoGenerated, pipeline_response)
             raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
 
-        deserialized = self._deserialize("StorageAccountMigration", pipeline_response)
+        deserialized = self._deserialize("StorageAccountMigration", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -1934,8 +1971,8 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         account_name: str,
         parameters: Union[_models.BlobRestoreParameters, IO[bytes]],
         **kwargs: Any
-    ) -> _models.BlobRestoreStatus:
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+    ) -> AsyncIterator[bytes]:
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1946,9 +1983,9 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
-        cls: ClsType[_models.BlobRestoreStatus] = kwargs.pop("cls", None)
+        cls: ClsType[AsyncIterator[bytes]] = kwargs.pop("cls", None)
 
         content_type = content_type or "application/json"
         _json = None
@@ -1969,10 +2006,10 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
-        _stream = False
+        _decompress = kwargs.pop("decompress", True)
+        _stream = True
         pipeline_response: PipelineResponse = await self._client._pipeline.run(  # pylint: disable=protected-access
             _request, stream=_stream, **kwargs
         )
@@ -1980,14 +2017,14 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         response = pipeline_response.http_response
 
         if response.status_code not in [200, 202]:
+            try:
+                await response.read()  # Load the body in memory and close the socket
+            except (StreamConsumedError, StreamClosedError):
+                pass
             map_error(status_code=response.status_code, response=response, error_map=error_map)
             raise HttpResponseError(response=response, error_format=ARMErrorFormat)
 
-        if response.status_code == 200:
-            deserialized = self._deserialize("BlobRestoreStatus", pipeline_response)
-
-        if response.status_code == 202:
-            deserialized = self._deserialize("BlobRestoreStatus", pipeline_response)
+        deserialized = response.stream_download(self._client._pipeline, decompress=_decompress)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -2014,14 +2051,14 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
          lower-case letters only. Required.
         :type account_name: str
         :param parameters: The parameters to provide for restore blob ranges. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.BlobRestoreParameters
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.BlobRestoreParameters
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
         :return: An instance of AsyncLROPoller that returns either BlobRestoreStatus or the result of
          cls(response)
         :rtype:
-         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2023_01_01.models.BlobRestoreStatus]
+         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2024_01_01.models.BlobRestoreStatus]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -2052,7 +2089,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :return: An instance of AsyncLROPoller that returns either BlobRestoreStatus or the result of
          cls(response)
         :rtype:
-         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2023_01_01.models.BlobRestoreStatus]
+         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2024_01_01.models.BlobRestoreStatus]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -2075,17 +2112,17 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :type account_name: str
         :param parameters: The parameters to provide for restore blob ranges. Is either a
          BlobRestoreParameters type or a IO[bytes] type. Required.
-        :type parameters: ~azure.mgmt.storage.v2023_01_01.models.BlobRestoreParameters or IO[bytes]
+        :type parameters: ~azure.mgmt.storage.v2024_01_01.models.BlobRestoreParameters or IO[bytes]
         :return: An instance of AsyncLROPoller that returns either BlobRestoreStatus or the result of
          cls(response)
         :rtype:
-         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2023_01_01.models.BlobRestoreStatus]
+         ~azure.core.polling.AsyncLROPoller[~azure.mgmt.storage.v2024_01_01.models.BlobRestoreStatus]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
         _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
 
-        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         content_type: Optional[str] = kwargs.pop("content_type", _headers.pop("Content-Type", None))
         cls: ClsType[_models.BlobRestoreStatus] = kwargs.pop("cls", None)
         polling: Union[bool, AsyncPollingMethod] = kwargs.pop("polling", True)
@@ -2103,10 +2140,11 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
                 params=_params,
                 **kwargs
             )
+            await raw_result.http_response.read()  # type: ignore
         kwargs.pop("error_map", None)
 
         def get_long_running_output(pipeline_response):
-            deserialized = self._deserialize("BlobRestoreStatus", pipeline_response)
+            deserialized = self._deserialize("BlobRestoreStatus", pipeline_response.http_response)
             if cls:
                 return cls(pipeline_response, deserialized, {})  # type: ignore
             return deserialized
@@ -2131,9 +2169,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         )
 
     @distributed_trace_async
-    async def revoke_user_delegation_keys(  # pylint: disable=inconsistent-return-statements
-        self, resource_group_name: str, account_name: str, **kwargs: Any
-    ) -> None:
+    async def revoke_user_delegation_keys(self, resource_group_name: str, account_name: str, **kwargs: Any) -> None:
         """Revoke user delegation keys.
 
         :param resource_group_name: The name of the resource group within the user's subscription. The
@@ -2147,7 +2183,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         :rtype: None
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+        error_map: MutableMapping = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2158,7 +2194,7 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
         _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._api_version or "2023-01-01"))
+        api_version: str = kwargs.pop("api_version", _params.pop("api-version", self._api_version or "2024-01-01"))
         cls: ClsType[None] = kwargs.pop("cls", None)
 
         _request = build_revoke_user_delegation_keys_request(
@@ -2169,7 +2205,6 @@ class StorageAccountsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        _request = _convert_request(_request)
         _request.url = self._client.format_url(_request.url)
 
         _stream = False