testutil: Handle silencing logging for tests

1 files changed, 15 insertions, 1 deletions

diff --git a/tests/testutil.py b/tests/testutil.py
index 170b83b..e02ea70 100644
--- a/tests/testutil.py
+++ b/tests/testutil.py
@@ -4,17 +4,30 @@
 # Licensed under the terms of 2-clause BSD license
 
 import io
+import logging
 import os
 import os.path
 import tempfile
 import unittest
 
 
-class TempDirTestCase(unittest.TestCase):
+class LoggingTestCase(unittest.TestCase):
+    def setUp(self):
+        self.log = io.StringIO()
+        self.log_handler = logging.getLogger().addHandler(
+                logging.StreamHandler(self.log))
+
+    def tearDown(self):
+        # TODO: make some use of the log output?
+        logging.getLogger().removeHandler(self.log_handler)
+
+
+class TempDirTestCase(LoggingTestCase):
     DIRS = []
     FILES = {}
 
     def setUp(self):
+        super(TempDirTestCase, self).setUp()
         self.dir = tempfile.mkdtemp()
         for k in self.DIRS:
             os.mkdir(os.path.join(self.dir, k))
@@ -28,3 +41,4 @@ class TempDirTestCase(unittest.TestCase):
         for k in reversed(self.DIRS):
             os.rmdir(os.path.join(self.dir, k))
         os.rmdir(self.dir)
+        super(TempDirTestCase, self).tearDown()