diff options
author | Aldo Cortesi <aldo@corte.si> | 2017-03-24 10:15:41 +1300 |
---|---|---|
committer | Aldo Cortesi <aldo@corte.si> | 2017-03-25 10:48:12 +1300 |
commit | 541c1e8b9f0e2e12bd770100888129639cc94827 (patch) | |
tree | 8939e7c9fc7e6810e9363e598e84882098cbde04 /test | |
parent | b531353ee0259425cff2b4857716ba2cd995a437 (diff) | |
download | mitmproxy-541c1e8b9f0e2e12bd770100888129639cc94827.tar.gz mitmproxy-541c1e8b9f0e2e12bd770100888129639cc94827.tar.bz2 mitmproxy-541c1e8b9f0e2e12bd770100888129639cc94827.zip |
addons: start -> load throughout
Diffstat (limited to 'test')
-rw-r--r-- | test/mitmproxy/addons/test_script.py | 20 | ||||
-rw-r--r-- | test/mitmproxy/addons/test_serverplayback.py | 12 | ||||
-rw-r--r-- | test/mitmproxy/data/addonscripts/addon.py | 8 | ||||
-rw-r--r-- | test/mitmproxy/data/addonscripts/concurrent_decorator_class.py | 2 | ||||
-rw-r--r-- | test/mitmproxy/data/addonscripts/print.py | 2 | ||||
-rw-r--r-- | test/mitmproxy/data/addonscripts/recorder.py | 2 | ||||
-rw-r--r-- | test/mitmproxy/script/test_concurrent.py | 6 |
7 files changed, 26 insertions, 26 deletions
diff --git a/test/mitmproxy/addons/test_script.py b/test/mitmproxy/addons/test_script.py index 84d36e2a..79416161 100644 --- a/test/mitmproxy/addons/test_script.py +++ b/test/mitmproxy/addons/test_script.py @@ -104,7 +104,7 @@ def test_load_script(): "mitmproxy/data/addonscripts/recorder.py" ), [] ) - assert ns.start + assert ns.load def test_script_print_stdout(): @@ -129,7 +129,7 @@ class TestScript: ) ) sc.load_script() - assert sc.ns.call_log[0][0:2] == ("solo", "start") + assert sc.ns.call_log[0][0:2] == ("solo", "load") sc.ns.call_log = [] f = tflow.tflow(resp=True) @@ -157,7 +157,7 @@ class TestScript: sc = script.Script( tutils.test_data.path("mitmproxy/data/addonscripts/error.py") ) - sc.start(tctx.options) + sc.load(tctx.options) f = tflow.tflow(resp=True) sc.request(f) assert tctx.master.logs[0].level == "error" @@ -173,10 +173,10 @@ class TestScript: "mitmproxy/data/addonscripts/addon.py" ) ) - sc.start(tctx.options) + sc.load(tctx.options) tctx.configure(sc) assert sc.ns.event_log == [ - 'scriptstart', 'addonstart', 'addonconfigure' + 'scriptload', 'addonload', 'addonconfigure' ] @@ -213,7 +213,7 @@ class TestScriptLoader: ), [f] ) evts = [i[1] for i in sc.ns.call_log] - assert evts == ['start', 'requestheaders', 'request', 'responseheaders', 'response', 'done'] + assert evts == ['load', 'requestheaders', 'request', 'responseheaders', 'response', 'done'] f = tflow.tflow(resp=True) with m.handlecontext(): @@ -271,15 +271,15 @@ class TestScriptLoader: ) debug = [i.msg for i in tctx.master.logs if i.level == "debug"] assert debug == [ - 'a start', + 'a load', 'a configure', 'a running', - 'b start', + 'b load', 'b configure', 'b running', - 'c start', + 'c load', 'c configure', 'c running', ] @@ -307,7 +307,7 @@ class TestScriptLoader: assert debug == [ 'c done', 'b done', - 'x start', + 'x load', 'x configure', 'x running', ] diff --git a/test/mitmproxy/addons/test_serverplayback.py b/test/mitmproxy/addons/test_serverplayback.py index 02642c35..7078b66e 100644 --- a/test/mitmproxy/addons/test_serverplayback.py +++ b/test/mitmproxy/addons/test_serverplayback.py @@ -44,12 +44,12 @@ def test_server_playback(): assert not sp.flowmap - sp.load([f]) + sp.load_flows([f]) assert sp.flowmap assert sp.next_flow(f) assert not sp.flowmap - sp.load([f]) + sp.load_flows([f]) assert sp.flowmap sp.clear() assert not sp.flowmap @@ -192,7 +192,7 @@ def test_load(): r2 = tflow.tflow(resp=True) r2.request.headers["key"] = "two" - s.load([r, r2]) + s.load_flows([r, r2]) assert s.count() == 2 @@ -218,7 +218,7 @@ def test_load_with_server_replay_nopop(): r2 = tflow.tflow(resp=True) r2.request.headers["key"] = "two" - s.load([r, r2]) + s.load_flows([r, r2]) assert s.count() == 2 s.next_flow(r) @@ -319,7 +319,7 @@ def test_server_playback_full(): f = tflow.tflow() f.response = mitmproxy.test.tutils.tresp(content=f.request.content) - s.load([f, f]) + s.load_flows([f, f]) tf = tflow.tflow() assert not tf.response @@ -352,7 +352,7 @@ def test_server_playback_kill(): f = tflow.tflow() f.response = mitmproxy.test.tutils.tresp(content=f.request.content) - s.load([f]) + s.load_flows([f]) f = tflow.tflow() f.request.host = "nonexistent" diff --git a/test/mitmproxy/data/addonscripts/addon.py b/test/mitmproxy/data/addonscripts/addon.py index f34f41cb..3186400a 100644 --- a/test/mitmproxy/data/addonscripts/addon.py +++ b/test/mitmproxy/data/addonscripts/addon.py @@ -6,8 +6,8 @@ class Addon: def event_log(self): return event_log - def start(self, opts): - event_log.append("addonstart") + def load(self, opts): + event_log.append("addonload") def configure(self, options, updated): event_log.append("addonconfigure") @@ -17,6 +17,6 @@ def configure(options, updated): event_log.append("addonconfigure") -def start(opts): - event_log.append("scriptstart") +def load(opts): + event_log.append("scriptload") return Addon() diff --git a/test/mitmproxy/data/addonscripts/concurrent_decorator_class.py b/test/mitmproxy/data/addonscripts/concurrent_decorator_class.py index 10ba24cd..d486923a 100644 --- a/test/mitmproxy/data/addonscripts/concurrent_decorator_class.py +++ b/test/mitmproxy/data/addonscripts/concurrent_decorator_class.py @@ -9,5 +9,5 @@ class ConcurrentClass: time.sleep(0.1) -def start(opts): +def load(opts): return ConcurrentClass() diff --git a/test/mitmproxy/data/addonscripts/print.py b/test/mitmproxy/data/addonscripts/print.py index fdfcc3d9..93b65a64 100644 --- a/test/mitmproxy/data/addonscripts/print.py +++ b/test/mitmproxy/data/addonscripts/print.py @@ -1,2 +1,2 @@ -def start(opts): +def load(l): print("stdoutprint") diff --git a/test/mitmproxy/data/addonscripts/recorder.py b/test/mitmproxy/data/addonscripts/recorder.py index aff524a8..5e0cdfc2 100644 --- a/test/mitmproxy/data/addonscripts/recorder.py +++ b/test/mitmproxy/data/addonscripts/recorder.py @@ -22,5 +22,5 @@ class CallLogger: raise AttributeError -def start(opts): +def load(opts): return CallLogger(*sys.argv[1:]) diff --git a/test/mitmproxy/script/test_concurrent.py b/test/mitmproxy/script/test_concurrent.py index 0e397b8f..ccae0d31 100644 --- a/test/mitmproxy/script/test_concurrent.py +++ b/test/mitmproxy/script/test_concurrent.py @@ -24,7 +24,7 @@ class TestConcurrent(tservers.MasterTest): "mitmproxy/data/addonscripts/concurrent_decorator.py" ) ) - sc.start(tctx.options) + sc.load(tctx.options) f1, f2 = tflow.tflow(), tflow.tflow() tctx.cycle(sc, f1) @@ -42,7 +42,7 @@ class TestConcurrent(tservers.MasterTest): "mitmproxy/data/addonscripts/concurrent_decorator_err.py" ) ) - sc.start(tctx.options) + sc.load(tctx.options) assert tctx.master.has_log("decorator not supported") def test_concurrent_class(self): @@ -52,7 +52,7 @@ class TestConcurrent(tservers.MasterTest): "mitmproxy/data/addonscripts/concurrent_decorator_class.py" ) ) - sc.start(tctx.options) + sc.load(tctx.options) f1, f2 = tflow.tflow(), tflow.tflow() tctx.cycle(sc, f1) |