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
|
import ssg.xml
import os
import xml.etree.ElementTree as ET
data_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "data"))
data_stream_path = os.path.join(data_dir, "simple_data_stream.xml")
def test_xml_content():
root = ET.parse(data_stream_path).getroot()
xml_content = ssg.xml.XMLContent(root)
expected_component_refs = {
"#scap_org.open-scap_comp_test_single_rule.oval.xml":
"scap_org.open-scap_cref_test_single_rule.oval.xml",
"#scap_org.open-scap_comp_cpe.oval.xml":
"scap_org.open-scap_cref_cpe.oval.xml"
}
assert xml_content.component_refs == expected_component_refs
expected_uris = {
"#scap_org.open-scap_cref_test_single_rule.oval.xml":
"test_single_rule.oval.xml",
"#scap_org.open-scap_cref_cpe.oval.xml":
"cpe.oval.xml"
}
assert xml_content.uris == expected_uris
assert "OVAL" in xml_content.components
assert "test_single_rule.oval.xml" in xml_content.components["OVAL"]
assert "cpe.oval.xml" in xml_content.components["OVAL"]
xml_component = xml_content.components["OVAL"]["test_single_rule.oval.xml"]
assert type(xml_component) is ssg.xml.XMLComponent
|