mirror of
https://gitlab.com/flectra-community/mis-builder.git
synced 2024-11-16 19:22:04 +00:00
69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
|
# Copyright 2023 ACSONE SA/NV (<http://acsone.eu>)
|
||
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
|
||
|
|
|
||
|
|
from flectra.tests.common import TransactionCase
|
||
|
|
|
||
|
|
from flectra.addons.mis_builder_budget.models.mis_report_instance import (
|
||
|
|
MisBudgetAwareExpressionEvaluator,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class TestBudgetAwareExpressionEvaluator(TransactionCase):
|
||
|
|
def _make_evaluator(self, kpi_data=None):
|
||
|
|
return MisBudgetAwareExpressionEvaluator(
|
||
|
|
date_from="2017-01-01",
|
||
|
|
date_to="2017-01-16",
|
||
|
|
kpi_data=kpi_data or {},
|
||
|
|
additional_move_line_filter=[],
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_no_expression(self):
|
||
|
|
evaluator = self._make_evaluator()
|
||
|
|
evaluator.eval_expressions([None], locals_dict={})
|
||
|
|
|
||
|
|
def test_one_expression(self):
|
||
|
|
evaluator = self._make_evaluator()
|
||
|
|
kpi = self.env["mis.report.kpi"].new({"name": "thekpi"})
|
||
|
|
expr = self.env["mis.report.kpi.expression"].new(
|
||
|
|
{"kpi_id": kpi.id, "name": "a"}
|
||
|
|
)
|
||
|
|
vals, drilldown_args, name_error = evaluator.eval_expressions(
|
||
|
|
[expr], locals_dict={"a": 1}
|
||
|
|
)
|
||
|
|
self.assertEqual(vals, [1])
|
||
|
|
self.assertEqual(drilldown_args, [None])
|
||
|
|
self.assertFalse(name_error)
|
||
|
|
|
||
|
|
def test_two_expressions(self):
|
||
|
|
evaluator = self._make_evaluator()
|
||
|
|
kpi = self.env["mis.report.kpi"].new({"name": "thekpi"})
|
||
|
|
expr1 = self.env["mis.report.kpi.expression"].new(
|
||
|
|
{"kpi_id": kpi.id, "name": "a"}
|
||
|
|
)
|
||
|
|
expr2 = self.env["mis.report.kpi.expression"].new(
|
||
|
|
{"kpi_id": kpi.id, "name": "b"}
|
||
|
|
)
|
||
|
|
vals, drilldown_args, _ = evaluator.eval_expressions(
|
||
|
|
[expr1, expr2], locals_dict={"a": 1, "b": 2}
|
||
|
|
)
|
||
|
|
self.assertEqual(vals, [1, 2])
|
||
|
|
self.assertEqual(drilldown_args, [None, None])
|
||
|
|
|
||
|
|
def test_two_expressions_with_budget(self):
|
||
|
|
kpi = self.env["mis.report.kpi"].new({"name": "thekpi", "budgetable": True})
|
||
|
|
expr1 = self.env["mis.report.kpi.expression"].new(
|
||
|
|
{"kpi_id": kpi.id, "name": "a"}
|
||
|
|
)
|
||
|
|
expr2 = self.env["mis.report.kpi.expression"].new(
|
||
|
|
{"kpi_id": kpi.id, "name": "b"}
|
||
|
|
)
|
||
|
|
kpi_data = {
|
||
|
|
expr1: 10,
|
||
|
|
expr2: 20,
|
||
|
|
}
|
||
|
|
evaluator = self._make_evaluator(kpi_data)
|
||
|
|
vals, _, _ = evaluator.eval_expressions(
|
||
|
|
[expr1, expr2], locals_dict={"a": 1, "b": 2}
|
||
|
|
)
|
||
|
|
self.assertEqual(vals, [10, 20]) # budget values instead of locals_dict values
|