Moved DummyParameterFactory to the dummy module

force_bdss/mco/parameters/tests/test_base_mco_parameter_factory.py

@@ -4,35 +4,13 @@ from envisage.plugin import Plugin
 from traits.trait_errors import TraitError
 
 from force_bdss.mco.base_mco_factory import BaseMCOFactory
+from force_bdss.tests.dummy_classes.mco import DummyMCOParameterFactory, \
+    DummyMCOParameter
 
 try:
     import mock
 except ImportError:
     from unittest import mock
-from traits.api import Int
-
-
-from force_bdss.mco.parameters.base_mco_parameter import BaseMCOParameter
-from force_bdss.mco.parameters.base_mco_parameter_factory import \
-    BaseMCOParameterFactory
-
-
-class DummyMCOParameter(BaseMCOParameter):
-    x = Int()
-
-
-class DummyMCOParameterFactory(BaseMCOParameterFactory):
-    def get_identifier(self):
-        return "foo"
-
-    def get_name(self):
-        return "bar"
-
-    def get_description(self):
-        return "description"
-
-    def get_model_class(self):
-        return DummyMCOParameter
 
 
 class TestBaseMCOParameterFactory(unittest.TestCase):
@@ -48,8 +26,8 @@ class TestBaseMCOParameterFactory(unittest.TestCase):
 
     def test_initialization(self):
         factory = DummyMCOParameterFactory(mco_factory=self.mco_factory)
-        self.assertEqual(factory.id, "mcoid.parameter.foo")
-        self.assertEqual(factory.name, "bar")
+        self.assertEqual(factory.id, "mcoid.parameter.dummy_mco_parameter")
+        self.assertEqual(factory.name, "Dummy MCO parameter")
         self.assertEqual(factory.description, "description")
         self.assertEqual(factory.model_class, DummyMCOParameter)
         self.assertIsInstance(factory.create_model(), DummyMCOParameter)

force_bdss/tests/dummy_classes/mco.py

+from traits.api import Int
 from force_bdss.mco.base_mco import BaseMCO
 from force_bdss.mco.base_mco_communicator import BaseMCOCommunicator
 from force_bdss.mco.base_mco_factory import BaseMCOFactory
 from force_bdss.mco.base_mco_model import BaseMCOModel
+from force_bdss.mco.parameters.base_mco_parameter import BaseMCOParameter
+from force_bdss.mco.parameters.base_mco_parameter_factory import \
+    BaseMCOParameterFactory
 
 
 class DummyMCO(BaseMCO):
@@ -21,6 +25,24 @@ class DummyMCOModel(BaseMCOModel):
     pass
 
 
+class DummyMCOParameter(BaseMCOParameter):
+    x = Int()
+
+
+class DummyMCOParameterFactory(BaseMCOParameterFactory):
+    def get_identifier(self):
+        return "dummy_mco_parameter"
+
+    def get_name(self):
+        return "Dummy MCO parameter"
+
+    def get_description(self):
+        return "description"
+
+    def get_model_class(self):
+        return DummyMCOParameter
+
+
 class DummyMCOFactory(BaseMCOFactory):
     def get_identifier(self):
         return "dummy_mco"
@@ -38,4 +60,4 @@ class DummyMCOFactory(BaseMCOFactory):
         return DummyMCO
 
     def parameter_factories(self):
-        return []
+        return [DummyMCOParameterFactory(mco_factory=self)]