Fix find test

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()