Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues 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
Show Stats 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")