def parse_xml_instance(filename, check_version = True):
"""Parse a MAEC instance and return the correct Binding and API objects
Returns a dictionary of MAEC Package or Bundle Binding/API Objects"""
object_dictionary = {}
entity_parser = EntityParser()
object_dictionary['binding'] = entity_parser.parse_xml_to_obj(filename, check_version)
object_dictionary['api'] = entity_parser.parse_xml(filename, check_version)
return object_dictionary
def parse_xml_instance(filename, check_version = True):
"""Parse a MAEC instance and return the correct Binding and API objects
Returns a dictionary of MAEC Package or Bundle Binding/API Objects"""
object_dictionary = {}
entity_parser = EntityParser()
object_dictionary['binding'] = entity_parser.parse_xml_to_obj(filename, check_version)
object_dictionary['api'] = entity_parser.parse_xml(filename, check_version)
return object_dictionary
def test_valid_bundle(self):
valid_bundle = """
<maecBundle:MAEC_Bundle
xmlns:maecBundle="http://maec.mitre.org/XMLSchema/maec-bundle-4"
id="example:bundle-1" schema_version="4.1">
</maecBundle:MAEC_Bundle>
"""
parser = EntityParser()
package = parser.parse_xml(StringIO(valid_bundle))
self.assertEqual("example:bundle-1", package.id_)
def test_valid_bundle(self):
valid_bundle = """
<maecBundle:MAEC_Bundle
xmlns:maecBundle="http://maec.mitre.org/XMLSchema/maec-bundle-4"
id="example:bundle-1" schema_version="4.1">
</maecBundle:MAEC_Bundle>
"""
parser = EntityParser()
package = parser.parse_xml(StringIO(valid_bundle))
self.assertEqual("example:bundle-1", package.id_)
def test_valid_package(self):
valid_package = """
<maecPackage:MAEC_Package
xmlns:maecPackage="http://maec.mitre.org/XMLSchema/maec-package-2"
id="example:package-1" schema_version="2.1">
</maecPackage:MAEC_Package>
"""
parser = EntityParser()
package = parser.parse_xml(StringIO(valid_package))
self.assertEqual(Package, type(package))
self.assertEqual("example:package-1", package.id_)
def test_valid_package(self):
valid_package = """
<maecPackage:MAEC_Package
xmlns:maecPackage="http://maec.mitre.org/XMLSchema/maec-package-2"
id="example:package-1" schema_version="2.1">
</maecPackage:MAEC_Package>
"""
parser = EntityParser()
package = parser.parse_xml(StringIO(valid_package))
self.assertEqual(Package, type(package))
self.assertEqual("example:package-1", package.id_)
def test_missing_version(self):
missing_version = """
<maecPackage:MAEC_Package
xmlns:maecPackage="http://maec.mitre.org/XMLSchema/maec-package-2"
id="example:package-1">
</maecPackage:MAEC_Package>
"""
parser = EntityParser()
self.assertRaises(UnknownVersionError,
parser.parse_xml, StringIO(missing_version))
package = parser.parse_xml(StringIO(missing_version),
check_version=False)
self.assertEqual("example:package-1", package.id_)
def test_missing_version(self):
missing_version = """
<maecPackage:MAEC_Package
xmlns:maecPackage="http://maec.mitre.org/XMLSchema/maec-package-2"
id="example:package-1">
</maecPackage:MAEC_Package>
"""
parser = EntityParser()
self.assertRaises(UnknownVersionError, parser.parse_xml,
StringIO(missing_version))
package = parser.parse_xml(StringIO(missing_version),
check_version=False)
self.assertEqual("example:package-1", package.id_)
def test_wrong_version(self):
wrong_version = """
<maecPackage:MAEC_Package
xmlns:maecPackage="http://maec.mitre.org/XMLSchema/maec-package-2"
id="example:package-1" schema_version="10.1.8">
</maecPackage:MAEC_Package>
"""
parser = EntityParser()
self.assertRaises(UnsupportedVersionError,
parser.parse_xml, StringIO(wrong_version))
package = parser.parse_xml(StringIO(wrong_version),
check_version=False)
self.assertEqual("example:package-1", package.id_)
self.assertEqual("10.1.8", package.schema_version)
def test_wrong_version(self):
wrong_version = """
<maecPackage:MAEC_Package
xmlns:maecPackage="http://maec.mitre.org/XMLSchema/maec-package-2"
id="example:package-1" schema_version="10.1.8">
</maecPackage:MAEC_Package>
"""
parser = EntityParser()
self.assertRaises(UnsupportedVersionError, parser.parse_xml,
StringIO(wrong_version))
package = parser.parse_xml(StringIO(wrong_version),
check_version=False)
self.assertEqual("example:package-1", package.id_)
self.assertEqual("10.1.8", package.schema_version)
def test_wrong_root_element(self):
wrong_root = """
<maecPackage:NotAPackage
xmlns:maecPackage="http://maec.mitre.org/XMLSchema/maec-package-2"
id="example:package-1" schema_version="2.1">
</maecPackage:NotAPackage>
"""
parser = EntityParser()
self.assertRaises(UnsupportedRootElementError, parser.parse_xml,
StringIO(wrong_root))
# If there's not a valid root element, there's no way to check the
# version number.
self.assertRaises(UnsupportedVersionError,
parser.parse_xml,
StringIO(wrong_root),
check_root=False)
