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