Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 08:11:00 +00:00
Code Issues Projects Releases Wiki Activity

test_dumper.py - Add sio_err everywhere + adjust test_simple

Browse Source
This commit is contained in:
Jessica Favin 2018-11-04 17:24:34 +01:00
parent 312f922316
commit 0cbbcffd89
1 changed files with 26 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
test/mitmproxy/addons/test_dumper.py
View File

@ -32,37 +32,50 @@ def test_configure():
def test_simple(): def test_simple():
sio = io.StringIO() sio = io.StringIO()
d = dumper.Dumper(sio) sio_err = io.StringIO()
d = dumper.Dumper(sio, sio_err)
with taddons.context(d) as ctx: with taddons.context(d) as ctx:
ctx.configure(d, flow_detail=0) ctx.configure(d, flow_detail=0)
d.response(tflow.tflow(resp=True)) d.response(tflow.tflow(resp=True))
assert not sio.getvalue() assert not sio.getvalue()
sio.truncate(0) sio.truncate(0)
assert not sio_err.getvalue()
sio_err.truncate(0)
ctx.configure(d, flow_detail=1) ctx.configure(d, flow_detail=1)
d.response(tflow.tflow(resp=True)) d.response(tflow.tflow(resp=True))
assert sio.getvalue() assert sio.getvalue()
sio.truncate(0) sio.truncate(0)
assert not sio_err.getvalue()
sio_err.truncate(0)
ctx.configure(d, flow_detail=1) ctx.configure(d, flow_detail=1)
d.error(tflow.tflow(err=True)) d.error(tflow.tflow(err=True))
assert sio.getvalue() assert sio.getvalue()
sio.truncate(0) sio.truncate(0)
assert not sio_err.getvalue()
sio_err.truncate(0)
ctx.configure(d, flow_detail=4) ctx.configure(d, flow_detail=4)
d.response(tflow.tflow(resp=True)) d.response(tflow.tflow(resp=True))
assert sio.getvalue() assert sio.getvalue()
sio.truncate(0) sio.truncate(0)
assert not sio_err.getvalue()
sio_err.truncate(0)
ctx.configure(d, flow_detail=4) ctx.configure(d, flow_detail=4)
d.response(tflow.tflow(resp=True)) d.response(tflow.tflow(resp=True))
assert "<<" in sio.getvalue() assert "<<" in sio.getvalue()
sio.truncate(0) sio.truncate(0)
assert not sio_err.getvalue()
sio_err.truncate(0)
ctx.configure(d, flow_detail=4) ctx.configure(d, flow_detail=4)
d.response(tflow.tflow(err=True)) d.response(tflow.tflow(err=True))
assert "<<" in sio.getvalue() assert "<<" in sio.getvalue()
sio.truncate(0) sio.truncate(0)
assert not sio_err.getvalue()
sio_err.truncate(0)
ctx.configure(d, flow_detail=4) ctx.configure(d, flow_detail=4)
flow = tflow.tflow() flow = tflow.tflow()
@ -75,6 +88,8 @@ def test_simple():
d.response(flow) d.response(flow)
assert sio.getvalue() assert sio.getvalue()
sio.truncate(0) sio.truncate(0)
assert not sio_err.getvalue()
sio_err.truncate(0)
ctx.configure(d, flow_detail=4) ctx.configure(d, flow_detail=4)
flow = tflow.tflow(resp=tutils.tresp(content=b"{")) flow = tflow.tflow(resp=tutils.tresp(content=b"{"))
@ -83,6 +98,8 @@ def test_simple():
d.response(flow) d.response(flow)
assert sio.getvalue() assert sio.getvalue()
sio.truncate(0) sio.truncate(0)
assert not sio_err.getvalue()
sio_err.truncate(0)
ctx.configure(d, flow_detail=4) ctx.configure(d, flow_detail=4)
flow = tflow.tflow() flow = tflow.tflow()
@ -92,6 +109,8 @@ def test_simple():
d.response(flow) d.response(flow)
assert "content missing" in sio.getvalue() assert "content missing" in sio.getvalue()
sio.truncate(0) sio.truncate(0)
assert not sio_err.getvalue()
sio_err.truncate(0)
def test_echo_body(): def test_echo_body():
@ -100,7 +119,8 @@ def test_echo_body():
f.response.content = b"foo bar voing\n" * 100 f.response.content = b"foo bar voing\n" * 100
sio = io.StringIO() sio = io.StringIO()
d = dumper.Dumper(sio) sio_err = io.StringIO()
d = dumper.Dumper(sio, sio_err)
with taddons.context(d) as ctx: with taddons.context(d) as ctx:
ctx.configure(d, flow_detail=3) ctx.configure(d, flow_detail=3)
d._echo_message(f.response) d._echo_message(f.response)
@ -110,7 +130,8 @@ def test_echo_body():
def test_echo_request_line(): def test_echo_request_line():
sio = io.StringIO() sio = io.StringIO()
d = dumper.Dumper(sio) sio_err = io.StringIO()
d = dumper.Dumper(sio, sio_err)
with taddons.context(d) as ctx: with taddons.context(d) as ctx:
ctx.configure(d, flow_detail=3, showhost=True) ctx.configure(d, flow_detail=3, showhost=True)
f = tflow.tflow(client_conn=None, server_conn=True, resp=True) f = tflow.tflow(client_conn=None, server_conn=True, resp=True)
@ -146,7 +167,8 @@ class TestContentView:
with mock.patch("mitmproxy.contentviews.auto.ViewAuto.__call__") as va: with mock.patch("mitmproxy.contentviews.auto.ViewAuto.__call__") as va:
va.side_effect = exceptions.ContentViewException("") va.side_effect = exceptions.ContentViewException("")
sio = io.StringIO() sio = io.StringIO()
d = dumper.Dumper(sio) sio_err = io.StringIO()
d = dumper.Dumper(sio, sio_err)
with taddons.context(d) as ctx: with taddons.context(d) as ctx:
ctx.configure(d, flow_detail=4) ctx.configure(d, flow_detail=4)
d.response(tflow.tflow()) d.response(tflow.tflow())