changeset:   90312:0c35d3616df5
branch:      3.4
parent:      90310:2755403682ad
user:        Yury Selivanov <yselivanov@sprymix.com>
date:        Mon Apr 14 22:21:52 2014 -0400
files:       Lib/asyncio/tasks.py Lib/test/test_asyncio/test_tasks.py Misc/NEWS
description:
asyncio.tasks: Fix CoroWrapper to workaround yield-from bug in CPython < 3.4.1

Closes issue #21209.


diff -r 2755403682ad -r 0c35d3616df5 Lib/asyncio/tasks.py
--- a/Lib/asyncio/tasks.py	Mon Apr 14 21:31:41 2014 -0400
+++ b/Lib/asyncio/tasks.py	Mon Apr 14 22:21:52 2014 -0400
@@ -49,7 +49,10 @@
     def __next__(self):
         return next(self.gen)
 
-    def send(self, value):
+    def send(self, *value):
+        # We use `*value` because of a bug in CPythons prior
+        # to 3.4.1. See issue #21209 and test_yield_from_corowrapper
+        # for details.  This workaround should be removed in 3.5.0.
         return self.gen.send(value)
 
     def throw(self, exc):
diff -r 2755403682ad -r 0c35d3616df5 Lib/test/test_asyncio/test_tasks.py
--- a/Lib/test/test_asyncio/test_tasks.py	Mon Apr 14 21:31:41 2014 -0400
+++ b/Lib/test/test_asyncio/test_tasks.py	Mon Apr 14 22:21:52 2014 -0400
@@ -1386,6 +1386,31 @@
         self.assertRaises(ValueError, self.loop.run_until_complete,
             asyncio.wait([], loop=self.loop))
 
+    def test_yield_from_corowrapper(self):
+        old_debug = asyncio.tasks._DEBUG
+        asyncio.tasks._DEBUG = True
+        try:
+            @asyncio.coroutine
+            def t1():
+                return (yield from t2())
+
+            @asyncio.coroutine
+            def t2():
+                f = asyncio.Future(loop=self.loop)
+                asyncio.Task(t3(f), loop=self.loop)
+                return (yield from f)
+
+            @asyncio.coroutine
+            def t3(f):
+                f.set_result((1, 2, 3))
+
+            task = asyncio.Task(t1(), loop=self.loop)
+            val = self.loop.run_until_complete(task)
+            self.assertEqual(val, (1, 2, 3))
+        finally:
+            asyncio.tasks._DEBUG = old_debug
+
+
 class GatherTestsBase:
 
     def setUp(self):
diff -r 2755403682ad -r 0c35d3616df5 Misc/NEWS
--- a/Misc/NEWS	Mon Apr 14 21:31:41 2014 -0400
+++ b/Misc/NEWS	Mon Apr 14 22:21:52 2014 -0400
@@ -147,6 +147,9 @@
   positional-or-keyword arguments passed as keyword arguments become
   keyword-only.
 
+- Issue #21209: Fix asyncio.tasks.CoroWrapper to workaround a bug
+  in yield-from implementation in CPythons prior to 3.4.1.
+
 IDLE
 ----