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 | 548c783c61f1183206c7a907d48576491a20fd6d (patch) | |
| tree | 9ac35e081a2e42e3d3eede0fc0ffaca35dae8632 /ishtar_common/tests.py | |
| parent | 6e25c161956c5c905e882ab2572a218ef7cb2914 (diff) | |
| download | Ishtar-548c783c61f1183206c7a907d48576491a20fd6d.tar.bz2 Ishtar-548c783c61f1183206c7a907d48576491a20fd6d.zip | |
Fix tests for treatments
Diffstat (limited to 'ishtar_common/tests.py')
| -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): |