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-14 21:01:01 +08:00 committed by GitHub
parent 26a17a3d82
commit a52d8c1dab
1 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -4,16 +4,7 @@ from mitmproxy import proxy
from mitmproxy import options
from .. import tservers
import pytest
from unittest import mock
class UrlError(Exception):
pass
def mock_log(message):
if "Invalid URL" in message:
raise UrlError(message)
from unittest import mock as Mock
class TestFlowlist(tservers.MasterTest):
@ -23,9 +14,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_log)
def test_new_request(self, test_func):
def test_new_request(self):
m = self.mkmaster()
x = flowlist.FlowListBox(m)
with pytest.raises(UrlError):
with Mock.patch('mitmproxy.tools.console.signals.status_message.send') as mock:
x.new_request("nonexistent url", "GET")
mock.assert_called_once_with(message = "Invalid URL: No hostname given")