feat: ClientResponse class support methods iter_chunked and iter_chunks.

CHANGES/5324.feature

@@ -0,0 +1 @@
+ClientResponse class support methods iter_chunked and iter_chunks.

CONTRIBUTORS.txt

@@ -120,6 +120,7 @@ Eugene Nikolaiev
 Eugene Tolmachev
 Evan Kepner
 Evert Lammerts
+fatelei
 Felix Yan
 Fernanda Guimarães
 FichteFoll

aiohttp/client_reqrep.py

@@ -14,6 +14,7 @@
 from typing import (
     TYPE_CHECKING,
     Any,
+    AsyncIterator,
     Dict,
     Iterable,
     List,
@@ -986,6 +987,18 @@ async def wait_for_close(self) -> None:
                 self._writer = None
         self.release()
 
+    async def iter_chunked(self, n) -> AsyncIterator[bytes]:
+        async for data in self.content.iter_chunked(n):
+            yield data
+            for trace in self._traces:
+                await trace.send_response_chunk_received(self.method, self.url, data)
+
+    async def iter_chunks(self) -> AsyncIterator[bytes]:
+        async for data, _ in self.content.iter_chunks():
+            yield data
+            for trace in self._traces:
+                await trace.send_response_chunk_received(self.method, self.url, data)
+
     async def read(self) -> bytes:
         """Read response payload."""
         if self._body is None:

tests/test_client_response.py

@@ -1265,3 +1265,45 @@ def test_response_not_closed_after_get_ok(mocker) -> None:
     assert not response.ok
     assert not response.closed
     assert spy.call_count == 0
+
+
+async def test_iter_chunks(loop: Any, session: Any) -> None:
+    response = ClientResponse(
+        "get",
+        URL("http://def-cl-resp.org"),
+        request_info=mock.Mock(),
+        writer=mock.Mock(),
+        continue100=None,
+        timer=TimerNoop(),
+        traces=[],
+        loop=loop,
+        session=session,
+    )
+
+    content = response.content = mock.MagicMock()
+    content.iter_chunks.__aiter__.return_value = [b"payload"]
+
+    async for data in response.iter_chunked(1):
+        assert data == b"payload"
+    assert response._connection is None
+
+
+async def test_iter_chunked(loop: Any, session: Any) -> None:
+    response = ClientResponse(
+        "get",
+        URL("http://def-cl-resp.org"),
+        request_info=mock.Mock(),
+        writer=mock.Mock(),
+        continue100=None,
+        timer=TimerNoop(),
+        traces=[],
+        loop=loop,
+        session=session,
+    )
+
+    content = response.content = mock.MagicMock()
+    content.iter_chunked.__aiter__.return_value = [b"payload"]
+
+    async for data in response.iter_chunked(1):
+        assert data == b"payload"
+    assert response._connection is None