def test_xpath_element(self):
doc = xml.etree.ElementTree.Element('root')
div = xml.etree.ElementTree.SubElement(doc, 'div')
p = xml.etree.ElementTree.SubElement(div, 'p')
p.text = 'Foo'
self.assertEqual(xpath_element(doc, 'div/p'), p)
self.assertEqual(xpath_element(doc, 'div/bar', default='default'), 'default')
self.assertTrue(xpath_element(doc, 'div/bar') is None)
self.assertRaises(ExtractorError, xpath_element, doc, 'div/bar', fatal=True)
def test_xpath_element(self):
doc = xml.etree.ElementTree.Element('root')
div = xml.etree.ElementTree.SubElement(doc, 'div')
p = xml.etree.ElementTree.SubElement(div, 'p')
p.text = 'Foo'
self.assertEqual(xpath_element(doc, 'div/p'), p)
self.assertEqual(xpath_element(doc, ['div/p']), p)
self.assertEqual(xpath_element(doc, ['div/bar', 'div/p']), p)
self.assertEqual(xpath_element(doc, 'div/bar', default='default'),
'default')
self.assertEqual(xpath_element(doc, ['div/bar'], default='default'),
'default')
self.assertTrue(xpath_element(doc, 'div/bar') is None)
self.assertTrue(xpath_element(doc, ['div/bar']) is None)
self.assertTrue(xpath_element(doc, ['div/bar'], 'div/baz') is None)
self.assertRaises(ExtractorError,
xpath_element,
doc,
'div/bar',
fatal=True)
self.assertRaises(ExtractorError,
xpath_element,
doc, ['div/bar'],
fatal=True)
self.assertRaises(ExtractorError,
xpath_element,
doc, ['div/bar', 'div/baz'],
fatal=True)
def test_xpath_element(self):
doc = xml.etree.ElementTree.Element("root")
div = xml.etree.ElementTree.SubElement(doc, "div")
p = xml.etree.ElementTree.SubElement(div, "p")
p.text = "Foo"
self.assertEqual(xpath_element(doc, "div/p"), p)
self.assertEqual(xpath_element(doc, ["div/p"]), p)
self.assertEqual(xpath_element(doc, ["div/bar", "div/p"]), p)
self.assertEqual(xpath_element(doc, "div/bar", default="default"), "default")
self.assertEqual(xpath_element(doc, ["div/bar"], default="default"), "default")
self.assertTrue(xpath_element(doc, "div/bar") is None)
self.assertTrue(xpath_element(doc, ["div/bar"]) is None)
self.assertTrue(xpath_element(doc, ["div/bar"], "div/baz") is None)
self.assertRaises(ExtractorError, xpath_element, doc, "div/bar", fatal=True)
self.assertRaises(ExtractorError, xpath_element, doc, ["div/bar"], fatal=True)
self.assertRaises(ExtractorError, xpath_element, doc, ["div/bar", "div/baz"], fatal=True)
