import logging
from ...constants import OVAL_NAMESPACES, STR_TO_BOOL, OVALREFATTR_TO_TAG
from ...xml import ElementTree
from ..general import OVALComponent, load_notes, required_attribute
[docs]
def load_test(oval_test_xml_el):
notes_el = oval_test_xml_el.find("./{%s}notes" % OVAL_NAMESPACES.definition)
test = Test(
oval_test_xml_el.tag,
required_attribute(oval_test_xml_el, "id"),
required_attribute(oval_test_xml_el, "check"),
)
test.check_existence = oval_test_xml_el.get(
"check_existence", "at_least_one_exists"
)
test.state_operator = oval_test_xml_el.get("state_operator", "AND")
test.comment = required_attribute(oval_test_xml_el, "comment")
test.deprecated = STR_TO_BOOL.get(oval_test_xml_el.get("deprecated", ""), False)
test.notes = load_notes(notes_el)
test.version = required_attribute(oval_test_xml_el, "version")
for child_node_el in oval_test_xml_el:
if child_node_el.tag.endswith("object"):
test.set_object_ref(child_node_el.get("object_ref"))
test.object_ref_tag = child_node_el.tag
elif child_node_el.tag.endswith("state"):
test.add_state_ref(child_node_el.get("state_ref"))
test.state_ref_tag = child_node_el.tag
else:
logging.warning("Unknown element '{}'\n".format(child_node_el.tag))
return test
[docs]
class ExceptionDuplicateObjectReferenceInTest(Exception):
pass
[docs]
class ExceptionMissingObjectReferenceInTest(Exception):
pass
[docs]
class Test(OVALComponent):
check_existence = "at_least_one_exists"
state_operator = "AND"
comment = ""
object_ref_tag = "object"
state_ref_tag = "state"
def __init__(self, tag, id_, check):
super(Test, self).__init__(tag, id_)
self.check = check
self.object_ref = ""
self.state_refs = []
[docs]
def set_object_ref(self, object_ref):
if self.object_ref != "":
raise ExceptionDuplicateObjectReferenceInTest(
"Problematic OVAL test: {}".format(self.id_)
)
self.object_ref = object_ref
[docs]
def add_state_ref(self, state_ref):
self.state_refs.append(state_ref)
[docs]
def get_xml_element(self):
test_el = super(Test, self).get_xml_element()
test_el.set("check", self.check)
test_el.set("comment", self.comment)
if self.check_existence != "at_least_one_exists":
test_el.set("check_existence", self.check_existence)
test_el.set("state_operator", self.state_operator)
if self.object_ref == "":
raise ExceptionMissingObjectReferenceInTest(
"Problematic OVAL test: {}".format(self.id_)
)
object_ref_el = ElementTree.Element(self.object_ref_tag)
object_ref_el.set("object_ref", self.object_ref)
test_el.append(object_ref_el)
for state_ref in self.state_refs:
state_ref_el = ElementTree.Element(self.state_ref_tag)
state_ref_el.set("state_ref", state_ref)
test_el.append(state_ref_el)
return test_el
[docs]
def translate_id(self, translator, store_defname=False):
super(Test, self).translate_id(translator)
self.object_ref = translator.generate_id(
"{%s}%s" % (OVAL_NAMESPACES.definition, OVALREFATTR_TO_TAG["object_ref"]),
self.object_ref,
)
translated_state_refs = []
for state_ref in self.state_refs:
translated_state_refs.append(
translator.generate_id(
"{%s}%s"
% (OVAL_NAMESPACES.definition, OVALREFATTR_TO_TAG["state_ref"]),
state_ref,
)
)
self.state_refs = translated_state_refs