diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/test_dump.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/test/test_dump.py b/test/test_dump.py index f860d1e0..95e7a98c 100644 --- a/test/test_dump.py +++ b/test/test_dump.py @@ -1,7 +1,7 @@ import os from cStringIO import StringIO import libpry -from libmproxy import dump, flow +from libmproxy import dump, flow, proxy import tutils class uStrFuncs(libpry.AutoTree): @@ -27,6 +27,7 @@ class uDumpMaster(libpry.AutoTree): m.handle_clientconnect(cc) m.handle_request(req) m.handle_response(resp) + m.handle_clientdisconnect(proxy.ClientDisconnect(cc)) def _dummy_cycle(self, filt, content, **options): cs = StringIO() @@ -73,7 +74,7 @@ class uDumpMaster(libpry.AutoTree): def test_basic(self): for i in (1, 2, 3): - assert "GET" in self._dummy_cycle("~s", "", verbosity=i) + assert "GET" in self._dummy_cycle("~s", "", verbosity=i, eventlog=True) assert "GET" in self._dummy_cycle("~s", "\x00\x00\x00", verbosity=i) assert "GET" in self._dummy_cycle("~s", "ascii", verbosity=i) |