summary refs log tree commit diff
path: root/tests/unittest.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unittest.py')
-rw-r--r--tests/unittest.py23
1 files changed, 22 insertions, 1 deletions
diff --git a/tests/unittest.py b/tests/unittest.py
index 00c3c532eb..19be03b96a 100644
--- a/tests/unittest.py
+++ b/tests/unittest.py
@@ -27,4 +27,25 @@ logging.getLogger().setLevel(NEVER)
 
 
 class TestCase(unittest.TestCase):
-    pass
+    def __init__(self, *args, **kwargs):
+        super(TestCase, self).__init__(*args, **kwargs)
+
+        level = getattr(self, "loglevel", NEVER)
+
+        orig_setUp = self.setUp
+
+        def setUp():
+            old_level = logging.getLogger().level
+
+            if old_level != level:
+                orig_tearDown = self.tearDown
+
+                def tearDown():
+                    ret = orig_tearDown()
+                    logging.getLogger().setLevel(old_level)
+                    return ret
+                self.tearDown = tearDown
+
+            logging.getLogger().setLevel(level)
+            return orig_setUp()
+        self.setUp = setUp