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\"'") | 
