From a453258a5171bd09167df217c4954303bf29ae02 Mon Sep 17 00:00:00 2001 From: Willi Ballenthin Date: Mon, 20 Jun 2022 14:25:18 -0600 Subject: [PATCH] tests: fix render test for MBC --- tests/test_render.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/tests/test_render.py b/tests/test_render.py index f87fd11c..cc569177 100644 --- a/tests/test_render.py +++ b/tests/test_render.py @@ -64,6 +64,9 @@ def test_render_meta_mbc(): rule: meta: name: test rule + scope: function + authors: + - foo mbc: - {:s} features: @@ -73,12 +76,12 @@ def test_render_meta_mbc(): ) ) r = capa.rules.Rule.from_yaml(rule) - rule_meta = capa.render.result_document.convert_meta_to_result_document(r.meta) - attack = rule_meta["mbc"][0] + rule_meta = capa.render.result_document.RuleMetadata.from_capa(r) + mbc = rule_meta.mbc[0] - assert attack["id"] == id - assert attack["objective"] == objective - assert attack["behavior"] == behavior - assert attack["method"] == method + assert mbc.id == id + assert mbc.objective == objective + assert mbc.behavior == behavior + assert mbc.method == method - assert capa.render.utils.format_parts_id(attack) == canonical + assert capa.render.utils.format_parts_id(mbc) == canonical