diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2011-02-17 10:18:38 +1300 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2011-02-17 10:18:38 +1300 |
commit | f0f1fb4b556b048c5a8f93541bea751981136aba (patch) | |
tree | 453fab260c9066135d2a4d61909f886a1aa00ec6 /test/test_dump.py | |
parent | 7758385ac15e02d06c89db684fd75846a130c318 (diff) | |
download | mitmproxy-f0f1fb4b556b048c5a8f93541bea751981136aba.tar.gz mitmproxy-f0f1fb4b556b048c5a8f93541bea751981136aba.tar.bz2 mitmproxy-f0f1fb4b556b048c5a8f93541bea751981136aba.zip |
Add file writing to mitmdump.
Diffstat (limited to 'test/test_dump.py')
-rw-r--r-- | test/test_dump.py | 27 |
1 files changed, 26 insertions, 1 deletions
diff --git a/test/test_dump.py b/test/test_dump.py index f9a6bdf4..9aee5a31 100644 --- a/test/test_dump.py +++ b/test/test_dump.py @@ -1,6 +1,7 @@ +import os from cStringIO import StringIO import libpry -from libmproxy import dump +from libmproxy import dump, flow import utils @@ -28,6 +29,30 @@ class uDumpMaster(libpry.AutoTree): self._dummy_cycle(m) assert "GET" in cs.getvalue() + def test_write(self): + d = self.tmpdir() + p = os.path.join(d, "a") + o = dump.Options( + wfile = p, + verbosity = 0 + ) + cs = StringIO() + m = dump.DumpMaster(None, o, cs) + self._dummy_cycle(m) + del m + assert len(list(flow.FlowReader(open(p)).stream())) == 1 + + def test_write_err(self): + o = dump.Options( + wfile = "nonexistentdir/foo", + verbosity = 0 + ) + cs = StringIO() + libpry.raises(dump.DumpError, dump.DumpMaster, None, o, cs) + + + + tests = [ |