From a8019d79fa3c1598d09f303a3f957490be02a7e3 Mon Sep 17 00:00:00 2001
From: Stefano Borini <sborini@enthought.com>
Date: Fri, 11 Aug 2017 00:04:22 +0100
Subject: [PATCH] tested lookup of notification listener

---
 .../tests/test_factory_registry_plugin.py       | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/force_bdss/tests/test_factory_registry_plugin.py b/force_bdss/tests/test_factory_registry_plugin.py
index 1d001d7..3a90f7b 100644
--- a/force_bdss/tests/test_factory_registry_plugin.py
+++ b/force_bdss/tests/test_factory_registry_plugin.py
@@ -5,6 +5,8 @@ from force_bdss.base_extension_plugin import (
 from force_bdss.ids import factory_id, mco_parameter_id
 from force_bdss.mco.parameters.base_mco_parameter_factory import \
     BaseMCOParameterFactory
+from force_bdss.notification_listeners.i_notification_listener_factory import \
+    INotificationListenerFactory
 
 try:
     import mock
@@ -61,6 +63,10 @@ class MySuperPlugin(BaseExtensionPlugin):
                 mock.Mock(spec=IKPICalculatorFactory,
                           id=factory_id("enthought", "kpi3"))]
 
+    def _notification_listener_factories_default(self):
+        return [mock.Mock(spec=INotificationListenerFactory,
+                          id=factory_id("enthought", "nl1"))]
+
 
 class TestFactoryRegistryWithContent(unittest.TestCase):
     def setUp(self):
@@ -89,6 +95,12 @@ class TestFactoryRegistryWithContent(unittest.TestCase):
             self.assertEqual(self.plugin.kpi_calculator_factory_by_id(id).id,
                              id)
 
+        for entry in ["nl1"]:
+            id = factory_id("enthought", entry)
+            self.assertEqual(
+                self.plugin.notification_listener_factory_by_id(id).id,
+                id)
+
         with self.assertRaises(KeyError):
             self.plugin.mco_factory_by_id(
                 factory_id("enthought", "foo"))
@@ -114,6 +126,11 @@ class TestFactoryRegistryWithContent(unittest.TestCase):
                 factory_id("enthought", "foo")
             )
 
+        with self.assertRaises(KeyError):
+            self.plugin.notification_listener_factory_by_id(
+                factory_id("enthought", "foo")
+            )
+
 
 if __name__ == '__main__':
     unittest.main()
-- 
GitLab