From 4eede10498e19aa609a697e44db360ce5ee3b70e Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Tue, 20 Dec 2016 18:43:34 +0100 Subject: Tests: get default find for treatment test --- archaeological_finds/tests.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index 20c0604d5..7a2485fc1 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -62,7 +62,9 @@ class FindInit(ContextRecordInit): self.finds.append(find) return self.finds, self.base_finds - def get_default_find(self): + def get_default_find(self, force=False): + if force: + return self.create_finds()[-1] return self.create_finds()[0] def tearDown(self): @@ -188,7 +190,11 @@ class ATreatmentWizardCreationTest(WizardTest, FindInit, TestCase): person.id = 1 person.save() if not models.Find.objects.filter(pk=1).count(): - self.create_finds({"pk": 1, "label": u"Find 1"}, force=True) + find = self.get_default_find(force=True) + self.form_datas[0].form_datas['selecfind-treatment_creation'][ + 'pk'] = find.pk + self.form_datas[0].form_datas['selecfind-treatment_creation'][ + 'resulting_pk'] = find.pk self.treatment_number = models.Treatment.objects.count() super(ATreatmentWizardCreationTest, self).pre_wizard() -- cgit v1.2.3