Add the test module so that it can be version tracked.

deluge/plugins/notifications/notifications/test.py

@@ -0,0 +1,86 @@
+# -*- coding: utf-8 -*-
+# vim: sw=4 ts=4 fenc=utf-8 et
+# ==============================================================================
+# Copyright © 2009 UfSoft.org - Pedro Algarvio <ufs@ufsoft.org>
+#
+# License: BSD - Please view the LICENSE file for additional information.
+# ==============================================================================
+
+from twisted.internet import task
+from deluge import component
+from deluge.event import DelugeEvent
+from deluge.log import LOG as log
+
+class FooEvent(DelugeEvent):
+    """foo Event"""
+
+class CustomEvent(DelugeEvent):
+    """Just a custom event to test"""
+
+class TestEmailNotifications(component.Component):
+    def __init__(self, imp):
+        component.Component.__init__(self, self.__class__.__name__, 5)
+        self.__imp = imp
+        self.lc = task.LoopingCall(self.update)
+        self.n = 1
+        self.events = [
+            FooEvent(),
+            CustomEvent()
+        ]
+        self.events_classes = []
+    def enable(self):
+        log.debug("\n\nEnabling %s", self.__class__.__name__)
+        for event in self.events:
+            if self.__imp == 'core':
+                component.get("Notifications").register_custom_email_notification(
+                    event.__class__.__name__,
+                    self.custom_email_message_provider
+                )
+            elif self.__imp == 'gtk':
+                component.get("Notifications").register_custom_popup_notification(
+                    event.__class__.__name__,
+                    self.custom_popup_message_provider
+                )
+                component.get("Notifications").register_custom_blink_notification(
+                    event.__class__.__name__,
+                    self.custom_blink_message_provider
+                )
+                component.get("Notifications").register_custom_sound_notification(
+                    event.__class__.__name__,
+                    self.custom_sound_message_provider
+                )
+
+        self.lc.start(15, False)
+
+    def disable(self):
+        log.debug("\n\nDisabling %s", self.__class__.__name__)
+        self.lc.stop()
+
+    def update(self):
+        log.debug("\n\nUpdating %s", self.__class__.__name__)
+        self.events.append(self.events.pop(0)) # Re-Queue
+        self.n += 1
+        try:
+            component.get("EventManager").emit(self.events[0])
+        except KeyError:
+            pass
+
+    def custom_email_message_provider(self, *evt_args, **evt_kwargs):
+        log.debug("Running custom email message provider: %s %s", evt_args, evt_kwargs)
+        subject = "%s Email Subject: %s" % (self.events[0].__class__.__name__, self.n)
+        message = "%s Email Message: %s" % (self.events[0].__class__.__name__, self.n)
+        return subject, message
+
+    def custom_popup_message_provider(self, *evt_args, **evt_kwargs):
+        log.debug("Running custom popup message provider: %s %s", evt_args, evt_kwargs)
+        title = "%s Popup Title: %s" % (self.events[0].__class__.__name__, self.n)
+        message = "%s Popup Message: %s" % (self.events[0].__class__.__name__, self.n)
+        return title, message
+
+    def custom_blink_message_provider(self, *evt_args, **evt_kwargs):
+        log.debug("Running custom blink message provider: %s %s", evt_args, evt_kwargs)
+        return True
+
+    def custom_sound_message_provider(self, *evt_args, **evt_kwargs):
+        log.debug("Running custom sound message provider: %s %s", evt_args, evt_kwargs)
+        return '/usr/share/kde4/apps/korganizer/sounds/alert.wav'