diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/mitmproxy/addons/test_termlog.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/test/mitmproxy/addons/test_termlog.py b/test/mitmproxy/addons/test_termlog.py index d9e18134..9d49a8c0 100644 --- a/test/mitmproxy/addons/test_termlog.py +++ b/test/mitmproxy/addons/test_termlog.py @@ -1,16 +1,18 @@ -import io - from mitmproxy.addons import termlog from mitmproxy import log from mitmproxy.tools import dump class TestTermLog: - def test_simple(self): - sio = io.StringIO() - t = termlog.TermLog(outfile=sio) + def test_simple(self, capsys): + t = termlog.TermLog() t.configure(dump.Options(verbosity = 2), set([])) t.log(log.LogEntry("one", "info")) - assert "one" in sio.getvalue() t.log(log.LogEntry("two", "debug")) - assert "two" not in sio.getvalue() + t.log(log.LogEntry("three", "warn")) + t.log(log.LogEntry("four", "error")) + out, err = capsys.readouterr() + assert "one" in out + assert "two" not in out + assert "three" in out + assert "four" in err |