Use self.io_loop in with_timeout tests.

tornadoweb · Feb 1, 2015 · 2716458 · 2716458
1 parent eb4649e
commit 2716458
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tornado/test/gen_test.py b/tornado/test/gen_test.py
@@ -1036,7 +1036,7 @@ def test_completes_before_timeout(self):
         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 @@ def test_fails_before_timeout(self):
             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')