diff options
| -rw-r--r-- | test/mitmproxy/console/test_flowlist.py | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/test/mitmproxy/console/test_flowlist.py b/test/mitmproxy/console/test_flowlist.py new file mode 100644 index 00000000..861be863 --- /dev/null +++ b/test/mitmproxy/console/test_flowlist.py @@ -0,0 +1,33 @@ +import mitmproxy.tools.console.flowlist as flowlist +from mitmproxy.test import tutils +from mitmproxy.test import tflow +from mitmproxy.tools import console +from mitmproxy import proxy +from mitmproxy import options +from mitmproxy.tools.console import common +from .. import mastertest +import pytest +from unittest import mock + +class ScriptError(Exception): + pass + +def mock_add_log(message): + raise ScriptError(message) + +class TestFlowlist(mastertest.MasterTest): + def mkmaster(self, **opts): + if "verbosity" not in opts: + opts["verbosity"] = 1 + 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) + def test_new_request(self,test_func): + m = self.mkmaster() + x = flowlist.FlowListBox(m) + with pytest.raises(ScriptError) as e: + x.new_request("nonexistent url", "GET") + assert "Invalid URL" in str(e) + + |