Browse Source

Additionally look first for a 'loglevel' attribute on the running test method, before the TestCase

Paul "LeoNerd" Evans 9 years ago
parent
commit
d9f3f322c5
1 changed files with 7 additions and 3 deletions
  1. 7 3
      tests/unittest.py

+ 7 - 3
tests/unittest.py

@@ -27,10 +27,14 @@ logging.getLogger().setLevel(NEVER)
 
 
 class TestCase(unittest.TestCase):
-    def __init__(self, *args, **kwargs):
-        super(TestCase, self).__init__(*args, **kwargs)
+    def __init__(self, methodName, *args, **kwargs):
+        super(TestCase, self).__init__(methodName, *args, **kwargs)
 
-        level = getattr(self, "loglevel", NEVER)
+        method = getattr(self, methodName)
+
+        level = getattr(method, "loglevel",
+                    getattr(self, "loglevel",
+                        NEVER))
 
         orig_setUp = self.setUp