From 49b34f039132b17f37bb7708aeb754222ea89715 Mon Sep 17 00:00:00 2001
From: Julien Guenat <julien@jguenat.ch>
Date: Fri, 8 Nov 2024 16:50:38 +0100
Subject: [PATCH] (fixup) add test

---
 mis_builder/tests/test_aep.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/mis_builder/tests/test_aep.py b/mis_builder/tests/test_aep.py
index 8b325d715..ddc562895 100644
--- a/mis_builder/tests/test_aep.py
+++ b/mis_builder/tests/test_aep.py
@@ -410,3 +410,28 @@ def test_invalid_field(self):
                 datetime.date(self.prev_year, 12, 1),
             )
         assert "Error while querying move line source" in str(cm.exception)
+
+    def test_aep_branch(self):
+        # create branch
+        self.branch = self.res_company.create(
+            {
+                "name": "AEP Branch",
+                "parent_id": self.company.id,
+            }
+        )
+        # create branch move in March this year
+        branch_move = self._create_move(
+            date=datetime.date(self.curr_year, 3, 1),
+            amount=50,
+            debit_acc=self.account_ar,
+            credit_acc=self.account_in,
+        )
+        branch_move.company_id = self.branch
+        self.aep = AEP(self.company | self.branch)
+        self.aep.parse_expr("balp[]")
+        self.aep.done_parsing()
+        self._do_queries(
+            datetime.date(self.curr_year, 3, 1), datetime.date(self.curr_year, 3, 31)
+        )
+        variation = self._eval_by_account_id("balp[]")
+        self.assertEqual(variation, {self.account_ar.id: 550, self.account_in.id: -550})