diff options
| author | Étienne Loks <etienne.loks@iggdrasil.net> | 2018-12-11 20:39:26 +0100 |
|---|---|---|
| committer | Étienne Loks <etienne.loks@iggdrasil.net> | 2018-12-11 20:39:26 +0100 |
| commit | 68427dd44be65b6e62bf222565198713ed1d664b (patch) | |
| tree | 9ac35e081a2e42e3d3eede0fc0ffaca35dae8632 /ishtar_common | |
| parent | b228ed1e8b626641a1429d2068b702597bf8aa0d (diff) | |
| download | Ishtar-68427dd44be65b6e62bf222565198713ed1d664b.tar.bz2 Ishtar-68427dd44be65b6e62bf222565198713ed1d664b.zip | |
Fix tests for treatments
Diffstat (limited to 'ishtar_common')
| -rw-r--r-- | ishtar_common/tests.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py index bd1833594..cf9f599c4 100644 --- a/ishtar_common/tests.py +++ b/ishtar_common/tests.py @@ -1235,10 +1235,14 @@ class ShortMenuTest(TestCase): self.assertFalse(str(tf.cached_label) in response.content) def _create_treatment(self): - from archaeological_finds.models import Treatment + from archaeological_finds.models import Treatment, TreatmentState + completed, created = TreatmentState.objects.get_or_create( + txt_idx='completed', defaults={"executed": True, "label": u"Done"} + ) return Treatment.objects.create( label="My treatment", - year=2052 + year=2052, + treatment_state=completed ) def test_treatment(self): |