src/notes/tests/api/auth.py
changeset 131 adad5563603c
parent 117 9864fe2067cd
--- a/src/notes/tests/api/auth.py	Sun Jul 30 01:02:09 2017 +0200
+++ b/src/notes/tests/api/auth.py	Mon Jul 31 23:18:38 2017 +0200
@@ -20,12 +20,12 @@
         User = get_user_model()
         self.user1 = User.objects.create_user(
             username='test_user1',
-            email='test_user@emial.com',
+            email='test_user@email.com',
             password='top_secret'
         )
         self.user2 = User.objects.create_user(
             username='test_user2',
-            email='test_user@emial.com',
+            email='test_user@email.com',
             password='top_secret'
         )
 
@@ -58,14 +58,24 @@
 
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
-        self.assertEqual(2, len(response.data), "Must have 2 groups")
-        for group_def in response.data:
-            self.assertIn('name', group_def.keys())
-            self.assertIn('owner', group_def.keys())
-            self.assertIn('description', group_def.keys())
+        resp_json = response.json()
+        self.assertIn('count', resp_json)
+        self.assertIn('results', resp_json)
+        self.assertEqual(4, resp_json['count'], "Must have 4 groups")
+        self.assertEqual(4, len(resp_json['results']), "Must have 4 groups")
+
+        for group_def in resp_json['results']:
+            self.assertIn('name', group_def)
+            self.assertIn('owner', group_def)
+            self.assertIn('is_personal', group_def)
+            self.assertIn('description', group_def)
             self.assertIn(group_def.get('owner'), ['test_user1', 'test_user2'])
-            self.assertIn(group_def.get('description'), ['This is the group 1', 'This is the group 2'])
-            self.assertIn(group_def.get('name'), ['group1', 'group2'])
+            self.assertIn(group_def.get('description'), [
+                'This is the group 1',
+                'This is the group 2',
+                'test_user1 personal group',
+                'test_user2 personal group'])
+            self.assertIn(group_def.get('name'), ['group1', 'group2', 'test_user1 group', 'test_user2 group'])
 
     def test_create_group(self):
         url = reverse('auth_group-list')