def test_xml_resource_defuse(self):
resource = XMLResource(self.vh_xml_file, defuse='never')
self.assertEqual(resource.defuse, 'never')
self.assertEqual(resource.parse, etree_parse)
self.assertEqual(resource.iterparse, etree_iterparse)
self.assertEqual(resource.fromstring, etree_fromstring)
resource.defuse = 'always'
self.assertEqual(resource.parse, defused_etree.parse)
self.assertEqual(resource.iterparse, defused_etree.iterparse)
self.assertEqual(resource.fromstring, defused_etree.fromstring)
resource.defuse = 'remote'
self.assertEqual(resource.parse, etree_parse)
self.assertEqual(resource.iterparse, etree_iterparse)
self.assertEqual(resource.fromstring, etree_fromstring)
resource._url = 'http://localhost'
self.assertEqual(resource.parse, defused_etree.parse)
self.assertEqual(resource.iterparse, defused_etree.iterparse)
self.assertEqual(resource.fromstring, defused_etree.fromstring)
self.assertRaises(ValueError,
XMLResource,
self.vh_xml_file,
defuse='all')
self.assertRaises(ValueError,
XMLResource,
self.vh_xml_file,
defuse=None)
def test_xml_resource_fromstring(self):
resource = XMLResource(self.vh_xml_file)
self.assertEqual(resource.defuse, 'remote')
self.assertEqual(resource.fromstring('<root/>').tag, 'root')
resource.defuse = 'always'
self.assertEqual(resource.fromstring('<root/>').tag, 'root')
def test_xml_resource_iterparse(self):
resource = XMLResource(self.vh_xml_file)
self.assertEqual(resource.defuse, 'remote')
for _, elem in resource.iterparse(self.col_xml_file, events=('end', )):
self.assertTrue(is_etree_element(elem))
resource.defuse = 'always'
for _, elem in resource.iterparse(self.col_xml_file, events=('end', )):
self.assertTrue(is_etree_element(elem))
def test_xml_resource_parse(self):
resource = XMLResource(self.vh_xml_file)
self.assertEqual(resource.defuse, 'remote')
xml_document = resource.parse(self.col_xml_file)
self.assertTrue(is_etree_element(xml_document.getroot()))
resource.defuse = 'always'
xml_document = resource.parse(self.col_xml_file)
self.assertTrue(is_etree_element(xml_document.getroot()))
