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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
import pytest
from ssg import boolean_expression
from xml.dom import expatbuilder
class PlatformFunction(boolean_expression.Function):
def as_cpe_lang_xml(self):
return '<cpe-lang:logical-test negate="' + ('true' if self.is_not() else 'false') + \
'" operator="' + ('OR' if self.is_or() else 'AND') + '">' + \
''.join([arg.as_cpe_lang_xml() for arg in self.args]) + "</cpe-lang:logical-test>"
class PlatformSymbol(boolean_expression.Symbol):
def as_cpe_lang_xml(self):
return '<cpe-lang:fact-ref name="cpe:/a:' + self.name + ':' + '"/>'
class PlatformAlgebra(boolean_expression.Algebra):
def __init__(self):
super(PlatformAlgebra, self).__init__(symbol_cls=PlatformSymbol, function_cls=PlatformFunction)
@staticmethod
def as_cpe_lang_xml(expr):
s = '<cpe-lang:platform id="' + expr.as_id() + '">' + expr.as_cpe_lang_xml() + '</cpe-lang:platform>'
# A primitive but simple way to pretty-print an XML string
return expatbuilder.parseString(s, False).toprettyxml()
@pytest.fixture
def algebra():
return PlatformAlgebra()
@pytest.fixture
def expression1(algebra):
return algebra.parse(u'(oranges | banana) and not ~apple + !pie', simplify=True)
def test_id(expression1):
assert str(expression1.as_id()) == 'apple_and_banana_or_oranges_or_not_pie'
def test_as_cpe_xml(algebra, expression1):
xml = algebra.as_cpe_lang_xml(algebra.dnf(expression1))
assert xml == """<?xml version="1.0" ?>
<cpe-lang:platform id="apple_and_banana_or_apple_and_oranges_or_not_pie">
\t<cpe-lang:logical-test negate="false" operator="OR">
\t\t<cpe-lang:logical-test negate="false" operator="AND">
\t\t\t<cpe-lang:fact-ref name="cpe:/a:apple:"/>
\t\t\t<cpe-lang:fact-ref name="cpe:/a:banana:"/>
\t\t</cpe-lang:logical-test>
\t\t<cpe-lang:logical-test negate="false" operator="AND">
\t\t\t<cpe-lang:fact-ref name="cpe:/a:apple:"/>
\t\t\t<cpe-lang:fact-ref name="cpe:/a:oranges:"/>
\t\t</cpe-lang:logical-test>
\t\t<cpe-lang:logical-test negate="true" operator="AND">
\t\t\t<cpe-lang:fact-ref name="cpe:/a:pie:"/>
\t\t</cpe-lang:logical-test>
\t</cpe-lang:logical-test>
</cpe-lang:platform>
"""
def test_expression_with_parameter(algebra):
exp = algebra.parse(u'package[test] and os_release[rhel]')
assert exp(**{"package[test]": True, "os_release[rhel]": True})
def test_as_id(algebra):
exp1 = algebra.parse(u'package')
assert exp1.as_id() == "package"
exp2 = algebra.parse(u'package[test]')
assert exp2.as_id() == "package_test"
exp3 = algebra.parse(u'package[test]>=1.0,<1.2.0-3')
assert exp3.as_id() == "package_test_le_1_2_0_3_gt_or_eq_1_0"
def test_as_dict(algebra):
exp1 = algebra.parse(u'package')
exp1_dict = {
"arg": "",
"id": "package",
"name": "package",
'ver_specs': [],
'ver_specs_cpe': '',
'ver_specs_id': '',
'ver_specs_title': ''
}
assert exp1.as_dict() == exp1_dict
exp2 = algebra.parse(u'package[test]')
exp2_dict = {
"arg": "test",
"id": "package_test",
"name": "package",
'ver_specs': [],
'ver_specs_cpe': '',
'ver_specs_id': '',
'ver_specs_title': '',
}
assert exp2.as_dict() == exp2_dict
exp3 = algebra.parse(u'package[test]>1.0')
exp3_dict = {
"arg": "test",
"id": "package_test_gt_1_0",
"name": "package",
'ver_specs': [{'evr_op': 'greater than',
'evr_ver': '0:1.0-0',
'ev_ver': '0:1.0',
'id': 'gt_1_0',
'op': '>',
'ver': '1.0'}],
'ver_specs_cpe': 'gt:1.0',
'ver_specs_id': 'gt_1_0',
'ver_specs_title': 'greater than 1.0',
}
assert exp3.as_dict() == exp3_dict
|
