diff options
Diffstat (limited to 'archaeological_finds/tests.py')
| -rw-r--r-- | archaeological_finds/tests.py | 49 |
1 files changed, 48 insertions, 1 deletions
diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index 346aa0b1e..3c38583ae 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -2204,7 +2204,20 @@ class FindQATest(FindInit, TestCase): ) packaging = models.TreatmentType.objects.get(txt_idx="packaging") packaging.change_reference_location = True + packaging.change_current_location = False packaging.save() + packaging2, __ = models.TreatmentType.objects.get_or_create( + txt_idx="packaging-2", defaults={"virtual": False} + ) + packaging2.change_reference_location = True + packaging2.change_current_location = True + packaging2.save() + packaging3, __ = models.TreatmentType.objects.get_or_create( + txt_idx="packaging-3", defaults={"virtual": False} + ) + packaging3.change_reference_location = False + packaging3.change_current_location = True + packaging3.save() data_check_lst = [ ( @@ -2242,9 +2255,32 @@ class FindQATest(FindInit, TestCase): {"container_ref": container2, "container": None}, 1, ), + ( + { + "qa-packaging-container": container2.pk, + "qa-packaging-container_to_change": "current-and-reference", + "qa-packaging-create_treatment": True, + "qa-packaging-year": 2019, + "qa-packaging-treatment_type": packaging2.pk, + }, + {"container_ref": container2, "container": container2}, + 1, + ), + ( + { + "qa-packaging-container": container2.pk, + "qa-packaging-container_to_change": "current", + "qa-packaging-create_treatment": True, + "qa-packaging-year": 2019, + "qa-packaging-treatment_type": packaging3.pk, + }, + {"container_ref": None, "container": container2}, + 1, + ), ] - for data, check, nb_treat in data_check_lst: + for idx, lst in enumerate(data_check_lst): + data, check, nb_treat = lst # reinit find_0.container, find_0.container_ref = None, None find_0.skip_history_when_saving = True @@ -2266,6 +2302,17 @@ class FindQATest(FindInit, TestCase): final_nb_treat = models.Treatment.objects.count() self.assertEqual(init_nb_treat + nb_treat, final_nb_treat) + if not final_nb_treat: + self.assertEqual(models.FindTreatment.objects.filter(find=find_0).count(), 0) + continue + + treatment = models.Treatment.objects.order_by("-pk")[0] + q = models.FindTreatment.objects.filter(treatment=treatment) + self.assertEqual(q.filter(find=find_0).count(), 1) + ft = q.all()[0] + self.assertTrue(ft.full_location) + self.assertTrue(ft.location_type) + class FindHistoryTest(FindInit, TestCase): fixtures = FIND_FIXTURES |