tests: fix render test for MBC

This commit is contained in:
Willi Ballenthin
2022-06-20 14:25:18 -06:00
parent 246ef58e7b
commit a453258a51

View File

@@ -64,6 +64,9 @@ def test_render_meta_mbc():
rule: rule:
meta: meta:
name: test rule name: test rule
scope: function
authors:
- foo
mbc: mbc:
- {:s} - {:s}
features: features:
@@ -73,12 +76,12 @@ def test_render_meta_mbc():
) )
) )
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["mbc"][0] mbc = rule_meta.mbc[0]
assert attack["id"] == id assert mbc.id == id
assert attack["objective"] == objective assert mbc.objective == objective
assert attack["behavior"] == behavior assert mbc.behavior == behavior
assert attack["method"] == method assert mbc.method == method
assert capa.render.utils.format_parts_id(attack) == canonical assert capa.render.utils.format_parts_id(mbc) == canonical