From d008f179c3e6fa620c6babc6cedf8e7df9093827 Mon Sep 17 00:00:00 2001 From: Matthew Shao Date: Thu, 17 Aug 2017 21:40:52 +0800 Subject: [PATCH] [web] Static_viewer coverage ++. --- test/mitmproxy/addons/test_static_viewer.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/test/mitmproxy/addons/test_static_viewer.py b/test/mitmproxy/addons/test_static_viewer.py index 7e34a9fc0..bb2b6777e 100644 --- a/test/mitmproxy/addons/test_static_viewer.py +++ b/test/mitmproxy/addons/test_static_viewer.py @@ -1,4 +1,5 @@ import json +from unittest import mock from mitmproxy.test import taddons from mitmproxy.test import tflow @@ -32,9 +33,11 @@ def test_save_flows(tmpdir): 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)] - static_viewer.save_flows_content(tmpdir, flows) + with mock.patch('time.time', mock.Mock(side_effect=[1, 2, 2] * 4)): + static_viewer.save_flows_content(tmpdir, flows) flows_path = tmpdir.join('flows') assert len(flows_path.listdir()) == len(flows) for p in flows_path.listdir():