summaryrefslogtreecommitdiff
path: root/tests/testutils
diff options
context:
space:
mode:
Diffstat (limited to 'tests/testutils')
-rw-r--r--tests/testutils/__init__.py3
-rw-r--r--tests/testutils/capture.py29
2 files changed, 31 insertions, 1 deletions
diff --git a/tests/testutils/__init__.py b/tests/testutils/__init__.py
index c94af64..092d377 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 0000000..60d5a12
--- /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()