Passed
Pull Request — master (#30)
by Jan
05:10
created

get_parser()   A

Complexity

Conditions 2

Size

Total Lines 5
Code Lines 5

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 2
eloc 5
nop 1
dl 0
loc 5
rs 10
c 0
b 0
f 0
1
import pytest
2
3
from oscap_report.scap_results_parser.data_structures import Report, Rule
4
from oscap_report.scap_results_parser.scap_results_parser import \
5
    SCAPResultsParser
6
7
from ..constants import (PATH_TO_ARF, PATH_TO_ARF_WITH_REMOVED_INFO,
8
                         PATH_TO_EMPTY_XML_FILE,
9
                         PATH_TO_RULE_AND_CPE_CHECK_ARF,
10
                         PATH_TO_RULE_AND_CPE_CHECK_XCCDF,
11
                         PATH_TO_SIMPLE_RULE_FAIL_ARF,
12
                         PATH_TO_SIMPLE_RULE_FAIL_XCCDF,
13
                         PATH_TO_SIMPLE_RULE_PASS_ARF,
14
                         PATH_TO_SIMPLE_RULE_PASS_XCCDF, PATH_TO_XCCDF,
15
                         PATH_TO_XCCDF_WITH_REMOVED_INFO)
16
17
18
def get_parser(file_path):
19
    xml_data = None
20
    with open(file_path, "r", encoding="utf-8") as xml_report:
21
        xml_data = xml_report.read().encode()
22
    return SCAPResultsParser(xml_data)
23
24
25
@pytest.mark.parametrize("file_path, result", [
26
    (PATH_TO_ARF, True),
27
    (PATH_TO_SIMPLE_RULE_PASS_ARF, True),
28
    (PATH_TO_SIMPLE_RULE_FAIL_ARF, True),
29
    (PATH_TO_RULE_AND_CPE_CHECK_ARF, True),
30
    (PATH_TO_ARF_WITH_REMOVED_INFO, True),
31
    (PATH_TO_XCCDF, False),
32
    (PATH_TO_SIMPLE_RULE_PASS_XCCDF, False),
33
    (PATH_TO_SIMPLE_RULE_FAIL_XCCDF, False),
34
    (PATH_TO_RULE_AND_CPE_CHECK_XCCDF, False),
35
    (PATH_TO_XCCDF_WITH_REMOVED_INFO, False),
36
    (PATH_TO_EMPTY_XML_FILE, False),
37
])
38
def test_validation(file_path, result):
39
    parser = get_parser(file_path)
40
    assert parser.validate(parser.arf_schemas_path) == result
41
42
43
@pytest.mark.parametrize("file_path, number_of_cpe_platforms", [
44
    (PATH_TO_ARF, 13),
45
    (PATH_TO_XCCDF, 13),
46
    (PATH_TO_SIMPLE_RULE_PASS_ARF, 0),
47
    (PATH_TO_SIMPLE_RULE_FAIL_ARF, 0),
48
    (PATH_TO_ARF_WITH_REMOVED_INFO, 0),
49
    (PATH_TO_RULE_AND_CPE_CHECK_ARF, 1),
50
    (PATH_TO_SIMPLE_RULE_PASS_XCCDF, 0),
51
    (PATH_TO_SIMPLE_RULE_FAIL_XCCDF, 0),
52
    (PATH_TO_XCCDF_WITH_REMOVED_INFO, 0),
53
    (PATH_TO_RULE_AND_CPE_CHECK_XCCDF, 1),
54
])
55
def test_get_profile_info(file_path, number_of_cpe_platforms):
56
    parser = get_parser(file_path)
57
    report = parser.get_profile_info()
58
    assert len(report.cpe_platforms) == number_of_cpe_platforms
59
60
61
@pytest.mark.parametrize("file_path, number_of_rules", [
62
    (PATH_TO_ARF, 714),
63
    (PATH_TO_XCCDF, 714),
64
    (PATH_TO_SIMPLE_RULE_PASS_ARF, 1),
65
    (PATH_TO_SIMPLE_RULE_FAIL_ARF, 1),
66
    (PATH_TO_ARF_WITH_REMOVED_INFO, 1),
67
    (PATH_TO_RULE_AND_CPE_CHECK_ARF, 3),
68
    (PATH_TO_SIMPLE_RULE_PASS_XCCDF, 1),
69
    (PATH_TO_SIMPLE_RULE_FAIL_XCCDF, 1),
70
    (PATH_TO_XCCDF_WITH_REMOVED_INFO, 1),
71
    (PATH_TO_RULE_AND_CPE_CHECK_XCCDF, 3),
72
])
73
def test_get_info_about_rules_in_profile(file_path, number_of_rules):
74
    parser = get_parser(file_path)
75
    rules = parser.get_info_about_rules_in_profile()
76
    assert len(rules.keys()) == number_of_rules
77
    for rule in rules.values():
78
        assert isinstance(rule, Rule)
79
80
81
@pytest.mark.parametrize("file_path, contains_oval_tree", [
82
    (PATH_TO_ARF, True),
83
    (PATH_TO_XCCDF, False),
84
])
85
def test_parse_report(file_path, contains_oval_tree):
86
    parser = get_parser(file_path)
87
    report = parser.parse_report()
88
    assert isinstance(report, Report)
89
    assert report.profile_name is not None
90
    assert report.rules is not None
91
    rule_id = "xccdf_org.ssgproject.content_rule_accounts_passwords_pam_faillock_deny"
92
    assert isinstance(report.rules[rule_id], Rule)
93
    assert (report.rules[rule_id].oval_tree is not None) == contains_oval_tree
94