From f028dc7e1bb030035ecb72fa0e9f175910d83656 Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Thu, 17 Feb 2011 10:44:08 +1300 Subject: Add filtering to mitmdump. --- test/test_dump.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) (limited to 'test/test_dump.py') diff --git a/test/test_dump.py b/test/test_dump.py index 9aee5a31..0d0a6219 100644 --- a/test/test_dump.py +++ b/test/test_dump.py @@ -19,13 +19,22 @@ class uDumpMaster(libpry.AutoTree): assert o.verbosity == 2 libpry.raises(AttributeError, dump.Options, nonexistent = 2) - def test_basic_verbosities(self): + def test_filter(self): + cs = StringIO() + o = dump.Options( + verbosity = 1 + ) + m = dump.DumpMaster(None, o, "~u foo", outfile=cs) + self._dummy_cycle(m) + assert not "GET" in cs.getvalue() + + def test_basic(self): for i in (1, 2, 3): cs = StringIO() o = dump.Options( verbosity = i ) - m = dump.DumpMaster(None, o, cs) + m = dump.DumpMaster(None, o, "~s", outfile=cs) self._dummy_cycle(m) assert "GET" in cs.getvalue() @@ -37,7 +46,7 @@ class uDumpMaster(libpry.AutoTree): verbosity = 0 ) cs = StringIO() - m = dump.DumpMaster(None, o, cs) + m = dump.DumpMaster(None, o, None, outfile=cs) self._dummy_cycle(m) del m assert len(list(flow.FlowReader(open(p)).stream())) == 1 @@ -48,7 +57,7 @@ class uDumpMaster(libpry.AutoTree): verbosity = 0 ) cs = StringIO() - libpry.raises(dump.DumpError, dump.DumpMaster, None, o, cs) + libpry.raises(dump.DumpError, dump.DumpMaster, None, o, None) -- cgit v1.2.3