aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorMaximilian Hils <git@maximilianhils.com>2014-11-26 04:18:21 +0100
committerMaximilian Hils <git@maximilianhils.com>2014-11-26 04:18:21 +0100
commited8249023fb7c0d429b9278c63b51ac071700987 (patch)
tree8bd83354e7484f15029ec678744234bd32e9906d /test
parent47a78e3c729f4ddb7971b72bfae30140562f4dd6 (diff)
downloadmitmproxy-ed8249023fb7c0d429b9278c63b51ac071700987.tar.gz
mitmproxy-ed8249023fb7c0d429b9278c63b51ac071700987.tar.bz2
mitmproxy-ed8249023fb7c0d429b9278c63b51ac071700987.zip
introduce revised views, port over changes from multiple_views branch
Diffstat (limited to 'test')
-rw-r--r--test/test_console.py8
-rw-r--r--test/test_flow.py40
-rw-r--r--test/test_server.py20
3 files changed, 34 insertions, 34 deletions
diff --git a/test/test_console.py b/test/test_console.py
index 3b6c941d..d66bd8b0 100644
--- a/test/test_console.py
+++ b/test/test_console.py
@@ -15,7 +15,7 @@ class TestConsoleState:
"""
c = console.ConsoleState()
f = self._add_request(c)
- assert f in c._flow_list
+ assert f in c.flows
assert c.get_focus() == (f, 0)
def test_focus(self):
@@ -52,19 +52,19 @@ class TestConsoleState:
def _add_request(self, state):
f = tutils.tflow()
- return state.add_request(f)
+ return state.add_flow(f)
def _add_response(self, state):
f = self._add_request(state)
f.response = tutils.tresp()
- state.add_response(f)
+ state.update_flow(f)
def test_add_response(self):
c = console.ConsoleState()
f = self._add_request(c)
f.response = tutils.tresp()
c.focus = None
- c.add_response(f)
+ c.update_flow(f)
def test_focus_view(self):
c = console.ConsoleState()
diff --git a/test/test_flow.py b/test/test_flow.py
index 22abb4d4..fdfac62f 100644
--- a/test/test_flow.py
+++ b/test/test_flow.py
@@ -364,7 +364,7 @@ class TestState:
def test_backup(self):
c = flow.State()
f = tutils.tflow()
- c.add_request(f)
+ c.add_flow(f)
f.backup()
c.revert(f)
@@ -376,42 +376,42 @@ class TestState:
"""
c = flow.State()
f = tutils.tflow()
- c.add_request(f)
+ c.add_flow(f)
assert f
assert c.flow_count() == 1
assert c.active_flow_count() == 1
newf = tutils.tflow()
- assert c.add_request(newf)
+ assert c.add_flow(newf)
assert c.active_flow_count() == 2
f.response = tutils.tresp()
- assert c.add_response(f)
+ assert c.update_flow(f)
assert c.flow_count() == 2
assert c.active_flow_count() == 1
_ = tutils.tresp()
- assert not c.add_response(None)
+ assert not c.update_flow(None)
assert c.active_flow_count() == 1
newf.response = tutils.tresp()
- assert c.add_response(newf)
+ assert c.update_flow(newf)
assert c.active_flow_count() == 0
def test_err(self):
c = flow.State()
f = tutils.tflow()
- c.add_request(f)
+ c.add_flow(f)
f.error = Error("message")
- assert c.add_error(f)
+ assert c.update_flow(f)
c = flow.State()
f = tutils.tflow()
- c.add_request(f)
+ c.add_flow(f)
c.set_limit("~e")
assert not c.view
f.error = tutils.terr()
- assert c.add_error(f)
+ assert c.update_flow(f)
assert c.view
def test_set_limit(self):
@@ -420,20 +420,20 @@ class TestState:
f = tutils.tflow()
assert len(c.view) == 0
- c.add_request(f)
+ c.add_flow(f)
assert len(c.view) == 1
c.set_limit("~s")
assert c.limit_txt == "~s"
assert len(c.view) == 0
f.response = tutils.tresp()
- c.add_response(f)
+ c.update_flow(f)
assert len(c.view) == 1
c.set_limit(None)
assert len(c.view) == 1
f = tutils.tflow()
- c.add_request(f)
+ c.add_flow(f)
assert len(c.view) == 2
c.set_limit("~q")
assert len(c.view) == 1
@@ -452,18 +452,18 @@ class TestState:
def _add_request(self, state):
f = tutils.tflow()
- state.add_request(f)
+ state.add_flow(f)
return f
def _add_response(self, state):
f = tutils.tflow()
- state.add_request(f)
+ state.add_flow(f)
f.response = tutils.tresp()
- state.add_response(f)
+ state.update_flow(f)
def _add_error(self, state):
f = tutils.tflow(err=True)
- state.add_request(f)
+ state.add_flow(f)
def test_clear(self):
c = flow.State()
@@ -487,7 +487,7 @@ class TestState:
c.clear()
c.load_flows(flows)
- assert isinstance(c._flow_list[0], Flow)
+ assert isinstance(c.flows[0], Flow)
def test_accept_all(self):
c = flow.State()
@@ -532,7 +532,7 @@ class TestSerialize:
s = flow.State()
fm = flow.FlowMaster(None, s)
fm.load_flows(r)
- assert len(s._flow_list) == 6
+ assert len(s.flows) == 6
def test_load_flows_reverse(self):
r = self._treader()
@@ -540,7 +540,7 @@ class TestSerialize:
conf = ProxyConfig(mode="reverse", upstream_server=[True,True,"use-this-domain",80])
fm = flow.FlowMaster(DummyServer(conf), s)
fm.load_flows(r)
- assert s._flow_list[0].request.host == "use-this-domain"
+ assert s.flows[0].request.host == "use-this-domain"
def test_filter(self):
sio = StringIO()
diff --git a/test/test_server.py b/test/test_server.py
index c81eab2b..a611d30f 100644
--- a/test/test_server.py
+++ b/test/test_server.py
@@ -747,19 +747,19 @@ class TestProxyChainingSSLReconnect(tservers.HTTPUpstreamProxTest):
assert req.content == "content"
assert req.status_code == 418
- assert not self.chain[1].tmaster.state._flow_list[0].response # killed
- assert self.chain[1].tmaster.state._flow_list[1].response
+ assert not self.chain[1].tmaster.state.flows[0].response # killed
+ assert self.chain[1].tmaster.state.flows[1].response
- assert self.proxy.tmaster.state._flow_list[0].request.form_in == "authority"
- assert self.proxy.tmaster.state._flow_list[1].request.form_in == "relative"
+ assert self.proxy.tmaster.state.flows[0].request.form_in == "authority"
+ assert self.proxy.tmaster.state.flows[1].request.form_in == "relative"
- assert self.chain[0].tmaster.state._flow_list[0].request.form_in == "authority"
- assert self.chain[0].tmaster.state._flow_list[1].request.form_in == "relative"
- assert self.chain[0].tmaster.state._flow_list[2].request.form_in == "authority"
- assert self.chain[0].tmaster.state._flow_list[3].request.form_in == "relative"
+ assert self.chain[0].tmaster.state.flows[0].request.form_in == "authority"
+ assert self.chain[0].tmaster.state.flows[1].request.form_in == "relative"
+ assert self.chain[0].tmaster.state.flows[2].request.form_in == "authority"
+ assert self.chain[0].tmaster.state.flows[3].request.form_in == "relative"
- assert self.chain[1].tmaster.state._flow_list[0].request.form_in == "relative"
- assert self.chain[1].tmaster.state._flow_list[1].request.form_in == "relative"
+ assert self.chain[1].tmaster.state.flows[0].request.form_in == "relative"
+ assert self.chain[1].tmaster.state.flows[1].request.form_in == "relative"
req = p.request("get:'/p/418:b\"content2\"'")