diff options
-rw-r--r-- | test/mitmproxy/console/test_flowlist.py | 17 |
1 files changed, 4 insertions, 13 deletions
diff --git a/test/mitmproxy/console/test_flowlist.py b/test/mitmproxy/console/test_flowlist.py index 88895ac8..16c5ad07 100644 --- a/test/mitmproxy/console/test_flowlist.py +++ b/test/mitmproxy/console/test_flowlist.py @@ -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") |