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

Update test_dumper.py

Browse Source
This commit is contained in:
Jessica Favin 2018-11-04 16:06:17 +01:00
parent 2e9a34faa9
commit ade136dc4d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -155,6 +155,7 @@ class TestContentView:
def test_tcp(): def test_tcp():
sio = io.StringIO() sio = io.StringIO()
sio_err = io.StringIO()
d = dumper.Dumper(sio) d = dumper.Dumper(sio)
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)
@ -165,11 +166,12 @@ def test_tcp():
f = tflow.ttcpflow(client_conn=True, err=True) f = tflow.ttcpflow(client_conn=True, err=True)
d.tcp_error(f) d.tcp_error(f)
assert "Error in TCP" in sio.getvalue() assert "Error in TCP" in sio_err.getvalue()
def test_websocket(): def test_websocket():
sio = io.StringIO() sio = io.StringIO()
sio_err = io.StringIO()
d = dumper.Dumper(sio) d = dumper.Dumper(sio)
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)
@ -183,4 +185,4 @@ def test_websocket():
f = tflow.twebsocketflow(client_conn=True, err=True) f = tflow.twebsocketflow(client_conn=True, err=True)
d.websocket_error(f) d.websocket_error(f)
assert "Error in WebSocket" in sio.getvalue() assert "Error in WebSocket" in sio_err.getvalue()