Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-27 02:24:18 +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 mitmproxy import options
from .. import tservers from .. import tservers
import pytest import pytest
from unittest import mock from unittest import mock as Mock
class UrlError(Exception):
pass
def mock_log(message):
if "Invalid URL" in message:
raise UrlError(message)
class TestFlowlist(tservers.MasterTest): class TestFlowlist(tservers.MasterTest):
@ -23,9 +14,9 @@ class TestFlowlist(tservers.MasterTest):
o = options.Options(**opts) o = options.Options(**opts)
return console.master.ConsoleMaster(o, proxy.DummyServer()) 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):
def test_new_request(self, test_func):
m = self.mkmaster() m = self.mkmaster()
x = flowlist.FlowListBox(m) 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") x.new_request("nonexistent url", "GET")
mock.assert_called_once_with(message = "Invalid URL: No hostname given")