]> git.ipfire.org Git - thirdparty/tornado.git/commitdiff
Fixed support for test generators 1153/head
authorNiko Wilbert <mail@nikowilbert.de>
Fri, 22 Aug 2014 08:48:42 +0000 (10:48 +0200)
committerNiko Wilbert <mail@nikowilbert.de>
Fri, 22 Aug 2014 08:48:42 +0000 (10:48 +0200)
This fixes the problem that AsyncTestCase no longer seemed to work with test generators (as supported by Nose, http://nose.readthedocs.org/en/latest/writing_tests.html#test-generators).

tornado/testing.py

index b3c47a03bf456b0165ac7ce16b3bf0a1db2e2415..5efd8334ede14e549d992a47d8d13b8bcae988ab 100644 (file)
@@ -114,8 +114,8 @@ class _TestMethodWrapper(object):
     def __init__(self, orig_method):
         self.orig_method = orig_method
 
-    def __call__(self):
-        result = self.orig_method()
+    def __call__(self, *args, **kwargs):
+        result = self.orig_method(*args, **kwargs)
         if isinstance(result, types.GeneratorType):
             raise TypeError("Generator test methods should be decorated with "
                             "tornado.testing.gen_test")