def test_load_rules(rulefile, expectation):
rules = WhisperRules(rulespath=rule_path("empty.yml"))
rulefile = rule_path(rulefile)
ruleyaml = load_yaml_from_file(Path(rulefile))
with expectation:
rules.load_rules(rulefile)
assert len(rules.rules) == len(ruleyaml)
for rule_id in ruleyaml.keys():
assert rule_id in rules.rules
def test_parse_rule(rulefile, expectation):
rules = WhisperRules()
rulefile = Path(rule_path(rulefile))
rule_id, rule = load_yaml_from_file(rulefile).popitem()
with expectation:
parsed_rule = rules.parse_rule(rule_id, rule)
for key in parsed_rule:
assert parsed_rule[key] == rule[key]
def test_load_rule(dups, expectation):
rules = WhisperRules()
rulefile = Path(rule_path("valid.yml"))
rule_id, rule = load_yaml_from_file(rulefile).popitem()
with expectation:
for _ in range(dups):
rules.load_rule(rule_id, rule)
assert rule_id in rules.rules
assert rules.rules[rule_id] == rule
def test_load_yaml_from_file(rulefile, expected_count):
rulefile = Path(rule_path(rulefile))
assert len(load_yaml_from_file(rulefile)) == expected_count
def test_match(value, expectation):
rules = WhisperRules(rulespath=rule_path("valid.yml"))
assert rules.match("valid", value) == expectation
def test_load_rules_from_dict(rulefile, rules_added):
rules = WhisperRules()
rules_len = len(rules.rules)
custom_rules = load_yaml_from_file(Path(rule_path(rulefile)))
rules.load_rules_from_dict(custom_rules)
assert len(rules.rules) == rules_len + rules_added
def test_load_rules_from_file(rulefile, expectation):
rules = WhisperRules()
rules_len = len(rules.rules)
with expectation:
rules.load_rules_from_file(Path(rule_path(rulefile)))
assert len(rules.rules) == rules_len + 1