From ee144d34a95f4365f44a53ddc51c95343ec6fd52 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Tue, 28 Feb 2023 18:03:36 +0100 Subject: [PATCH] Adjust core test to avoid lingering task (#88918) --- tests/test_core.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_core.py b/tests/test_core.py index eb81efae920..f627475270f 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -2034,7 +2034,8 @@ async def test_shutdown_does_not_block_on_shielded_tasks( ) -> None: """Ensure shutdown does not block on shielded tasks.""" result = asyncio.Future() - shielded_task = asyncio.shield(asyncio.sleep(10)) + sleep_task = asyncio.ensure_future(asyncio.sleep(10)) + shielded_task = asyncio.shield(sleep_task) async def test_task(): try: @@ -2050,3 +2051,6 @@ async def test_shutdown_does_not_block_on_shielded_tasks( assert result.done() assert task.done() assert time.monotonic() - start < 0.5 + + # Cleanup lingering task after test is done + sleep_task.cancel()