mirror of
https://github.com/hanzoai/sentinel.git
synced 2026-09-29 22:34:33 +00:00
26 lines
1.0 KiB
Python
26 lines
1.0 KiB
Python
from django.conf import settings
|
|
|
|
from sentry.models.userrole import UserRole, manage_default_super_admin_role
|
|
from sentry.testutils.cases import TestCase
|
|
from sentry.testutils.silo import control_silo_test
|
|
|
|
|
|
@control_silo_test
|
|
class UserRoleTest(TestCase):
|
|
def setUp(self) -> None:
|
|
manage_default_super_admin_role()
|
|
|
|
def test_permissions_for_user(self):
|
|
user = self.create_user(email="[email protected]")
|
|
user2 = self.create_user(email="[email protected]")
|
|
role = UserRole.objects.create(name="test", permissions=["test1", "test2"])
|
|
role.users.add(user)
|
|
role2 = UserRole.objects.create(name="test2", permissions=["test2", "test3"])
|
|
role2.users.add(user)
|
|
assert sorted(UserRole.permissions_for_user(user.id)) == ["test1", "test2", "test3"]
|
|
assert sorted(UserRole.permissions_for_user(user2.id)) == []
|
|
|
|
def test_creates_super_admin_role(self):
|
|
role = UserRole.objects.get(name="Super Admin")
|
|
assert sorted(role.permissions) == sorted(settings.SENTRY_USER_PERMISSIONS)
|