]> git.ipfire.org Git - thirdparty/tornado.git/commitdiff
Use self.io_loop in with_timeout tests.
authorA. Jesse Jiryu Davis <jesse@mongodb.com>
Fri, 30 Jan 2015 20:08:08 +0000 (15:08 -0500)
committerBen Darnell <ben@bendarnell.com>
Sun, 1 Feb 2015 15:59:22 +0000 (10:59 -0500)
tornado/test/gen_test.py

index efc48644a72b796acf35a10d7a672aa42731998c..5d646d15007a0b100d5f9fa0c00623ba48690372 100644 (file)
@@ -1036,7 +1036,7 @@ class WithTimeoutTest(AsyncTestCase):
         self.io_loop.add_timeout(datetime.timedelta(seconds=0.1),
                                  lambda: future.set_result('asdf'))
         result = yield gen.with_timeout(datetime.timedelta(seconds=3600),
-                                        future)
+                                        future, io_loop=self.io_loop)
         self.assertEqual(result, 'asdf')
 
     @gen_test
@@ -1046,14 +1046,15 @@ class WithTimeoutTest(AsyncTestCase):
             datetime.timedelta(seconds=0.1),
             lambda: future.set_exception(ZeroDivisionError()))
         with self.assertRaises(ZeroDivisionError):
-            yield gen.with_timeout(datetime.timedelta(seconds=3600), future)
+            yield gen.with_timeout(datetime.timedelta(seconds=3600),
+                                   future, io_loop=self.io_loop)
 
     @gen_test
     def test_already_resolved(self):
         future = Future()
         future.set_result('asdf')
         result = yield gen.with_timeout(datetime.timedelta(seconds=3600),
-                                        future)
+                                        future, io_loop=self.io_loop)
         self.assertEqual(result, 'asdf')
 
     @unittest.skipIf(futures is None, 'futures module not present')