]> granicus.if.org Git - python/commitdiff
Issue 24004: Add a unittest for @asyncio.coroutine supporting Awaitables
authorYury Selivanov <yselivanov@sprymix.com>
Sun, 31 May 2015 01:04:37 +0000 (21:04 -0400)
committerYury Selivanov <yselivanov@sprymix.com>
Sun, 31 May 2015 01:04:37 +0000 (21:04 -0400)
Lib/test/test_asyncio/test_pep492.py

index 8fc6cba1312e7632afbc7c4f0e2a6ea191c9019d..d123f088cb6aed319fb533f2f256fdfbece1d71e 100644 (file)
@@ -106,6 +106,19 @@ class CoroutineTests(BaseTest):
 
         self.assertTrue(asyncio.iscoroutine(FakeCoro()))
 
+    def test_function_returning_awaitable(self):
+        class Awaitable:
+            def __await__(self):
+                return ('spam',)
+
+        @asyncio.coroutine
+        def func():
+            return Awaitable()
+
+        coro = func()
+        self.assertEquals(coro.send(None), 'spam')
+        coro.close()
+
 
 if __name__ == '__main__':
     unittest.main()