✨ treatment state refactoring -> distinct input status and treatment status (refs #6182) - treatments: search criteria

1 files changed, 9 insertions, 5 deletions

diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py
index d00156d90..896014c0d 100644
--- a/archaeological_warehouse/tests.py
+++ b/archaeological_warehouse/tests.py
@@ -41,9 +41,10 @@ from ishtar_common.tests import (
 from ishtar_common.models import IshtarSiteProfile, ProfileType, UserProfile
 from archaeological_operations.models import Operation
 from archaeological_context_records.models import ContextRecord
-from archaeological_finds.models import \
-    Find, MaterialType, Treatment, TreatmentType, FindTreatment, TreatmentState
-
+from archaeological_finds.models import (
+    Find, MaterialType, Treatment, TreatmentType, FindTreatment,
+    TreatmentStatus, TreatmentInputStatus
+)
 from archaeological_warehouse import models, views, forms, serializers
 
 from archaeological_operations.serializers import operation_serialization
@@ -947,13 +948,16 @@ class ContainerTest(FindInit, TestCase):
         self.assertEqual(find.container_fisrt_full_location, full_localisation)
         self.assertEqual(find.container_ref_fisrt_full_location, full_localisation)
 
-        treat_state = TreatmentState.get_completed_state()
+        treat_input_status = TreatmentInputStatus.get_validated_state()
+        treat_status = TreatmentStatus.get_completed_state()
+
         t = Treatment.objects.create(
             container=container_2,
             year=2024,
             start_date=datetime.date(2024, 2, 29),
             location=self.main_warehouse,
-            treatment_state=treat_state
+            input_status=treat_input_status,
+            treatment_status=treat_status,
         )
         treat_type = TreatmentType.objects.get(txt_idx="packaging")
         t.treatment_types.add(treat_type)