aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@corte.si>2017-03-24 10:15:41 +1300
committerAldo Cortesi <aldo@corte.si>2017-03-25 10:48:12 +1300
commit541c1e8b9f0e2e12bd770100888129639cc94827 (patch)
tree8939e7c9fc7e6810e9363e598e84882098cbde04 /test
parentb531353ee0259425cff2b4857716ba2cd995a437 (diff)
downloadmitmproxy-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.py20
-rw-r--r--test/mitmproxy/addons/test_serverplayback.py12
-rw-r--r--test/mitmproxy/data/addonscripts/addon.py8
-rw-r--r--test/mitmproxy/data/addonscripts/concurrent_decorator_class.py2
-rw-r--r--test/mitmproxy/data/addonscripts/print.py2
-rw-r--r--test/mitmproxy/data/addonscripts/recorder.py2
-rw-r--r--test/mitmproxy/script/test_concurrent.py6
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)