class UserPermissionsTestCase(SimpleTestCase):
def setUp(self, *args, **kwargs):
super(UserPermissionsTestCase, self).setUp(*args, **kwargs)
self.user = MojUser(5, generate_tokens(), {
'first_name': 'Sam',
'last_name': 'Hall',
'permissions': [
'allowed_permission_1',
'allowed_permission_2',
'allowed_permission_3'
]
})
def test_has_perm_succeeds_if_present(self):
self.assertTrue(self.user.has_perm('allowed_permission_1'))
def test_has_perm_fails_if_absent(self):
self.assertFalse(self.user.has_perm('forbidden_permission'))
def test_has_perms_succeeds_if_all_present(self):
self.assertTrue(self.user.has_perms([
'allowed_permission_1',
'allowed_permission_2'
]))
def test_has_perms_fails_if_any_absent(self):
self.assertFalse(self.user.has_perms([
'allowed_permission_1',
'forbidden_permission'
]))
def test_no_permissions_fails_gracefully(self):
user = MojUser(6, generate_tokens(), {
'first_name': 'Sam',
'last_name': 'Halle',
})
user.get_all_permissions()
self.assertFalse(user.has_perm('forbidden_permission'))
self.assertFalse(user.has_perms([
'allowed_permission_1',
'forbidden_permission'
]))
def test_no_permissions_fails_gracefully(self):
user = MojUser(6, generate_tokens(), {
'first_name': 'Sam',
'last_name': 'Halle',
})
user.get_all_permissions()
self.assertFalse(user.has_perm('forbidden_permission'))
self.assertFalse(user.has_perms([
'allowed_permission_1',
'forbidden_permission'
]))
