Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 18:18:25 +00:00
Code Issues Projects Releases Wiki Activity

Update test_flowlist.py

Browse Source
This commit is contained in:
lymanZerga11 2017-02-12 23:23:23 +08:00 committed by GitHub
parent c622e4a649
commit df8a5aa9be
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,33 +1,31 @@
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):
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)