diff options
Diffstat (limited to 'archaeological_context_records/tests.py')
| -rw-r--r-- | archaeological_context_records/tests.py | 16 |
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) |