def test_update(self, mock_create_rules, mock_delete_rule,
mock_rule_exists):
"""
Unittest to update a rule.
:param mock_create_rules:
:param mock_delete_rule:
:param mock_rule_exists:
:return:
"""
param = {
"type": "System Rule",
"rule_info": {
"action": "always",
"filter": "exit",
"systemcall": "init_module,delete_module"
",finit_module",
"field": ["arch=b32", "arch=b64"],
"key": "abcde"
}
}
old_rule = '-a always,exit -F arch=b32 -F arch=b64 -S init_module,' \
'delete_module,finit_module -F key=abc99'
new_rule = '-a always,exit -F arch=b32 -F arch=b64 -S init_module,' \
'delete_module,finit_module -F key=abcde'
mock_rule_exists.return_value = True
mock_delete_rule.return_value = {}
mock_create_rules.return_value = new_rule
ruleModel = RuleModel()
out_rule = ruleModel.update(old_rule, param)
self.assertEquals(out_rule, new_rule)
def test_update(self, mock_create_rules, mock_delete_rule,
mock_rule_exists):
"""
Unittest to update a rule.
:param mock_create_rules:
:param mock_delete_rule:
:param mock_rule_exists:
:return:
"""
param = {"type": "System Rule",
"rule_info": {"action": "always",
"filter": "exit",
"systemcall": "init_module,delete_module"
",finit_module",
"field": ["arch=b32", "arch=b64"],
"key": "abcde"}}
old_rule = '-a always,exit -F arch=b32 -F arch=b64 -S init_module,' \
'delete_module,finit_module -F key=abc99'
new_rule = '-a always,exit -F arch=b32 -F arch=b64 -S init_module,' \
'delete_module,finit_module -F key=abcde'
mock_rule_exists.return_value = True
mock_delete_rule.return_value = {}
mock_create_rules.return_value = new_rule
ruleModel = RuleModel()
out_rule = ruleModel.update(old_rule, param)
self.assertEquals(out_rule, new_rule)
