Updates expected number of groups in tests

This commit is contained in:
Mouse Reeve 2022-12-11 11:35:48 -08:00
parent f779c57f33
commit a25a5b1032

View file

@ -12,7 +12,7 @@ class InitDB(TestCase):
def test_init_groups(self): def test_init_groups(self):
"""Create groups""" """Create groups"""
initdb.init_groups() initdb.init_groups()
self.assertEqual(Group.objects.count(), 3) self.assertEqual(Group.objects.count(), 4)
self.assertTrue(Group.objects.filter(name="admin").exists()) self.assertTrue(Group.objects.filter(name="admin").exists())
self.assertTrue(Group.objects.filter(name="moderator").exists()) self.assertTrue(Group.objects.filter(name="moderator").exists())
self.assertTrue(Group.objects.filter(name="editor").exists()) self.assertTrue(Group.objects.filter(name="editor").exists())
@ -87,7 +87,7 @@ class InitDB(TestCase):
command.handle() command.handle()
# everything should have been called # everything should have been called
self.assertEqual(Group.objects.count(), 3) self.assertEqual(Group.objects.count(), 4)
self.assertTrue(Permission.objects.exists()) self.assertTrue(Permission.objects.exists())
self.assertEqual(models.Connector.objects.count(), 3) self.assertEqual(models.Connector.objects.count(), 3)
self.assertEqual(models.SiteSettings.objects.count(), 1) self.assertEqual(models.SiteSettings.objects.count(), 1)
@ -99,7 +99,7 @@ class InitDB(TestCase):
command.handle(limit="group") command.handle(limit="group")
# everything should have been called # everything should have been called
self.assertEqual(Group.objects.count(), 3) self.assertEqual(Group.objects.count(), 4)
self.assertEqual(models.Connector.objects.count(), 0) self.assertEqual(models.Connector.objects.count(), 0)
self.assertEqual(models.SiteSettings.objects.count(), 0) self.assertEqual(models.SiteSettings.objects.count(), 0)
self.assertEqual(models.LinkDomain.objects.count(), 0) self.assertEqual(models.LinkDomain.objects.count(), 0)