def test_add_delete_get_policy_rule(self): id = uuidutils.generate_uuid() rule_str = "p(x) :- q(x)" policy_name = "classification" comment = "None" db_policy_rules.add_policy_rule(id=id, policy_name=policy_name, rule=rule_str, comment=comment) db_policy_rules.delete_policy_rule(id) rule = db_policy_rules.get_policy_rule(id, policy_name) self.assertIsNone(rule)
def test_add_delete_get_policy_rule(self): id = uuidutils.generate_uuid() rule_str = "p(x) :- q(x)" policy_name = "classification" comment = "None" db_policy_rules.add_policy_rule(id=id, policy_name=policy_name, rule=rule_str, comment=comment) db_policy_rules.delete_policy_rule(id) rule = db_policy_rules.get_policy_rule(id, policy_name) self.assertEqual(rule, None)
def test_add_delete_get_deleted_policy_rule(self): id = uuidutils.generate_uuid() rule_str = "p(x) :- q(x)" policy_name = "classification" comment = "None" rule1 = db_policy_rules.add_policy_rule(id=id, policy_name=policy_name, rule=rule_str, comment=comment) db_policy_rules.delete_policy_rule(id) rule2 = db_policy_rules.get_policy_rule(id, policy_name, deleted=True) self.assertEqual(rule1.id, rule2.id) self.assertNotEqual(rule1.deleted, rule2.deleted)
def test_add_get_policy_rule(self): id = uuidutils.generate_uuid() rule_str = "p(x) :- q(x)" policy_name = "classification" comment = "None" db_policy_rules.add_policy_rule(id=id, policy_name=policy_name, rule=rule_str, comment=comment) rule = db_policy_rules.get_policy_rule(id, policy_name) self.assertEqual(id, rule.id) self.assertEqual(policy_name, rule.policy_name) self.assertEqual(rule_str, rule.rule) self.assertEqual(comment, rule.comment)
def get_item(self, id_, params, context=None): """Retrieve item with id id_ from model. Args: id_: The ID of the item to retrieve params: A dict-like object containing parameters from the request query string and body. context: Key-values providing frame of reference of request Returns: The matching item or None if item with id_ does not exist. """ policy_name = self.policy_name(context) rule = db_policy_rules.get_policy_rule(id_, policy_name) if rule is None: return d = {'rule': rule.rule, 'id': rule.id, 'comment': rule.comment} return d
def get_item(self, id_, params, context=None): """Retrieve item with id id_ from model. Args: id_: The ID of the item to retrieve params: A dict-like object containing parameters from the request query string and body. context: Key-values providing frame of reference of request Returns: The matching item or None if item with id_ does not exist. """ policy_name = self.policy_name(context) rule = db_policy_rules.get_policy_rule(id_, policy_name) if rule is None: return d = { 'rule': rule.rule, 'id': rule.id, 'comment': rule.comment, 'name': rule.name } return d