🐛 treatment container history: fix history on container move (refs #6532)

1 files changed, 63 insertions, 7 deletions

diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py
index 20e0e20f0..c8e16a50c 100644
--- a/archaeological_warehouse/tests.py
+++ b/archaeological_warehouse/tests.py
@@ -628,9 +628,12 @@ class ContainerQATest(FindInit, TestCase):
 
     def test_move(self):
         c = Client()
-        find_0 = self.finds[0]
+        find_0 = Find.objects.get(pk=self.finds[0].pk)
         find_0.container = self.container_3
+        find_0.container_ref = self.container_3
         find_0.save()
+        find = Find.objects.get(pk=find_0.pk)
+        self.assertTrue(find.container_fisrt_full_location)
 
         pks_1 = f"{self.container_3.pk}-{self.container_4.pk}"
         pks_2 = str(self.container_1.pk)
@@ -653,18 +656,33 @@ class ContainerQATest(FindInit, TestCase):
 
         packaging = TreatmentType.objects.get(txt_idx="packaging")
 
-        def check_location_of_children():
+        def check_location_of_children(idx):
             # check change location of children
             container3 = models.Container.objects.get(pk=self.container_3.pk)
             self.assertEqual(container3.location, self.alt_warehouse)
             container4 = models.Container.objects.get(pk=self.container_4.pk)
             self.assertEqual(container4.location, self.alt_warehouse)
 
-        def check_reinit_parent():
+        def check_reinit_parent(idx):
             # reinit parent when not provided and location changed
             container = models.Container.objects.get(pk=self.container_3.pk)
             self.assertEqual(container.parent, None)
 
+        def init_treatment(find_0):
+            # create a first packaging treatment
+            treat_state = TreatmentState.get_completed_state()
+            t = Treatment.objects.create(
+                container=self.container_3,
+                year=2024,
+                start_date=datetime.date(2024, 2, 29),
+                location=self.main_warehouse,
+                treatment_state=treat_state
+            )
+            treat_type = TreatmentType.objects.get(txt_idx="packaging")
+            t.treatment_types.add(treat_type)
+            t.save()
+            t.finds.add(find_0)
+            t.save()  # force find container history
 
         data_check_lst = [
             (
@@ -674,6 +692,7 @@ class ContainerQATest(FindInit, TestCase):
                 {"parent": self.container_2},
                 0,
                 pks_1,
+                None,
                 None
             ),
             (
@@ -683,6 +702,7 @@ class ContainerQATest(FindInit, TestCase):
                 {"location": self.alt_warehouse},
                 0,
                 pks_1,
+                None,
                 None
             ),
             (
@@ -695,10 +715,24 @@ class ContainerQATest(FindInit, TestCase):
                 {"parent": self.container_2},
                 2,
                 pks_1,
+                None,
                 None
             ),
             (
                 {
+                    "qa-move-qaparent": self.container_2.pk,
+                    "qa-move-create_treatment": True,
+                    "qa-move-year": 2019,
+                    "qa-move-treatment_type": packaging.pk,
+                },
+                {"parent": self.container_2},
+                2,
+                pks_1,
+                None,
+                init_treatment
+            ),
+            (
+                {
                     "qa-move-qalocation": self.alt_warehouse.pk,
                     "qa-move-create_treatment": True,
                     "qa-move-year": 2019,
@@ -707,6 +741,7 @@ class ContainerQATest(FindInit, TestCase):
                 {"location": self.alt_warehouse},
                 2,
                 pks_1,
+                None,
                 None
             ),
             (
@@ -716,7 +751,8 @@ class ContainerQATest(FindInit, TestCase):
                 {"location": self.alt_warehouse},
                 0,
                 pks_2,
-                check_location_of_children
+                check_location_of_children,
+                None
             ),
             (
                 {
@@ -725,12 +761,13 @@ class ContainerQATest(FindInit, TestCase):
                 {"location": self.alt_warehouse},
                 0,
                 pks_1,
-                check_reinit_parent
+                check_reinit_parent,
+                None
             ),
         ]
 
         for idx, lst in enumerate(data_check_lst):
-            data, check, nb_treat, pks, extra_check = lst
+            data, check, nb_treat, pks, extra_check, pre_init = lst
             # reinit
             self.container_1.location = self.main_warehouse
             self.container_1.parent = None
@@ -741,6 +778,12 @@ class ContainerQATest(FindInit, TestCase):
             self.container_4.location = self.main_warehouse
             self.container_4.parent = self.container_1
             self.container_4.save()
+            find_0.save()
+            FindTreatment.objects.exclude(id__isnull=True).delete()
+            Treatment.objects.exclude(id__isnull=True).delete()
+
+            if pre_init:
+                pre_init(find_0)
 
             init_nb_treat = Treatment.objects.count()
 
@@ -755,7 +798,7 @@ class ContainerQATest(FindInit, TestCase):
             self.assertEqual(init_nb_treat + nb_treat, final_nb_treat)
 
             if extra_check:
-                extra_check()
+                extra_check(idx)
 
             if not final_nb_treat:
                 self.assertEqual(FindTreatment.objects.filter(find=find_0).count(), 0)
@@ -771,6 +814,19 @@ class ContainerQATest(FindInit, TestCase):
             self.assertTrue(ft.full_location)
             self.assertTrue(ft.location_type)
 
+            find = Find.objects.get(pk=find_0.pk)
+            if pre_init:
+                # pre init add a packaging -> first full location is not changed
+                self.assertNotEqual(
+                    find.container_fisrt_full_location,
+                    find.container.generate_full_location()
+                )
+            else:
+                self.assertEqual(
+                    find.container_fisrt_full_location,
+                    find.container.generate_full_location()
+                )
+
 
 class ContainerTest(FindInit, TestCase):
     fixtures = WAREHOUSE_FIXTURES