From: Ben Darnell <ben@bendarnell.com>
Date: Sun, 26 Mar 2017 15:57:09 +0000 (-0400)
Subject: gen: Add tests for `yield None`
X-Git-Tag: v4.5.0~18^2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=528ebe97cfceafbb818c56da3f60cafce9eec56c;p=thirdparty%2Ftornado.git

gen: Add tests for `yield None`

See #1716
---

diff --git a/tornado/test/gen_test.py b/tornado/test/gen_test.py
index 73ccd4e9d..bdaf889bd 100644
--- a/tornado/test/gen_test.py
+++ b/tornado/test/gen_test.py
@@ -276,6 +276,13 @@ class GenEngineTest(AsyncTestCase):
             pass
         self.orphaned_callback()
 
+    def test_none(self):
+        @gen.engine
+        def f():
+            yield None
+            self.stop()
+        self.run_gen(f)
+
     def test_multi(self):
         @gen.engine
         def f():
@@ -752,6 +759,21 @@ class GenCoroutineTest(AsyncTestCase):
         self.assertEqual(result, 42)
         self.finished = True
 
+    @skipBefore35
+    @gen_test
+    def test_asyncio_sleep_zero(self):
+        # asyncio.sleep(0) turns into a special case (equivalent to
+        # `yield None`)
+        namespace = exec_test(globals(), locals(), """
+        async def f():
+            import asyncio
+            await asyncio.sleep(0)
+            return 42
+        """)
+        result = yield namespace['f']()
+        self.assertEqual(result, 42)
+        self.finished = True
+
     @skipBefore35
     @gen_test
     def test_async_await_mixed_multi_native_future(self):