From 3c65510ef54c5773c22cd439488ae21dac95ee74 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Sat, 6 Sep 2014 13:09:57 +0200 Subject: coverage++ --- test/tutils.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'test/tutils.py') diff --git a/test/tutils.py b/test/tutils.py index dc049adb..05f65a21 100644 --- a/test/tutils.py +++ b/test/tutils.py @@ -1,5 +1,7 @@ +from cStringIO import StringIO import os, shutil, tempfile, argparse from contextlib import contextmanager +import sys from libmproxy import flow, utils, controller from libmproxy.protocol import http from libmproxy.proxy.connection import ClientConnection, ServerConnection @@ -185,4 +187,12 @@ def raises(exc, obj, *args, **kwargs): ) raise AssertionError("No exception raised.") + +@contextmanager +def capture_stderr(command, *args, **kwargs): + out, sys.stderr = sys.stderr, StringIO() + command(*args, **kwargs) + yield sys.stderr.getvalue() + sys.stderr = out + test_data = utils.Data(__name__) -- cgit v1.2.3