65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.core.exceptions import PermissionDenied
|
|
|
|
from mayan.apps.common.tests import BaseTestCase
|
|
|
|
from ..classes import Permission, PermissionNamespace
|
|
from ..models import StoredPermission
|
|
from ..permissions import permission_role_view
|
|
|
|
from .literals import (
|
|
TEST_INVALID_PERMISSION_NAME, TEST_INVALID_PERMISSION_NAMESPACE_NAME,
|
|
TEST_PERMISSION_LABEL, TEST_PERMISSION_NAME,
|
|
TEST_PERMISSION_NAMESPACE_LABEL, TEST_PERMISSION_NAMESPACE_NAME
|
|
)
|
|
|
|
|
|
class PermissionTestCase(BaseTestCase):
|
|
def test_no_permissions(self):
|
|
with self.assertRaises(PermissionDenied):
|
|
Permission.check_user_permission(
|
|
permission=permission_role_view, user=self._test_case_user
|
|
)
|
|
|
|
def test_with_permissions(self):
|
|
self._test_case_group.user_set.add(self._test_case_user)
|
|
self._test_case_role.permissions.add(permission_role_view.stored_permission)
|
|
self._test_case_role.groups.add(self._test_case_group)
|
|
|
|
try:
|
|
Permission.check_user_permission(
|
|
permission=permission_role_view, user=self._test_case_user
|
|
)
|
|
except PermissionDenied:
|
|
self.fail('PermissionDenied exception was not expected.')
|
|
|
|
|
|
class StoredPermissionManagerTestCase(BaseTestCase):
|
|
create_test_case_superuser = False
|
|
create_test_case_user = False
|
|
|
|
def test_purge_obsolete_with_invalid(self):
|
|
StoredPermission.objects.create(
|
|
namespace=TEST_INVALID_PERMISSION_NAMESPACE_NAME,
|
|
name=TEST_INVALID_PERMISSION_NAME
|
|
)
|
|
|
|
StoredPermission.objects.purge_obsolete()
|
|
|
|
self.assertEqual(StoredPermission.objects.count(), 0)
|
|
|
|
def test_purge_obsolete_with_valid(self):
|
|
test_permission_namespace = PermissionNamespace(
|
|
label=TEST_PERMISSION_NAMESPACE_LABEL,
|
|
name=TEST_PERMISSION_NAMESPACE_NAME
|
|
)
|
|
test_permission = test_permission_namespace.add_permission(
|
|
label=TEST_PERMISSION_LABEL, name=TEST_PERMISSION_NAME
|
|
)
|
|
test_permission.stored_permission
|
|
|
|
StoredPermission.objects.purge_obsolete()
|
|
|
|
self.assertEqual(StoredPermission.objects.count(), 1)
|