diff options
Diffstat (limited to 'archaeological_context_records/tests.py')
| -rw-r--r-- | archaeological_context_records/tests.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py index c56964670..87eb80c57 100644 --- a/archaeological_context_records/tests.py +++ b/archaeological_context_records/tests.py @@ -231,7 +231,10 @@ class ContextRecordSearchTest(ContextRecordInit, TestCase): c.login(username=self.username, password=self.password) response = c.get(reverse('get-contextrecord'), search) self.assertEqual(response.status_code, 200) - self.assertEqual(json.loads(response.content)['total'], 1) + res = json.loads(response.content) + self.assertTrue(res['total'] == 1) + self.assertEqual(res["rows"][0]["unit"], + unicode(dig)) # no result for the brother search = {'unit': dest.pk} @@ -268,7 +271,8 @@ class ContextRecordSearchTest(ContextRecordInit, TestCase): c.login(username=self.username, password=self.password) response = c.get(reverse('get-contextrecord'), search) self.assertEqual(response.status_code, 200) - self.assertEqual(json.loads(response.content)['total'], 1) + res = json.loads(response.content) + self.assertTrue(res['total'] == 1) # no result for the brother search = {'datings__period': recent_neo.pk} |