Adapt tests for new fixtures

1 files changed, 9 insertions, 7 deletions

diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index 4226b42ea..e697459b5 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -60,7 +60,7 @@ class ImportContextRecordTest(ImportTest, TestCase):
                 unit__txt_idx='not_in_context').count(), 3)
         self.assertEqual(
             models.ContextRecord.objects.filter(
-                unit__txt_idx='layer').count(), 1)
+                unit__txt_idx='negative').count(), 1)
 
     def test_model_limitation(self):
         old_nb = models.ContextRecord.objects.count()
@@ -405,12 +405,14 @@ class ContextRecordSearchTest(ContextRecordInit, TestCase):
         cr = self.context_records[0]
         c = Client()
 
+        su = models.Unit.objects.get(txt_idx='stratigraphic-unit')
         neg = models.Unit.objects.get(txt_idx='negative')
-        dig = models.Unit.objects.get(txt_idx='digging')
-        dest = models.Unit.objects.get(txt_idx='partial_destruction')
-        cr.unit = (dig)
+        dest = models.Unit.objects.get(txt_idx='sector')
+        dest.parent = su
+        dest.save()
+        cr.unit = (neg)
         cr.save()
-        search = {'unit': dig.pk}
+        search = {'unit': neg.pk}
 
         # no result when no authentication
         response = c.get(reverse('get-contextrecord'), search)
@@ -424,7 +426,7 @@ class ContextRecordSearchTest(ContextRecordInit, TestCase):
         res = json.loads(response.content)
         self.assertTrue(res['total'] == 1)
         self.assertEqual(res["rows"][0]["unit"],
-                         unicode(dig))
+                         unicode(neg))
 
         # no result for the brother
         search = {'unit': dest.pk}
@@ -433,7 +435,7 @@ class ContextRecordSearchTest(ContextRecordInit, TestCase):
         self.assertEqual(json.loads(response.content)['total'], 0)
 
         # one result for the father
-        search = {'unit': neg.pk}
+        search = {'unit': su.pk}
         response = c.get(reverse('get-contextrecord'), search)
         self.assertEqual(response.status_code, 200)
         self.assertEqual(json.loads(response.content)['total'], 1)