reporting-engine/kpi/tests/test_kpi.py

# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from flectra.tests.common import TransactionCase


class TestKPI(TransactionCase):
    @classmethod
    def setUpClass(cls):
        super().setUpClass()

    def test_invalid_threshold_range(self):
        range1 = self.env["kpi.threshold.range"].create(
            {
                "name": "Range1",
                "min_type": "static",
                "max_type": "static",
                "min_fixed_value": 3,
                "max_fixed_value": 1,
            }
        )
        range2 = self.env["kpi.threshold.range"].create(
            {
                "name": "Range2",
                "min_type": "static",
                "max_type": "static",
                "min_fixed_value": 4,
                "max_fixed_value": 10,
            }
        )
        self.assertFalse(range1.valid)
        self.assertTrue(range2.valid)

    def test_invalid_threshold(self):
        range1 = self.env["kpi.threshold.range"].create(
            {
                "name": "Range1",
                "min_type": "static",
                "max_type": "static",
                "min_fixed_value": 1,
                "max_fixed_value": 4,
            }
        )
        range2 = self.env["kpi.threshold.range"].create(
            {
                "name": "Range2",
                "min_type": "static",
                "max_type": "static",
                "min_fixed_value": 4,
                "max_fixed_value": 10,
            }
        )
        range3 = self.env["kpi.threshold.range"].create(
            {
                "name": "Range3",
                "min_type": "static",
                "max_type": "static",
                "min_fixed_value": 1,
                "max_fixed_value": 3,
            }
        )
        range_invalid = self.env["kpi.threshold.range"].create(
            {
                "name": "RangeInvalid",
                "min_type": "static",
                "max_type": "static",
                "min_fixed_value": 3,
                "max_fixed_value": 1,
            }
        )

        threshold1 = self.env["kpi.threshold"].create(
            {"name": "Threshold1", "range_ids": [(6, 0, [range1.id, range2.id])]}
        )

        threshold2 = self.env["kpi.threshold"].create(
            {"name": "Threshold1", "range_ids": [(6, 0, [range3.id, range2.id])]}
        )

        threshold3 = self.env["kpi.threshold"].create(
            {
                "name": "Threshold1",
                "range_ids": [(6, 0, [range_invalid.id, range2.id])],
            }
        )

        self.assertFalse(threshold1.valid)
        self.assertTrue(threshold2.valid)
        self.assertFalse(threshold3.valid)

    def test_invalid_threshold_range_exception(self):
        range_error = self.env["kpi.threshold.range"].create(
            {
                "name": "RangeError",
                "min_type": "python",
                "min_code": "<Not a valid python expression>",
                "max_type": "static",
                "max_fixed_value": 1,
            }
        )
        self.assertFalse(range_error.valid)

    def test_kpi_python(self):
        kpi_category = self.env["kpi.category"].create({"name": "Dynamic KPIs"})
        kpi_threshold = self.env["kpi.threshold"].create(
            {"name": "KPI Threshold for dynamic KPIs"}
        )
        kpi_code = """
            {
                'value': 1.0,
                'color': '#00FF00'
            }
        """
        kpi = self.env["kpi"].create(
            {
                "name": "Dynamic python kpi",
                "description": "Dynamic python kpi",
                "category_id": kpi_category.id,
                "threshold_id": kpi_threshold.id,
                "periodicity": 1,
                "periodicity_uom": "day",
                "kpi_type": "python",
                "kpi_code": kpi_code,
            }
        )
        kpi.update_kpi_value()
        kpi_history = self.env["kpi.history"].search([("kpi_id", "=", kpi.id)])
        self.assertEqual(len(kpi_history), 1)
        self.assertEqual(kpi_history.color, "#00FF00")
        self.assertEqual(kpi_history.value, 1.0)