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()))