Added more coverage

force_bdss/core_plugins/dummy/dummy_kpi_calculator/tests/test_dummy_kpi_calculator_bundle.py

+import unittest
+
+from force_bdss.core_plugins.dummy.dummy_kpi_calculator.dummy_kpi_calculator\
+    import \
+    DummyKPICalculator
+from force_bdss.core_plugins.dummy.dummy_kpi_calculator\
+    .dummy_kpi_calculator_bundle import \
+    DummyKPICalculatorBundle
+from force_bdss.core_plugins.dummy.dummy_kpi_calculator\
+    .dummy_kpi_calculator_model import \
+    DummyKPICalculatorModel
+from force_bdss.core_plugins.dummy.tests.kpi_calculator_bundle_test_mixin \
+    import \
+    KPICalculatorBundleTestMixin
+
+
+class TestDummyKPICalculatorBundle(
+        KPICalculatorBundleTestMixin, unittest.TestCase):
+
+    @property
+    def bundle_class(self):
+        return DummyKPICalculatorBundle
+
+    @property
+    def kpi_calculator_class(self):
+        return DummyKPICalculator
+
+    @property
+    def model_class(self):
+        return DummyKPICalculatorModel

force_bdss/core_plugins/dummy/kpi_adder/tests/test_kpi_adder_bundle.py

+import unittest
+
+from force_bdss.core_plugins.dummy.kpi_adder.kpi_adder_bundle import \
+    KPIAdderBundle
+from force_bdss.core_plugins.dummy.kpi_adder.kpi_adder_calculator import \
+    KPIAdderCalculator
+from force_bdss.core_plugins.dummy.kpi_adder.kpi_adder_model import \
+    KPIAdderModel
+from force_bdss.core_plugins.dummy.tests.kpi_calculator_bundle_test_mixin \
+    import \
+    KPICalculatorBundleTestMixin
+
+
+class TestDummyKPICalculatorBundle(
+        KPICalculatorBundleTestMixin, unittest.TestCase):
+
+    @property
+    def bundle_class(self):
+        return KPIAdderBundle
+
+    @property
+    def kpi_calculator_class(self):
+        return KPIAdderCalculator
+
+    @property
+    def model_class(self):
+        return KPIAdderModel

force_bdss/core_plugins/dummy/tests/kpi_calculator_bundle_test_mixin.py

+try:
+    import mock
+except ImportError:
+    from unittest import mock
+
+from envisage.api import Plugin
+
+
+class KPICalculatorBundleTestMixin(object):
+    def setUp(self):
+        self.plugin = mock.Mock(spec=Plugin)
+        super(KPICalculatorBundleTestMixin, self).setUp()
+
+    @property
+    def bundle_class(self):
+        raise NotImplementedError()
+
+    @property
+    def model_class(self):
+        raise NotImplementedError()
+
+    @property
+    def kpi_calculator_class(self):
+        raise NotImplementedError()
+
+    def test_initialization(self):
+        bundle = self.bundle_class(self.plugin)
+        self.assertNotEqual(bundle.id, "")
+        self.assertEqual(bundle.plugin, self.plugin)
+
+    def test_create_model(self):
+        bundle = self.bundle_class(self.plugin)
+        model = bundle.create_model({})
+        self.assertIsInstance(model, self.model_class)
+
+        model = bundle.create_model()
+        self.assertIsInstance(model, self.model_class)
+
+    def test_create_kpi_calculator(self):
+        bundle = self.bundle_class(self.plugin)
+        ds = bundle.create_kpi_calculator()
+        self.assertIsInstance(ds, self.kpi_calculator_class)