From 7c32d4ea2a435484e83aa459831f74ca483d8e3c Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Wed, 19 Oct 2016 14:48:42 +1300 Subject: flow.state -> addons.state --- test/mitmproxy/addons/test_state.py | 20 ++++++++++++++++++++ test/mitmproxy/test_flow.py | 27 ++++++++++++++------------- test/mitmproxy/test_flow_state.py | 19 ------------------- test/mitmproxy/tservers.py | 2 +- 4 files changed, 35 insertions(+), 33 deletions(-) create mode 100644 test/mitmproxy/addons/test_state.py delete mode 100644 test/mitmproxy/test_flow_state.py (limited to 'test') diff --git a/test/mitmproxy/addons/test_state.py b/test/mitmproxy/addons/test_state.py new file mode 100644 index 00000000..71c46dcb --- /dev/null +++ b/test/mitmproxy/addons/test_state.py @@ -0,0 +1,20 @@ +from mitmproxy import proxy +from mitmproxy import master +from mitmproxy.addons import state + +from .. import tutils + + +class TestState: + def test_duplicate_flow(self): + s = state.State() + fm = master.Master(None, proxy.DummyServer()) + fm.addons.add(s) + f = tutils.tflow(resp=True) + fm.load_flow(f) + assert s.flow_count() == 1 + + f2 = s.duplicate_flow(f) + assert f2.response + assert s.flow_count() == 2 + assert s.index(f2) == 1 diff --git a/test/mitmproxy/test_flow.py b/test/mitmproxy/test_flow.py index 2b387f5c..86cd7d16 100644 --- a/test/mitmproxy/test_flow.py +++ b/test/mitmproxy/test_flow.py @@ -4,6 +4,7 @@ import io import netlib.utils from netlib.http import Headers from mitmproxy import flowfilter, flow, options +from mitmproxy.addons import state from mitmproxy.contrib import tnetstring from mitmproxy.exceptions import FlowReadException, Kill from mitmproxy.models import Error @@ -110,7 +111,7 @@ class TestHTTPFlow: def test_killall(self): srv = DummyServer(None) - s = flow.State() + s = state.State() fm = master.Master(None, srv) fm.addons.add(s) @@ -190,7 +191,7 @@ class TestTCPFlow: class TestState: def test_backup(self): - c = flow.State() + c = state.State() f = tutils.tflow() c.add_flow(f) f.backup() @@ -202,7 +203,7 @@ class TestState: connect -> request -> response """ - c = flow.State() + c = state.State() f = tutils.tflow() c.add_flow(f) assert f @@ -226,13 +227,13 @@ class TestState: assert c.active_flow_count() == 0 def test_err(self): - c = flow.State() + c = state.State() f = tutils.tflow() c.add_flow(f) f.error = Error("message") assert c.update_flow(f) - c = flow.State() + c = state.State() f = tutils.tflow() c.add_flow(f) c.set_view_filter("~e") @@ -242,7 +243,7 @@ class TestState: assert c.view def test_set_view_filter(self): - c = flow.State() + c = state.State() f = tutils.tflow() assert len(c.view) == 0 @@ -270,7 +271,7 @@ class TestState: assert "Invalid" in c.set_view_filter("~") def test_set_intercept(self): - c = flow.State() + c = state.State() assert not c.set_intercept("~q") assert c.intercept_txt == "~q" assert "Invalid" in c.set_intercept("~") @@ -293,7 +294,7 @@ class TestState: state.add_flow(f) def test_clear(self): - c = flow.State() + c = state.State() f = self._add_request(c) f.intercepted = True @@ -301,7 +302,7 @@ class TestState: assert c.flow_count() == 0 def test_dump_flows(self): - c = flow.State() + c = state.State() self._add_request(c) self._add_response(c) self._add_request(c) @@ -317,7 +318,7 @@ class TestState: assert isinstance(c.flows[0], Flow) def test_accept_all(self): - c = flow.State() + c = state.State() self._add_request(c) self._add_response(c) self._add_request(c) @@ -363,7 +364,7 @@ class TestSerialize: def test_load_flows(self): r = self._treader() - s = flow.State() + s = state.State() fm = master.Master(None, DummyServer()) fm.addons.add(s) fm.load_flows(r) @@ -371,7 +372,7 @@ class TestSerialize: def test_load_flows_reverse(self): r = self._treader() - s = flow.State() + s = state.State() opts = options.Options( mode="reverse", upstream_server="https://use-this-domain" @@ -440,7 +441,7 @@ class TestFlowMaster: assert fm.create_request("GET", "http", "example.com", 80, "/") def test_all(self): - s = flow.State() + s = state.State() fm = master.Master(None, DummyServer()) fm.addons.add(s) f = tutils.tflow(req=None) diff --git a/test/mitmproxy/test_flow_state.py b/test/mitmproxy/test_flow_state.py deleted file mode 100644 index 05f4cbb4..00000000 --- a/test/mitmproxy/test_flow_state.py +++ /dev/null @@ -1,19 +0,0 @@ -from mitmproxy import flow -from mitmproxy import proxy -from mitmproxy import master -from . import tutils - - -class TestState: - def test_duplicate_flow(self): - s = flow.State() - fm = master.Master(None, proxy.DummyServer()) - fm.addons.add(s) - f = tutils.tflow(resp=True) - fm.load_flow(f) - assert s.flow_count() == 1 - - f2 = s.duplicate_flow(f) - assert f2.response - assert s.flow_count() == 2 - assert s.index(f2) == 1 diff --git a/test/mitmproxy/tservers.py b/test/mitmproxy/tservers.py index 254af2f0..1243bca0 100644 --- a/test/mitmproxy/tservers.py +++ b/test/mitmproxy/tservers.py @@ -7,7 +7,7 @@ import sys from mitmproxy.proxy.config import ProxyConfig from mitmproxy.proxy.server import ProxyServer from mitmproxy import master -from mitmproxy.flow import state +from mitmproxy.addons import state import pathod.test import pathod.pathoc from mitmproxy import controller, options -- cgit v1.2.3