From bc6cf94def73926e6f6bb986d61e5861a17584e2 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Fri, 21 Feb 2025 17:06:53 +0100 Subject: ✨ treatment state refactoring -> distinct input status and treatment status (refs #6182) - treatments: search criteria MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ishtar_common/tests.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) (limited to 'ishtar_common/tests.py') diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py index 0341d932a..5692c9863 100644 --- a/ishtar_common/tests.py +++ b/ishtar_common/tests.py @@ -2525,13 +2525,15 @@ class ShortMenuTest(TestCase): def _create_treatment(self): Treatment = apps.get_model("archaeological_finds", "Treatment") - TreatmentState = apps.get_model("archaeological_finds", "TreatmentState") - - completed, created = TreatmentState.objects.get_or_create( - txt_idx="completed", defaults={"executed": True, "label": "Done"} + TreatmentInputStatus = apps.get_model("archaeological_finds", "TreatmentInputStatus") + completed, created = TreatmentInputStatus.objects.get_or_create( + txt_idx="validated", defaults={"executed": True, "label": "Validated"} ) + completed.executed = True + completed.save() + return Treatment.objects.create( - label="My treatment", year=2052, treatment_state=completed + label="My treatment", year=2052, input_status=completed ) def test_treatment(self): -- cgit v1.2.3