add tests for formatting

This commit is contained in:
William Ballenthin
2020-06-21 13:03:07 -06:00
parent 09fa5a4227
commit fea1177c5e

71
tests/test_fmt.py Normal file
View File

@@ -0,0 +1,71 @@
import textwrap
import capa.rules
EXPECTED = textwrap.dedent('''\
rule:
meta:
name: test rule
author: user@domain.com
scope: function
examples:
- foo1234
- bar5678
features:
- and:
- number: 1
- number: 2''')
def test_rule_reformat_top_level_elements():
rule = textwrap.dedent('''\
rule:
features:
- and:
- number: 1
- number: 2
meta:
name: test rule
author: user@domain.com
scope: function
examples:
- foo1234
- bar5678''')
assert capa.rules.Rule.from_yaml(rule).to_yaml() == EXPECTED
def test_rule_reformat_indentation():
rule = textwrap.dedent('''\
rule:
meta:
name: test rule
author: user@domain.com
scope: function
examples:
- foo1234
- bar5678
features:
- and:
- number: 1
- number: 2''')
assert capa.rules.Rule.from_yaml(rule).to_yaml() == EXPECTED
def test_rule_reformat_order():
rule = textwrap.dedent('''\
rule:
meta:
author: user@domain.com
examples:
- foo1234
- bar5678
scope: function
name: test rule
features:
- and:
- number: 1
- number: 2''')
assert capa.rules.Rule.from_yaml(rule).to_yaml() == EXPECTED