From 6bd2d5059b0beeb1aa9996c1d0619f167d98925e Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@corte.si>
Date: Thu, 3 May 2018 08:43:30 +1200
Subject: [PATCH] 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.
---
 mitmproxy/addonmanager.py           |  2 +-
 test/mitmproxy/test_addonmanager.py | 35 ++++++++++++++++-------------
 2 files changed, 21 insertions(+), 16 deletions(-)

diff --git a/mitmproxy/addonmanager.py b/mitmproxy/addonmanager.py
index ac5610738..a9f6551f3 100644
--- a/mitmproxy/addonmanager.py
+++ b/mitmproxy/addonmanager.py
@@ -243,7 +243,7 @@ class AddonManager:
             Invoke an event on an addon and all its children.
         """
         if name not in eventsequence.Events:
-            name = "event_" + name
+            raise exceptions.AddonManagerError("Unknown event: %s" % name)
         for a in traverse([addon]):
             func = getattr(a, name, None)
             if func:
diff --git a/test/mitmproxy/test_addonmanager.py b/test/mitmproxy/test_addonmanager.py
index 02c8f2e23..a1f0a9436 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")