import sys
import unittest
from lxml import etree
from lxml.tests.common_imports import HelperTestCase
from lxml.doctestcompare import LXMLOutputChecker, PARSE_HTML, PARSE_XML
class DummyInput:
def __init__(self, **kw):
for name, value in kw.items():
setattr(self, name, value)
def indent(elem, level=0):
i = "\n" + level*" "
if len(elem):
if not elem.text or not elem.text.strip():
elem.text = i + " "
if not elem.tail or not elem.tail.strip():
elem.tail = i
for elem in elem:
indent(elem, level+1)
if not elem.tail or not elem.tail.strip():
elem.tail = i
else:
if level and (not elem.tail or not elem.tail.strip()):
elem.tail = i
class DoctestCompareTest(HelperTestCase):
_checker = LXMLOutputChecker()
def compare(self, want, got, html=False):
if html:
options = PARSE_HTML
else:
options = PARSE_XML
parse = self._checker.get_parser(want, got, options)
want_doc = parse(want)
got_doc = parse(got)
return self._checker.collect_diff(
want_doc, got_doc, html, indent=0).lstrip()
def assert_diff(self, want, got, diff, html=False):
self.assertEqual(self.compare(want, got, html), diff)
def assert_nodiff(self, want, got, html=False):
root = etree.fromstring(want)
root.tail = '\n'
indent(root)
diff = etree.tostring(
root, encoding='unicode', method=html and 'html' or 'xml')
self.assert_diff(want, got, diff, html=html)
def test_equal_input(self):
self.assert_nodiff(
'Expected
',
'Expected
')
def test_differing_tags(self):
self.assert_diff(
'Expected
',
'Expected',
'Expected
\n')
def test_tags_upper_lower_case(self):
self.assert_diff(
'Expected
',
'Expected
',
'Expected
\n')
def test_tags_upper_lower_case_html(self):
self.assert_nodiff(
'Expected
',
'Expected
',
html=True)
def test_differing_attributes(self):
self.assert_diff(
'Expected
',
'Actual
',
'Expected (got: Actual)
\n')
def test_extra_children(self):
# https://bugs.launchpad.net/lxml/+bug/1238503
self.assert_diff(
'One
',
'OneTwoThree
',
'\n'
' One\n'
' +Two\n'
' +Three\n'
'
\n')
def test_missing_children(self):
self.assert_diff(
'OneTwoThree
',
'One
',
'\n'
' One\n'
' -Two\n'
' -Three\n'
'
\n')
def test_extra_attributes(self):
self.assert_diff(
'Text
',
'Text
',
'\n'
' Text\n'
'
\n')
def test_missing_attributes(self):
self.assert_diff(
'Text
',
'Text
',
'\n'
' Text\n'
'
\n')
def test_suite():
suite = unittest.TestSuite()
if sys.version_info >= (2,4):
suite.addTests([unittest.makeSuite(DoctestCompareTest)])
return suite
if __name__ == '__main__':
unittest.main()