def test_xml_document_decode_with_xsi_type(self):
xml_data = '<root xmlns:xsi="{}" xmlns:xs="{}" ' \
'xsi:type="xs:integer">10</root>'.format(XSI_NAMESPACE, XSD_NAMESPACE)
xml_document = XmlDocument(xml_data)
self.assertEqual(
xml_document.decode(), {
'@xmlns:xsi': 'http://www.w3.org/2001/XMLSchema-instance',
'@xmlns:xs': 'http://www.w3.org/2001/XMLSchema',
'@xsi:type': 'xs:integer',
'$': 10
})
def test_xml_document_init_without_schema(self):
with self.assertRaises(ValueError) as ctx:
XmlDocument('<empty/>')
self.assertIn('no schema can be retrieved for the XML resource',
str(ctx.exception))
xml_document = XmlDocument('<empty/>', validation='skip')
self.assertIsNone(xml_document.schema)
self.assertIsInstance(xml_document._fallback_schema, XMLSchema10)
self.assertEqual(xml_document._fallback_schema.target_namespace, '')
xml_document = XmlDocument(
'<tns:empty xmlns:tns="http://example.com/ns" />',
validation='skip')
self.assertIsNone(xml_document.schema)
self.assertIsInstance(xml_document._fallback_schema, XMLSchema10)
self.assertEqual(xml_document._fallback_schema.target_namespace,
xml_document.namespace)
def test_get_context_with_schema(self):
source, schema = get_context(self.col_xml_file)
self.assertIsInstance(source, XMLResource)
self.assertIsInstance(schema, XMLSchema10)
source, schema = get_context(self.col_xml_file, self.col_xsd_file)
self.assertIsInstance(source, XMLResource)
self.assertIsInstance(schema, XMLSchema10)
col_schema = XMLSchema10(self.col_xsd_file)
source, schema = get_context(self.col_xml_file, col_schema)
self.assertIsInstance(source, XMLResource)
self.assertIs(schema, col_schema)
source, schema = get_context(self.vh_xml_file, cls=XMLSchema10)
self.assertIsInstance(source, XMLResource)
self.assertIsInstance(schema, XMLSchema10)
source, schema = get_context(self.col_xml_file, cls=XMLSchema11)
self.assertIsInstance(source, XMLResource)
self.assertIsInstance(schema, XMLSchema11)
source, schema = get_context(XMLResource(self.vh_xml_file))
self.assertIsInstance(source, XMLResource)
self.assertIsInstance(schema, XMLSchema10)
xml_document = XmlDocument(self.vh_xml_file)
source, schema = get_context(xml_document)
self.assertIsInstance(source, XMLResource)
self.assertIsInstance(schema, XMLSchema10)
self.assertIs(xml_document.schema, schema)
# Issue #145
with open(self.vh_xml_file) as f:
source, schema = get_context(f, schema=self.vh_xsd_file)
self.assertIsInstance(source, XMLResource)
self.assertIsInstance(schema, XMLSchema10)
with open(self.vh_xml_file) as f:
source, schema = get_context(XMLResource(f),
schema=self.vh_xsd_file)
self.assertIsInstance(source, XMLResource)
self.assertIsInstance(schema, XMLSchema10)
with open(self.vh_xml_file) as f:
source, schema = get_context(f, base_url=self.vh_dir)
self.assertIsInstance(source, XMLResource)
self.assertIsInstance(schema, XMLSchema10)
def test_xml_document_with_lxml(self):
vh_tree = lxml_etree.parse(self.vh_xml_file)
xml_document = XmlDocument(vh_tree, base_url=self.vh_dir)
self.assertIs(xml_document.get_etree_document(), vh_tree)
xml_document = XmlDocument(vh_tree.getroot(), base_url=self.vh_dir)
etree_document = xml_document.get_etree_document()
self.assertIsNot(etree_document, vh_tree)
self.assertTrue(is_etree_document(etree_document))
self.assertTrue(hasattr(etree_document, 'xpath'))
self.assertTrue(hasattr(etree_document, 'xslt'))
def test_xml_document_init_with_schema(self):
xml_document = XmlDocument(self.vh_xml_file)
self.assertEqual(os.path.basename(xml_document.url), 'vehicles.xml')
self.assertEqual(xml_document.errors, ())
self.assertIsInstance(xml_document.schema, XMLSchema10)
xml_document = XmlDocument(self.vh_xml_file, cls=XMLSchema11)
self.assertIsInstance(xml_document.schema, XMLSchema11)
xml_document = XmlDocument(self.vh_xml_file, self.vh_xsd_file)
self.assertIsInstance(xml_document.schema, XMLSchema10)
vh_schema = XMLSchema10(self.vh_xsd_file)
xml_document = XmlDocument(self.vh_xml_file, vh_schema)
self.assertIsInstance(xml_document.schema, XMLSchema10)
with self.assertRaises(XMLSchemaValidationError) as ctx:
XmlDocument(self.vh_xml_file, self.col_xsd_file)
self.assertIn('is not an element of the schema', str(ctx.exception))
xml_document = XmlDocument(self.col_xml_file)
self.assertEqual(os.path.basename(xml_document.url), 'collection.xml')
self.assertIsInstance(xml_document.schema, XMLSchema10)
xml_file = casepath('examples/collection/collection-1_error.xml')
with self.assertRaises(XMLSchemaValidationError) as ctx:
XmlDocument(xml_file)
self.assertIn('invalid literal for int() with base 10',
str(ctx.exception))
xml_document = XmlDocument(xml_file, validation='lax')
self.assertEqual(os.path.basename(xml_document.url),
'collection-1_error.xml')
self.assertIsInstance(xml_document.schema, XMLSchema10)
self.assertTrue(len(xml_document.errors), 1)
with self.assertRaises(ValueError) as ctx:
XmlDocument(xml_file, validation='foo')
self.assertEqual(str(ctx.exception), "'foo': not a validation mode")
def test_xml_document_decode_without_schema(self):
xml_document = XmlDocument('<x:root xmlns:x="ns" />',
validation='skip')
self.assertIsNone(xml_document.decode())
xml_document = XmlDocument(
'<x:root xmlns:x="ns" a="true"><b1>10</b1><b2/></x:root>',
validation='skip')
self.assertEqual(xml_document.decode(), {
'@a': 'true',
'b1': ['10'],
'b2': [None]
})
def test_xml_document_tostring(self):
xml_document = XmlDocument(self.vh_xml_file)
self.assertTrue(xml_document.tostring().startswith('<vh:vehicles'))
with self.assertRaises(XMLResourceError):
XmlDocument(self.vh_xml_file, lazy=True).tostring()
def test_xml_document_etree_interface(self):
xml_document = XmlDocument(self.vh_xml_file)
self.assertIs(xml_document.getroot(), xml_document._root)
self.assertTrue(is_etree_element(xml_document.getroot()))
self.assertTrue(is_etree_document(xml_document.get_etree_document()))
xml_document = XmlDocument(self.vh_xml_file, lazy=1)
with self.assertRaises(XMLResourceError) as ctx:
xml_document.get_etree_document()
self.assertIn('cannot create an ElementTree from a lazy resource',
str(ctx.exception))
vh_tree = ElementTree.parse(self.vh_xml_file)
xml_document = XmlDocument(vh_tree, base_url=self.vh_dir)
self.assertIs(xml_document.source, vh_tree)
self.assertIs(xml_document.get_etree_document(), vh_tree)
def test_xml_document_write(self):
with tempfile.TemporaryDirectory() as dirname:
col_file_path = pathlib.Path(dirname).joinpath('collection.xml')
xml_document = XmlDocument(self.col_xml_file)
xml_document.write(col_file_path.open(mode='wb'))
schema = XMLSchema10(self.col_xsd_file)
xml_document = XmlDocument(str(col_file_path), schema=schema)
self.assertEqual(xml_document.root.tag,
'{http://example.com/ns/collection}collection')
self.assertIs(xml_document.schema, schema)
col_file_path.unlink()
xml_document.write(str(col_file_path))
xml_document = XmlDocument(str(col_file_path), schema=schema)
self.assertIs(xml_document.schema, schema)
col_file_path.unlink()
xml_document.write(str(col_file_path), encoding='unicode')
xml_document = XmlDocument(str(col_file_path), schema=schema)
self.assertIs(xml_document.schema, schema)
col_file_path.unlink()
xml_document.write(
str(col_file_path),
default_namespace="http://example.com/ns/collection")
xml_document = XmlDocument(str(col_file_path), schema=schema)
self.assertIs(xml_document.schema, schema)
if lxml_etree is not None:
col_file_path.unlink()
col_etree_document = lxml_etree.parse(self.col_xml_file)
xml_document = XmlDocument(col_etree_document,
base_url=self.col_dir)
xml_document.write(
str(col_file_path),
default_namespace="http://example.com/ns/collection")
xml_document = XmlDocument(str(col_file_path), schema=schema)
self.assertIs(xml_document.schema, schema)
col_file_path.unlink()
xml_document = XmlDocument(self.col_xml_file, lazy=True)
with self.assertRaises(XMLResourceError) as ctx:
xml_document.write(str(col_file_path))
self.assertEqual(str(ctx.exception),
"cannot serialize a lazy XML document")
def test_xml_document_to_json(self):
xml_document = XmlDocument(self.col_xml_file, lazy=True)
json_data = xml_document.to_json()
self.assertIsInstance(json_data, str)
self.assertIn('"@xmlns:col"', json_data)
self.assertIn(r'"name": "Joan Mir\u00f3"', json_data)
self.assertEqual(xml_document.to_json(validation='lax')[0], json_data)
self.assertEqual(xml_document.to_json(namespaces=None), json_data)
with self.assertRaises(TypeError) as ctx:
xml_document.to_json(decimal_type=Decimal)
self.assertIn("is not JSON serializable", str(ctx.exception))
fp = io.StringIO()
xml_document.to_json(fp=fp)
self.assertEqual(fp.getvalue(), json_data)
fp.close()
fp = io.StringIO()
self.assertEqual(xml_document.to_json(fp=fp, validation='lax'), ())
self.assertEqual(fp.getvalue(), json_data)
fp.close()
col_1_error_xml_file = casepath(
'examples/collection/collection-1_error.xml')
xml_document = XmlDocument(col_1_error_xml_file, validation='lax')
json_data, errors = xml_document.to_json()
self.assertEqual(len(errors), 1)
self.assertIsInstance(errors[0], XMLSchemaDecodeError)
self.assertIn('"position": null', json_data)
xml_document = XmlDocument(col_1_error_xml_file,
validation='lax',
lazy=True)
json_data, errors = xml_document.to_json(
json_options={'default': lambda x: None})
self.assertEqual(len(errors), 0)
self.assertIn('"object": [null, null]', json_data)
def test_xml_document_decode_with_schema(self):
xml_document = XmlDocument(self.vh_xml_file)
vh_schema = XMLSchema10(self.vh_xsd_file)
self.assertEqual(xml_document.decode(),
vh_schema.decode(self.vh_xml_file))
namespaces = {'vh': 'http://example.com/ns'}
self.assertEqual(
xml_document.decode(namespaces=namespaces),
vh_schema.decode(self.vh_xml_file, namespaces=namespaces))
self.assertNotEqual(xml_document.decode(namespaces=namespaces),
vh_schema.decode(self.vh_xml_file))
xml_file = casepath('examples/collection/collection-1_error.xml')
xml_document = XmlDocument(xml_file, validation='lax')
col_schema = XMLSchema10(self.col_xsd_file)
self.assertEqual(xml_document.decode(),
col_schema.decode(xml_file, validation='lax')[0])
xml_document = XmlDocument(xml_file, validation='skip')
self.assertEqual(xml_document.decode(),
col_schema.decode(xml_file, validation='skip'))
self.assertEqual(xml_document.decode(validation='lax'),
col_schema.decode(xml_file, validation='lax')[0])
def test_xml_document_parse(self):
xml_document = XmlDocument(self.vh_xml_file)
self.assertEqual(os.path.basename(xml_document.url), 'vehicles.xml')
self.assertFalse(xml_document.is_lazy())
xml_file = casepath('examples/vehicles/vehicles-1_error.xml')
with self.assertRaises(XMLSchemaValidationError):
xml_document.parse(xml_file)
xml_document.parse(self.vh_xml_file, lazy=True)
self.assertEqual(os.path.basename(xml_document.url), 'vehicles.xml')
self.assertTrue(xml_document.is_lazy())
xml_document = XmlDocument(self.vh_xml_file, validation='lax')
xml_document.parse(xml_file)
self.assertEqual(len(xml_document.errors), 1)