diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_dump.py | 4 | ||||
| -rw-r--r-- | test/test_server.py | 6 |
2 files changed, 6 insertions, 4 deletions
diff --git a/test/test_dump.py b/test/test_dump.py index f13245ed..5e2fcdc4 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 def test_strfuncs(): @@ -20,6 +20,8 @@ class TestDumpMaster: def _cycle(self, m, content): req = tutils.treq() req.content = content + l = proxy.Log("connect") + m.handle_log(l) cc = req.client_conn cc.connection_error = "error" resp = tutils.tresp(req) diff --git a/test/test_server.py b/test/test_server.py index e4a62fa5..1f83c496 100644 --- a/test/test_server.py +++ b/test/test_server.py @@ -55,6 +55,6 @@ class TestProxy(tutils.HTTPProxTest): assert f.status_code == 304 l = self.log() - assert l[0].address - assert "host" in l[1].headers - assert l[2].code == 304 + assert l[1].address + assert "host" in l[2].headers + assert l[3].code == 304 |