Restore total_bytes to EmptyStreamReader (#10387)

(cherry picked from commit 95b28c7)
aio-libs · Feb 5, 2025 · d514063 · d514063
1 parent 6709f53
commit d514063
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/CHANGES/10387.bugfix.rst b/CHANGES/10387.bugfix.rst
@@ -0,0 +1 @@
+Restored the missing ``total_bytes`` attribute to ``EmptyStreamReader`` -- by :user:`bdraco`.
diff --git a/aiohttp/streams.py b/aiohttp/streams.py
@@ -555,6 +555,7 @@ class EmptyStreamReader(StreamReader):  # lgtm [py/missing-call-to-init]
 
     def __init__(self) -> None:
         self._read_eof_chunk = False
+        self.total_bytes = 0
 
     def __repr__(self) -> str:
         return "<%s>" % self.__class__.__name__

diff --git a/tests/test_streams.py b/tests/test_streams.py
@@ -1141,6 +1141,7 @@ async def test_empty_stream_reader() -> None:
     with pytest.raises(asyncio.IncompleteReadError):
         await s.readexactly(10)
     assert s.read_nowait() == b""
+    assert s.total_bytes == 0
 
 
 async def test_empty_stream_reader_iter_chunks() -> None:
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Restored the missing ``total_bytes`` attribute to ``EmptyStreamReader`` -- by :user:`bdraco`.