add set_read_checks

sdk/core/azure-core/azure/core/_pipeline_client_async.py

@@ -209,7 +209,7 @@ async def _make_pipeline_call(self, request, **kwargs):
                     # the body is loaded. instead of doing response.read(), going to set the body
                     # to the internal content
                     rest_response._content = response.body()  # pylint: disable=protected-access
-                    await rest_response.read()
+                    await rest_response._set_read_checks()  # pylint: disable=protected-access
                     await rest_response.close()
                 except Exception as exc:
                     await rest_response.close()

sdk/core/azure-core/azure/core/rest/_http_response_impl_async.py

@@ -46,6 +46,10 @@ class AsyncHttpResponseImpl(_HttpResponseBaseImpl, _AsyncHttpResponse):
     :keyword Callable stream_download_generator: The stream download generator that we use to stream the response.
     """
 
+    async def _set_read_checks(self):
+        self._is_stream_consumed = True
+        await self.close()
+
     async def read(self) -> bytes:
         """Read the response's bytes into memory.
 
@@ -57,7 +61,7 @@ async def read(self) -> bytes:
             async for part in self.iter_bytes():
                 parts.append(part)
             self._content = b"".join(parts)
-        self._is_stream_consumed = True
+        await self._set_read_checks()
         return self._content
 
     async def iter_raw(self) -> AsyncIterator[bytes]: