Create probe classes for the notification listeners

force_bdss/tests/probe_classes/factory_registry_plugin.py

@@ -5,6 +5,7 @@ from force_bdss.factory_registry_plugin import FactoryRegistryPlugin
 from .mco import ProbeMCOFactory
 from .kpi_calculator import ProbeKPICalculatorFactory
 from .data_source import ProbeDataSourceFactory
+from .notification_listener import ProbeNotificationListenerFactory
 
 
 class ProbeFactoryRegistryPlugin(FactoryRegistryPlugin):
@@ -21,3 +22,6 @@ class ProbeFactoryRegistryPlugin(FactoryRegistryPlugin):
 
     def _data_source_factories_default(self):
         return [ProbeDataSourceFactory(self)]
+
+    def _notification_listener_factories_default(self):
+        return [ProbeNotificationListenerFactory(self)]

force_bdss/tests/probe_classes/notification_listener.py

+from traits.api import Bool, Str, Type
+
+from force_bdss.api import (
+    BaseNotificationListener, BaseNotificationListenerModel,
+    BaseNotificationListenerFactory)
+
+
+class ProbeNotificationListener(BaseNotificationListener):
+    initialize_called = Bool(False)
+    deliver_called = Bool(False)
+    finalize_called = Bool(False)
+
+    def initialize(self, model):
+        self.initialize_called = True
+
+    def deliver(self, event):
+        self.deliver_called = True
+
+    def finalize(self):
+        self.finalize_called = True
+
+
+class ProbeNotificationListenerModel(BaseNotificationListenerModel):
+    pass
+
+
+class ProbeNotificationListenerFactory(BaseNotificationListenerFactory):
+    id = Str("enthought.test.notification_listener")
+    name = "test_notification_listener"
+
+    model_class = Type(ProbeNotificationListenerModel)
+
+    listener_class = Type(ProbeNotificationListener)
+
+    def create_listener(self):
+        return self.listener_class(self)
+
+    def create_model(self, model_data=None):
+        return self.model_class(self, model_data=model_data)