diff --git a/force_bdss/ui_hooks/tests/test_base_ui_hooks_factory.py b/force_bdss/ui_hooks/tests/test_base_ui_hooks_factory.py index 9a364605b8323b292cecd2bee0cfcc7032523583..b50dfda1a24ef39e5afeab86c2b006c402cb2b9a 100644 --- a/force_bdss/ui_hooks/tests/test_base_ui_hooks_factory.py +++ b/force_bdss/ui_hooks/tests/test_base_ui_hooks_factory.py @@ -1,11 +1,16 @@ import unittest -from ..base_ui_hooks_factory import BaseUIHooksFactory try: import mock except ImportError: from unittest import mock +from envisage.api import Plugin +from ..base_ui_hooks_factory import BaseUIHooksFactory + class TestBaseUIHooksFactory(unittest.TestCase): - + def test_initialize(self): + mock_plugin = mock.Mock(spec=Plugin) + factory = BaseUIHooksFactory(plugin=mock_plugin) + self.assertEqual(factory.plugin, mock_plugin)