Skip to content

Commit 076d983

Browse files
committed
Fix tests.
1 parent a7cdc8c commit 076d983

File tree

1 file changed

+13
-9
lines changed

1 file changed

+13
-9
lines changed

example/tests/test_model_viewsets.py

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -26,15 +26,19 @@ def test_key_in_list_result(self):
2626
response = self.client.get(self.list_url)
2727
self.assertEqual(response.status_code, 200)
2828

29-
expected = {"user": []}
30-
for user in get_user_model().objects.all():
31-
expected['user'].append({
32-
'id': user.pk,
33-
'first_name': user.first_name,
34-
'last_name': user.last_name,
35-
'email': user.email})
36-
37-
self.assertEquals(expected, json.loads(response.content))
29+
user = get_user_model().objects.all()[0]
30+
expected = {"user": [{
31+
'id': user.pk,
32+
'first_name': user.first_name,
33+
'last_name': user.last_name,
34+
'email': user.email
35+
}]}
36+
37+
json_content = json.loads(response.content)
38+
self.assertEquals(expected.get('user'), json_content.get('user'))
39+
self.assertEquals(
40+
json_content.get("meta", {}).get('count', 0),
41+
get_user_model().objects.count())
3842

3943
def test_key_in_detail_result(self):
4044
"""

0 commit comments

Comments
 (0)