tests: fix render test for ATT&CK metadata

This commit is contained in:
Willi Ballenthin
2022-06-20 14:24:01 -06:00
parent a5979d3b4d
commit 246ef58e7b

View File

@@ -28,6 +28,9 @@ def test_render_meta_attack():
rule: rule:
meta: meta:
name: test rule name: test rule
scope: function
authors:
- foo
att&ck: att&ck:
- {:s} - {:s}
features: features:
@@ -37,13 +40,13 @@ def test_render_meta_attack():
) )
) )
r = capa.rules.Rule.from_yaml(rule) r = capa.rules.Rule.from_yaml(rule)
rule_meta = capa.render.result_document.convert_meta_to_result_document(r.meta) rule_meta = capa.render.result_document.RuleMetadata.from_capa(r)
attack = rule_meta["att&ck"][0] attack = rule_meta.attack[0]
assert attack["id"] == id assert attack.id == id
assert attack["tactic"] == tactic assert attack.tactic == tactic
assert attack["technique"] == technique assert attack.technique == technique
assert attack["subtechnique"] == subtechnique assert attack.subtechnique == subtechnique
assert capa.render.utils.format_parts_id(attack) == canonical assert capa.render.utils.format_parts_id(attack) == canonical