diff --git a/allianceauth/tests/auth_utils.py b/allianceauth/tests/auth_utils.py index d5642d08..7120c2de 100644 --- a/allianceauth/tests/auth_utils.py +++ b/allianceauth/tests/auth_utils.py @@ -25,6 +25,7 @@ from allianceauth.services.signals import ( process_main_character_change, process_main_character_update ) +from allianceauth.groupmanagement.models import AuthGroup class AuthUtils: @@ -69,6 +70,28 @@ class AuthUtils: cls.connect_signals() return state + @classmethod + def _create_group(cls, group_name): + group = Group.objects.create(name=group_name) + AuthGroup.objects.create(group=group) + return group + + @classmethod + def create_group(cls, group_name, disconnect_signals=False): + """ + Create a new group and its authgroup. + + Args: + group_name: name of the group + disconnect_signals: whether or not to run without signals. + """ + if disconnect_signals: + cls.disconnect_signals() + group = cls._create_group(group_name) + if disconnect_signals: + cls.connect_signals() + return group + @classmethod def get_member_state(cls): try: diff --git a/allianceauth/tests/test_auth_utils.py b/allianceauth/tests/test_auth_utils.py index 1eb5d4a5..d6c5644a 100644 --- a/allianceauth/tests/test_auth_utils.py +++ b/allianceauth/tests/test_auth_utils.py @@ -6,6 +6,7 @@ from django.test import TestCase from allianceauth.eveonline.models import ( EveCorporationInfo, EveAllianceInfo, EveCharacter ) +from allianceauth.groupmanagement.models import AuthGroup from .auth_utils import AuthUtils @@ -15,7 +16,12 @@ class TestAuthUtils(TestCase): def test_can_create_user(self): user = AuthUtils.create_user('Bruce Wayne') self.assertTrue(User.objects.filter(username='Bruce Wayne').exists()) - + + def test_can_create_group(self): + group = AuthUtils.create_group('Test Group') + self.assertTrue(Group.objects.filter(name='Test Group').exists()) + self.assertTrue(AuthGroup.objects.filter(group=group).exists()) + def test_can_add_main_character_2(self): user = AuthUtils.create_user('Bruce Wayne') character = AuthUtils.add_main_character_2(