src/notes/tests/models/auth.py
changeset 131 adad5563603c
parent 117 9864fe2067cd
child 142 56850f5c73f6
--- a/src/notes/tests/models/auth.py	Sun Jul 30 01:02:09 2017 +0200
+++ b/src/notes/tests/models/auth.py	Mon Jul 31 23:18:38 2017 +0200
@@ -20,6 +20,10 @@
             email='test_user@emial.com',
             password='top_secret'
         )
+
+        group = Group(name='test_user2 group')
+        group.save()
+
         self.user2 = User.objects.create_user(
             username='test_user2',
             email='test_user@emial.com',
@@ -32,3 +36,23 @@
         self.assertIsNotNone(self.group)
         self.assertIsNotNone(self.group.profile)
 
+    def test_create_personal_group(self):
+        self.assertIsNotNone(self.user1.profile)
+        self.assertIsNotNone(self.user2.profile)
+        default_group1 = self.user1.profile.default_group
+        self.assertIsNotNone(default_group1)
+        self.assertIsNotNone(default_group1.profile)
+        self.assertTrue(default_group1.profile.is_personal)
+        self.assertEqual(default_group1.profile.owner, self.user1)
+        self.assertEqual(default_group1.profile.owner_personal, self.user1)
+
+    def test_create_personal_group_new_name(self):
+        default_group2 = self.user2.profile.default_group
+        self.assertIsNotNone(default_group2)
+        self.assertEqual(default_group2.name, "test_user2 group 1")
+
+    def test_create_personal_group(self):
+        personal_group = Group.objects.get(profile__owner_personal=self.user1)
+        self.assertIsNotNone(personal_group)
+        self.assertIn(self.user1, personal_group.user_set.all())
+