testutils: Add a context manager to capture stderr

and use it to test the help output

2 files changed, 31 insertions, 1 deletions

diff --git a/tests/testutils/__init__.py b/tests/testutils/__init__.py
index c94af648..092d377c 100644
--- a/tests/testutils/__init__.py
+++ b/tests/testutils/__init__.py
@@ -13,9 +13,10 @@ from gbp.deb.changelog import ChangeLog
 
 from . gbplogtester import GbpLogTester
 from . debiangittestrepo import DebianGitTestRepo
+from . capture import capture_stderr
 
 __all__ = ['GbpLogTester', 'DebianGitTestRepo', 'OsReleaseFile',
-           'MockedChangeLog', 'get_dch_default_urgency' ]
+           'MockedChangeLog', 'get_dch_default_urgency', 'capture_stderr' ]
 
 class OsReleaseFile(object):
     """Repesents a simple file with key-value pairs"""
diff --git a/tests/testutils/capture.py b/tests/testutils/capture.py
new file mode 100644
index 00000000..60d5a12e
--- /dev/null
+++ b/tests/testutils/capture.py
@@ -0,0 +1,29 @@
+# vim: set fileencoding=utf-8 :
+
+import sys
+from contextlib import contextmanager
+from StringIO import StringIO
+
+
+class StderrCapture(StringIO):
+    def save(self):
+        self.safed = sys.stderr
+        sys.stderr = self
+
+    def restore(self):
+        if self.safed is not None:
+            sys.stderr = self.safed
+            self.safed = None
+
+    def output(self):
+        self.seek(0)
+        return self.read()
+
+
+@contextmanager
+def capture_stderr():
+    """Capture an output and return it's content"""
+    c = StderrCapture()
+    c.save()
+    yield c
+    c.restore()