diff options
author | Maximilian Hils <git@maximilianhils.com> | 2014-11-26 04:18:21 +0100 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2014-11-26 04:18:21 +0100 |
commit | ed8249023fb7c0d429b9278c63b51ac071700987 (patch) | |
tree | 8bd83354e7484f15029ec678744234bd32e9906d /test | |
parent | 47a78e3c729f4ddb7971b72bfae30140562f4dd6 (diff) | |
download | mitmproxy-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.py | 8 | ||||
-rw-r--r-- | test/test_flow.py | 40 | ||||
-rw-r--r-- | test/test_server.py | 20 |
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\"'") |