aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@corte.si>2018-05-03 08:43:30 +1200
committerAldo Cortesi <aldo@corte.si>2018-05-03 09:42:51 +1200
commit6bd2d5059b0beeb1aa9996c1d0619f167d98925e (patch)
tree7289e91d1bad45764f4affe810ca61b049f84700 /test
parentf380a77deedbc4c11f11f273384d5fdf14d3f4fe (diff)
downloadmitmproxy-6bd2d5059b0beeb1aa9996c1d0619f167d98925e.tar.gz
mitmproxy-6bd2d5059b0beeb1aa9996c1d0619f167d98925e.tar.bz2
mitmproxy-6bd2d5059b0beeb1aa9996c1d0619f167d98925e.zip
Remove custom events
These were always weird, and were added to support exit when processing completed to mitmdump. We now have better ways to do this.
Diffstat (limited to 'test')
-rw-r--r--test/mitmproxy/test_addonmanager.py35
1 files changed, 20 insertions, 15 deletions
diff --git a/test/mitmproxy/test_addonmanager.py b/test/mitmproxy/test_addonmanager.py
index 02c8f2e2..a1f0a943 100644
--- a/test/mitmproxy/test_addonmanager.py
+++ b/test/mitmproxy/test_addonmanager.py
@@ -16,7 +16,7 @@ class TAddon:
def __init__(self, name, addons=None):
self.name = name
self.response = True
- self.custom_called = False
+ self.running_called = False
if addons:
self.addons = addons
@@ -30,12 +30,12 @@ class TAddon:
def done(self):
pass
- def event_custom(self):
- self.custom_called = True
+ def running(self):
+ self.running_called = True
class THalt:
- def event_custom(self):
+ def running(self):
raise exceptions.AddonHalt
@@ -59,12 +59,13 @@ def test_halt():
a.add(halt)
a.add(end)
- a.trigger("custom")
- assert not end.custom_called
+ assert not end.running_called
+ a.trigger("running")
+ assert not end.running_called
a.remove(halt)
- a.trigger("custom")
- assert end.custom_called
+ a.trigger("running")
+ assert end.running_called
@pytest.mark.asyncio
@@ -126,6 +127,10 @@ async def test_simple():
assert not a.chain
a.add(TAddon("one"))
+
+ a.trigger("nonexistent")
+ assert await tctx.master.await_log("unknown event")
+
a.trigger("running")
a.trigger("response")
assert await tctx.master.await_log("not callable")
@@ -140,8 +145,8 @@ async def test_simple():
ta = TAddon("one")
a.add(ta)
- a.trigger("custom")
- assert ta.custom_called
+ a.trigger("running")
+ assert ta.running_called
assert ta in a
@@ -174,11 +179,11 @@ def test_nesting():
assert a.get("three")
assert a.get("four")
- a.trigger("custom")
- assert a.get("one").custom_called
- assert a.get("two").custom_called
- assert a.get("three").custom_called
- assert a.get("four").custom_called
+ a.trigger("running")
+ assert a.get("one").running_called
+ assert a.get("two").running_called
+ assert a.get("three").running_called
+ assert a.get("four").running_called
a.remove(a.get("three"))
assert not a.get("three")