commit 17ef4319a34f5a2f95e7823dfb5f5b8cff11882d
author Richard Kojedzinszky <rkojedzinszky@users.noreply.github.com> Fri Dec 18 18:26:04 2020 +0100
committer GitHub <noreply@github.com> Fri Dec 18 09:26:04 2020 -0800
tree 769a121c883be857a22432fc1b5c57b80e60dd6a
parent 2179349d8cf45b1202775547df384b1fde31630a

bpo-41891: ensure asyncio.wait_for waits for task completion (#22461)

Lib/asyncio/tasks.py

diff --git a/Lib/asyncio/tasks.py b/Lib/asyncio/tasks.py
index 0d3a24b..52f1e66 100644
--- a/Lib/asyncio/tasks.py
+++ b/Lib/asyncio/tasks.py
@@ -437,7 +437,10 @@ async def wait_for(fut, timeout):
                 return fut.result()
             else:
                 fut.remove_done_callback(cb)
-                fut.cancel()
+                # We must ensure that the task is not running
+                # after wait_for() returns.
+                # See https://bugs.python.org/issue32751
+                await _cancel_and_wait(fut, loop=loop)
                 raise
 
         if fut.done():