def test_user_has_permission(self):
resolver = RuleEnforcementPermissionsResolver()
# Admin user, should always return true
user_db = self.users['admin']
permission_type = PermissionType.RULE_ENFORCEMENT_LIST
self.assertTrue(
resolver.user_has_permission(user_db=user_db,
permission_type=permission_type))
# Observer, should always return true for VIEW permissions
user_db = self.users['observer']
self.assertTrue(
resolver.user_has_permission(user_db=user_db,
permission_type=permission_type))
# No roles, should return false for everything
user_db = self.users['no_roles']
self.assertFalse(
resolver.user_has_permission(user_db=user_db,
permission_type=permission_type))
# Custom role with no permission grants, should return false for everything
user_db = self.users['1_custom_role_no_permissions']
self.assertFalse(
resolver.user_has_permission(user_db=user_db,
permission_type=permission_type))
# Custom role with "rule_list" grant
user_db = self.users['custom_role_rule_list_grant']
self.assertTrue(
resolver.user_has_permission(user_db=user_db,
permission_type=permission_type))
def test_user_has_resource_db_permission(self):
resolver = RuleEnforcementPermissionsResolver()
all_permission_types = PermissionType.get_valid_permissions_for_resource_type(
ResourceType.RULE_ENFORCEMENT)
# Admin user, should always return true
resource_db = self.resources['rule_enforcement_1']
user_db = self.users['admin']
self.assertTrue(
self._user_has_resource_db_permissions(
resolver=resolver,
user_db=user_db,
resource_db=resource_db,
permission_types=all_permission_types))
# Observer, should always return true for VIEW permission
user_db = self.users['observer']
self.assertTrue(
resolver.user_has_resource_db_permission(
user_db=user_db,
resource_db=self.resources['rule_enforcement_1'],
permission_type=PermissionType.RULE_ENFORCEMENT_VIEW))
self.assertTrue(
resolver.user_has_resource_db_permission(
user_db=user_db,
resource_db=self.resources['rule_enforcement_2'],
permission_type=PermissionType.RULE_ENFORCEMENT_VIEW))
# No roles, should return false for everything
user_db = self.users['no_roles']
self.assertFalse(
self._user_has_resource_db_permissions(
resolver=resolver,
user_db=user_db,
resource_db=resource_db,
permission_types=all_permission_types))
# Custom role with no permission grants, should return false for everything
user_db = self.users['1_custom_role_no_permissions']
self.assertFalse(
self._user_has_resource_db_permissions(
resolver=resolver,
user_db=user_db,
resource_db=resource_db,
permission_types=all_permission_types))
# Custom role with unrelated permission grant to parent pack
user_db = self.users['custom_role_pack_grant']
self.assertFalse(
resolver.user_has_resource_db_permission(
user_db=user_db,
resource_db=self.resources['rule_enforcement_1'],
permission_type=PermissionType.RULE_ENFORCEMENT_VIEW))
self.assertFalse(
resolver.user_has_resource_db_permission(
user_db=user_db,
resource_db=self.resources['rule_enforcement_2'],
permission_type=PermissionType.RULE_ENFORCEMENT_VIEW))
# Custom role with with grant on the parent pack
user_db = self.users['custom_role_rule_pack_grant']
self.assertTrue(
resolver.user_has_resource_db_permission(
user_db=user_db,
resource_db=self.resources['rule_enforcement_1'],
permission_type=PermissionType.RULE_ENFORCEMENT_VIEW))
self.assertTrue(
resolver.user_has_resource_db_permission(
user_db=user_db,
resource_db=self.resources['rule_enforcement_2'],
permission_type=PermissionType.RULE_ENFORCEMENT_VIEW))
# Custom role with a direct grant on rule
user_db = self.users['custom_role_rule_grant']
self.assertTrue(
resolver.user_has_resource_db_permission(
user_db=user_db,
resource_db=self.resources['rule_enforcement_3'],
permission_type=PermissionType.RULE_ENFORCEMENT_VIEW))
# Custom role - "rule_all" grant on the rule parent pack
user_db = self.users['custom_role_pack_rule_all_grant']
resource_db = self.resources['rule_enforcement_1']
self.assertTrue(
self._user_has_resource_db_permissions(
resolver=resolver,
user_db=user_db,
resource_db=resource_db,
permission_types=all_permission_types))
# Custom role - "rule_all" grant on the rule
user_db = self.users['custom_role_rule_all_grant']
resource_db = self.resources['rule_enforcement_1']
self.assertTrue(
self._user_has_resource_db_permissions(
resolver=resolver,
user_db=user_db,
resource_db=resource_db,
permission_types=all_permission_types))
# Custom role - "rule_modify" grant on rule_1
user_db = self.users['custom_role_rule_modify_grant']
resource_db = self.resources['rule_enforcement_1']
# "modify" also grants "view"
self.assertTrue(
resolver.user_has_resource_db_permission(
user_db=user_db,
resource_db=resource_db,
permission_type=PermissionType.RULE_ENFORCEMENT_VIEW))
