test: synchronize stream cancellation test on an event instead of a sleep (#188)

test_cancellation_propagates_cleanly slept 10 ms and assumed body consumption had begun and that the body's 1 s sleep was still running when it cancelled. A stall over 1 s let the body finish, and a cancel before consumption began still passed. The body now signals once the first chunk is consumed and then waits on an event that is never set.

Author: lesnik512

tests/test_client_stream.py

@@ -162,10 +162,13 @@ def handler(request: httpx2.Request) -> httpx2.Response:  # noqa: ARG001
 
 
 async def test_cancellation_propagates_cleanly() -> None:
+    first_chunk_consumed = asyncio.Event()
+
     async def slow_body() -> typing.AsyncIterator[bytes]:
         yield b"first"
-        await asyncio.sleep(1.0)
-        yield b"second"  # pragma: no cover — the task is cancelled during the sleep before this yield
+        first_chunk_consumed.set()
+        await asyncio.Event().wait()
+        yield b"second"  # pragma: no cover — the task is cancelled while the body waits before this yield
 
     def handler(request: httpx2.Request) -> httpx2.Response:
         return httpx2.Response(HTTPStatus.OK, request=request, content=slow_body())
@@ -178,7 +181,8 @@ async def consume() -> None:
                 pass
 
     task = asyncio.create_task(consume())
-    await asyncio.sleep(0.01)  # let body consumption begin
+    async with asyncio.timeout(5.0):
+        await first_chunk_consumed.wait()
     task.cancel()
     with pytest.raises(asyncio.CancelledError):
         await task