make http_disconnect_while_sending test more robust in the face of sc…

…heduling issues
encode · Jul 2, 2022 · 1722e94 · 1722e94
1 parent 8a97bbb
commit 1722e94
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/tests/middleware/test_base.py b/tests/middleware/test_base.py
@@ -288,13 +288,18 @@ async def cancel_on_disconnect():
 
             task_group.start_soon(cancel_on_disconnect)
 
-            await send(
-                {
-                    "type": "http.response.body",
-                    "body": b"chunk",
-                    "more_body": True,
-                }
-            )
+            # A timeout is set for 0.1 second in order to ensure that
+            # cancel_on_disconnect is scheduled by the event loop
+            with anyio.move_on_after(0.1):
+                while True:
+                    await send(
+                        {
+                            "type": "http.response.body",
+                            "body": b"chunk ",
+                            "more_body": True,
+                        }
+                    )
+
             pytest.fail(
                 "http.disconnect should have been received and canceled the scope"
             )  # pragma: no cover