add except handler tests

tests/test_events.py

@@ -1,3 +1,4 @@
+import pytest
 from pyflowlauncher.event import EventHandler
 
 
@@ -9,6 +10,10 @@ def temp_method2():
     return None
 
 
+def except_method():
+    raise Exception
+
+
 def test_add_method():
     handler = EventHandler()
     handler.add_method(temp_method1)
@@ -25,3 +30,16 @@ def test_call():
     handler = EventHandler()
     handler.add_method(temp_method1)
     assert handler("temp_method1") is None
+
+
+def test_add_exception_handler():
+    handler = EventHandler()
+    handler.add_exception_handler(Exception, temp_method1)
+    assert handler._handlers == {Exception: temp_method1}
+
+
+def test_call_exception():
+    handler = EventHandler()
+    handler.add_method(except_method)
+    with pytest.raises(Exception):
+        handler("except_method")

tests/test_plugin.py

@@ -57,3 +57,14 @@ def test_action():
     plugin = Plugin()
     action = plugin.action(query)
     assert action == {'method': 'query', 'parameters': []}
+
+
+def test_exception_handler():
+    plugin = Plugin()
+
+    @plugin.on_except(KeyError)
+    def action(e: Exception):
+        print('OH NO!')
+        return {'result': [{'title': 'title', 'subtitle': 'subtitle', 'icon': 'icon.png'}]}
+
+    assert plugin._event_handler._handlers == {KeyError: action}