diff options
Diffstat (limited to 'ishtar_common/tests.py')
| -rw-r--r-- | ishtar_common/tests.py | 12 |
1 files changed, 7 insertions, 5 deletions
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): |