bpo-41891: ensure asyncio.wait_for waits for task completion

Author: rkojedzinszky

Lib/asyncio/tasks.py

@@ -469,7 +469,10 @@ async def wait_for(fut, timeout, *, loop=None):
                 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():

Misc/NEWS.d/next/Library/2020-09-30-13-35-29.bpo-41891.pNAeYI.rst

@@ -0,0 +1 @@
+Ensure asyncio.wait_for waits for task completion