diff options
Diffstat (limited to 'archaeological_warehouse/tests.py')
| -rw-r--r-- | archaeological_warehouse/tests.py | 14 |
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) |