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

[web] Static_viewer coverage ++.

Browse Source
This commit is contained in:
Matthew Shao 2017-08-17 21:40:52 +08:00
parent af30930ae8
commit d008f179c3
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
test/mitmproxy/addons/test_static_viewer.py
View File

@ -1,4 +1,5 @@
import json import json
from unittest import mock
from mitmproxy.test import taddons from mitmproxy.test import taddons
from mitmproxy.test import tflow from mitmproxy.test import tflow
@ -32,8 +33,10 @@ def test_save_flows(tmpdir):
assert tmpdir.join('flows.json').read() == json.dumps([flow_to_json(f) for f in flows]) assert tmpdir.join('flows.json').read() == json.dumps([flow_to_json(f) for f in flows])
def test_save_flows_content(tmpdir): @mock.patch('mitmproxy.ctx.log')
def test_save_flows_content(ctx, tmpdir):
flows = [tflow.tflow(req=True, resp=None), tflow.tflow(req=True, resp=True)] flows = [tflow.tflow(req=True, resp=None), tflow.tflow(req=True, resp=True)]
with mock.patch('time.time', mock.Mock(side_effect=[1, 2, 2] * 4)):
static_viewer.save_flows_content(tmpdir, flows) static_viewer.save_flows_content(tmpdir, flows)
flows_path = tmpdir.join('flows') flows_path = tmpdir.join('flows')
assert len(flows_path.listdir()) == len(flows) assert len(flows_path.listdir()) == len(flows)