diff options
| -rw-r--r-- | archaeological_finds/tests.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index 569f81669..1c30f57b0 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -63,9 +63,10 @@ class FindInit(ContextRecordInit): return self.finds, self.base_finds def get_default_find(self, force=False): + finds, base_finds = self.create_finds(force=force) if force: - return self.create_finds()[-1] - return self.create_finds()[0] + return finds[-1], base_finds[-1] + return finds[0], base_finds[0] def tearDown(self): super(FindInit, self).tearDown() @@ -119,7 +120,7 @@ class AFindWizardCreationTest(WizardTest, FindInit, TestCase): ] def pre_wizard(self): - cr = self.create_context_record()[-1] + cr = self.get_default_context_record(force=True) self.form_datas[0].form_datas['selecrecord-find_creation']['pk'] = cr.pk self.find_number = models.Find.objects.count() self.basefind_number = models.BaseFind.objects.count() @@ -184,7 +185,7 @@ class ATreatmentWizardCreationTest(WizardTest, FindInit, TestCase): person = Person.objects.create(name="default") person.id = 1 person.save() - find = self.get_default_find(force=True) + find, base_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'][ |