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' ]))