mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2025-07-09 12:30:15 +02:00
Add authutil for creating groups with authgroups.
This commit is contained in:
parent
eaa1cde01a
commit
15db817382
@ -25,6 +25,7 @@ from allianceauth.services.signals import (
|
|||||||
process_main_character_change,
|
process_main_character_change,
|
||||||
process_main_character_update
|
process_main_character_update
|
||||||
)
|
)
|
||||||
|
from allianceauth.groupmanagement.models import AuthGroup
|
||||||
|
|
||||||
|
|
||||||
class AuthUtils:
|
class AuthUtils:
|
||||||
@ -69,6 +70,28 @@ class AuthUtils:
|
|||||||
cls.connect_signals()
|
cls.connect_signals()
|
||||||
return state
|
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
|
@classmethod
|
||||||
def get_member_state(cls):
|
def get_member_state(cls):
|
||||||
try:
|
try:
|
||||||
|
@ -6,6 +6,7 @@ from django.test import TestCase
|
|||||||
from allianceauth.eveonline.models import (
|
from allianceauth.eveonline.models import (
|
||||||
EveCorporationInfo, EveAllianceInfo, EveCharacter
|
EveCorporationInfo, EveAllianceInfo, EveCharacter
|
||||||
)
|
)
|
||||||
|
from allianceauth.groupmanagement.models import AuthGroup
|
||||||
|
|
||||||
from .auth_utils import AuthUtils
|
from .auth_utils import AuthUtils
|
||||||
|
|
||||||
@ -15,7 +16,12 @@ class TestAuthUtils(TestCase):
|
|||||||
def test_can_create_user(self):
|
def test_can_create_user(self):
|
||||||
user = AuthUtils.create_user('Bruce Wayne')
|
user = AuthUtils.create_user('Bruce Wayne')
|
||||||
self.assertTrue(User.objects.filter(username='Bruce Wayne').exists())
|
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):
|
def test_can_add_main_character_2(self):
|
||||||
user = AuthUtils.create_user('Bruce Wayne')
|
user = AuthUtils.create_user('Bruce Wayne')
|
||||||
character = AuthUtils.add_main_character_2(
|
character = AuthUtils.add_main_character_2(
|
||||||
|
Loading…
x
Reference in New Issue
Block a user