test: fix stalling future (microsoft#2282)

mxschmitt · web-flow · commit 752467d1a40f · 2024-02-09T14:52:17.000+01:00
diff --git a/tests/async/test_unroute_behavior.py b/tests/async/test_unroute_behavior.py
@@ -351,6 +351,7 @@ async def _unroute_promise() -> None:
 async def test_page_close_does_not_wait_for_active_route_handlers(
     page: Page, server: Server
 ) -> None:
+    stalling_future: "asyncio.Future[None]" = asyncio.Future()
     second_handler_called = False
 
     def _handler1(route: Route) -> None:
@@ -365,7 +366,7 @@ def _handler1(route: Route) -> None:
 
     async def _handler2(route: Route) -> None:
         route_future.set_result(route)
-        await asyncio.Future()
+        await stalling_future
 
     await page.route(
         "**/*",
@@ -383,6 +384,7 @@ async def _goto_ignore_exceptions() -> None:
     await page.close()
     await asyncio.sleep(0.5)
     assert not second_handler_called
+    stalling_future.cancel()
 
 
 async def test_route_continue_should_not_throw_if_page_has_been_closed(
