1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
import os
import ssg.rules
data_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "data"))
rule_dir = os.path.join(data_dir, "group_dir", "rule_dir")
def test_get_rule_dir_id():
assert ssg.rules.get_rule_dir_id("/some/path/fix_all_vulns/rule.yml") == "fix_all_vulns"
assert ssg.rules.get_rule_dir_id("/some/path/fix_all_vulns") == "fix_all_vulns"
assert ssg.rules.get_rule_dir_id(rule_dir) == 'rule_dir'
def test_is_rule_dir():
assert ssg.rules.is_rule_dir(rule_dir)
def test_applies_to_product():
assert ssg.rules.applies_to_product('shared', None)
assert ssg.rules.applies_to_product('rhel', None)
assert ssg.rules.applies_to_product('shared', 'rhel')
assert ssg.rules.applies_to_product('rhel', 'rhel')
assert not ssg.rules.applies_to_product('ol', 'rhel')
def test_find_rule_dirs():
rule_dirs = list(ssg.rules.find_rule_dirs(data_dir))
rule_ids = list(map(ssg.rules.get_rule_dir_id, rule_dirs))
assert rule_dir in rule_dirs
assert 'rule_dir' in rule_ids
assert 'random_dir' not in rule_ids
def test_get_rule_dir_ovals():
ovals = ssg.rules.get_rule_dir_ovals(rule_dir)
oval_files = list(map(os.path.basename, ovals))
assert len(ovals) == 2
assert 'shared.xml' in oval_files
assert 'rhel.xml' in oval_files
assert oval_files.index('shared.xml') > oval_files.index('rhel.xml')
rhel_ovals = ssg.rules.get_rule_dir_ovals(rule_dir, 'rhel')
assert rhel_ovals == ovals
ol_ovals = ssg.rules.get_rule_dir_ovals(rule_dir, 'ol')
assert ol_ovals != ovals
assert len(ol_ovals) == 1
assert 'rhel.xml' not in ol_ovals
|
