Issue #23140, asyncio: Fix cancellation of Process.wait(). Check the state of

vstinner · vstinner · commit c447ba04e78a · 2015-01-06T01:13:49.000+01:00
the waiter future before setting its result.
diff --git a/Lib/asyncio/subprocess.py b/Lib/asyncio/subprocess.py
@@ -96,7 +96,8 @@ def process_exited(self):
         returncode = self._transport.get_returncode()
         while self._waiters:
             waiter = self._waiters.popleft()
-            waiter.set_result(returncode)
+            if not waiter.cancelled():
+                waiter.set_result(returncode)
 
 
 class Process:
diff --git a/Lib/test/test_asyncio/test_subprocess.py b/Lib/test/test_asyncio/test_subprocess.py
@@ -223,6 +223,34 @@ def len_message(message):
         self.assertEqual(output.rstrip(), b'3')
         self.assertEqual(exitcode, 0)
 
+    def test_cancel_process_wait(self):
+        # Issue #23140: cancel Process.wait()
+
+        @asyncio.coroutine
+        def wait_proc(proc, event):
+            event.set()
+            yield from proc.wait()
+
+        @asyncio.coroutine
+        def cancel_wait():
+            proc = yield from asyncio.create_subprocess_exec(
+                                          *PROGRAM_BLOCKED,
+                                          loop=self.loop)
+
+            # Create an internal future waiting on the process exit
+            event = asyncio.Event(loop=self.loop)
+            task = self.loop.create_task(wait_proc(proc, event))
+            yield from event.wait()
+
+            # Cancel the future
+            task.cancel()
+
+            # Kill the process and wait until it is done
+            proc.kill()
+            yield from proc.wait()
+
+        self.loop.run_until_complete(cancel_wait())
+
 
 if sys.platform != 'win32':
     # Unix
