kylefang/mitmproxy
1
0
Fork 0
mirror of https://github.com/zhigang1992/mitmproxy.git synced 2026-03-26 08:54:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update test_flowlist.py

Browse Source
This commit is contained in:
lymanZerga11
2017-02-13 15:05:29 +08:00
committed by GitHub
parent a912d67c06
commit 26a17a3d82
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test/mitmproxy/console/test_flowlist.py
View File

@@ -11,8 +11,9 @@ class UrlError(Exception):
pass
def mock_add_log(message):
raise UrlError(message)
def mock_log(message):
if "Invalid URL" in message:
raise UrlError(message)
class TestFlowlist(tservers.MasterTest):
@@ -22,9 +23,9 @@ class TestFlowlist(tservers.MasterTest):
o = options.Options(**opts)
return console.master.ConsoleMaster(o, proxy.DummyServer())
@mock.patch('mitmproxy.tools.console.signals.status_message.send', side_effect = mock_add_log)
@mock.patch('mitmproxy.tools.console.signals.status_message.send', side_effect = mock_log)
def test_new_request(self, test_func):
m = self.mkmaster()
x = flowlist.FlowListBox(m)
with pytest.raises(UrlError) as e:
with pytest.raises(UrlError):
x.new_request("nonexistent url", "GET")