Fixed coverage problems

Author: agronholm

httpcore/_models.py

@@ -486,10 +486,11 @@ async def aiter_stream(self) -> AsyncGenerator[bytes]:
             )
         self._stream_consumed = True
         async with AsyncExitStack() as stack:
-            if isasyncgen(self.stream):
-                stack.push_async_callback(self.stream.aclose)
+            iterator = self.stream.__aiter__()
+            if isasyncgen(iterator):
+                stack.push_async_callback(iterator.aclose)
 
-            async for chunk in self.stream:
+            async for chunk in iterator:
                 yield chunk
 
     async def aclose(self) -> None:

httpcore/_utils.py

@@ -4,9 +4,9 @@
 import socket
 import sys
 
-if sys.version_info >= (3, 10):
+if sys.version_info >= (3, 10):  # pragma: no cover
     from contextlib import aclosing as aclosing
-else:
+else:  # pragma: no cover
     from contextlib import AbstractAsyncContextManager
     from typing import Any, Awaitable, Protocol, TypeVar
 

tests/test_models.py

@@ -163,9 +163,6 @@ async def __aiter__(self) -> typing.AsyncIterator[bytes]:
         for chunk in self._chunks:
             yield chunk
 
-    async def aclose(self) -> None:
-        pass
-
 
 @pytest.mark.trio
 async def test_response_async_read():