diff options
Diffstat (limited to 'archaeological_finds/tests.py')
| -rw-r--r-- | archaeological_finds/tests.py | 6 | 
1 files changed, 6 insertions, 0 deletions
| diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index e7afcbf66..2f871ae23 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -23,6 +23,7 @@ from django.test import TestCase  from ishtar_common.models import ImporterType, IshtarUser, ImporterColumn,\      FormaterType, ImportTarget +from ishtar_common.models import Person  from archaeological_finds import models, views  from archaeological_warehouse.models import Warehouse, WarehouseType @@ -266,6 +267,11 @@ class TreatmentWizardCreationTest(WizardTest, FindInit, TestCase):                  name="default", warehouse_type=WarehouseType.objects.all()[0])              warehouse.id = 1              warehouse.save() +        q = Person.objects.filter(pk=1) +        if not q.count(): +            person = Person.objects.create(name="default") +            person.id = 1 +            person.save()          self.create_finds({"label": u"Find 1"}, force=True)          self.treatment_number = models.Treatment.objects.count()          super(TreatmentWizardCreationTest, self).pre_wizard() | 
