# -*- coding: utf-8 -*-
import unittest
from repoze.xmliter import decorator
from repoze.xmliter import serializer
from repoze.xmliter import utils
import lxml.html
import lxml.etree
import sys
if sys.version_info > (3,):
unicode = str
class TestIterator(unittest.TestCase):
def create_tree(self):
return lxml.html.fromstring(''.join(self.create_iterable()))
def create_iterable(self, preamble="", body=""):
return [preamble,
"",
"",
"My homepage",
"",
"",
"Hello, wörld!",
body,
""
"",
]
def test_html_serialization(self):
"""Test HTML serialization."""
@decorator.lazy(serializer=lxml.html.tostring)
def app(a, b, c=u""):
tree = self.create_tree()
tree.find('body').attrib['class'] = " ".join((a, b, c))
return tree
result = app("a", "b", c="c")
self.assertEqual(
lxml.html.tostring(result.tree),
b"".join(result))
# With Unicode encoding:
self.assertEqual(
lxml.html.tostring(result.tree, encoding='unicode'),
u"".join(result.serialize(encoding=unicode)))
def test_xml_serialization(self):
"""Test XML serialization."""
@decorator.lazy
def app(a, b, c=u""):
tree = self.create_tree()
tree.find('body').attrib['class'] = " ".join((a, b, c))
return tree
result = app("a", "b", c="c")
self.assertEqual(
lxml.etree.tostring(result.tree),
b"".join(result))
# With Unicode encoding:
self.assertEqual(
lxml.etree.tostring(result.tree, encoding='unicode'),
u"".join(result.serialize(encoding=unicode)))
def test_decorator_instancemethod(self):
class test(object):
@decorator.lazy
def process(self, tree):
return tree
def __call__(self, tree):
return self.process(tree)
result = test()(self.create_tree())
self.assertEqual(
lxml.etree.tostring(result.tree),
b"".join(result))
self.assertEqual(
lxml.etree.tostring(result.tree, encoding='unicode'),
u"".join(result.serialize(encoding=unicode)))
def test_getXMLSerializer(self):
t = utils.getXMLSerializer(self.create_iterable())
self.failUnless(isinstance(t, serializer.XMLSerializer))
t2 = utils.getXMLSerializer(t)
self.failUnless(t2 is t)
self.assertEqual(
b"My homepageHello, wörld!",
b"".join(t2))
self.assertEqual(
u"My homepageHello, wörld!",
u"".join(t2.serialize(encoding=unicode)))
def test_length(self):
t = utils.getXMLSerializer(self.create_iterable())
self.failUnless(len(t) == 1)
self.failUnless(len(list(t)) == 1)
def test_getHTMLSerializer(self):
t = utils.getHTMLSerializer(self.create_iterable(body='
'), pretty_print=True)
self.failUnless(isinstance(t, serializer.XMLSerializer))
t2 = utils.getXMLSerializer(t)
self.failUnless(t2 is t)
self.assertEqual(
b'\n\nMy homepage\nHello, wörld!\n\n\n',
b"".join(t2))
self.assertEqual(
u'\n\nMy homepage\nHello, wörld!\n\n\n',
u"".join(t2.serialize(encoding=unicode)))
def test_getHTMLSerializer_doctype_xhtml_serializes_to_xhtml(self):
t = utils.getHTMLSerializer(self.create_iterable(preamble='\n', body=''), pretty_print=True)
self.failUnless(isinstance(t, serializer.XMLSerializer))
t2 = utils.getXMLSerializer(t)
self.failUnless(t2 is t)
self.assertEqual(
b'\n\n \n \n My homepage\n \n Hello, wörld!\n\n',
b"".join(t2))
self.assertEqual(
u'\n\n \n \n My homepage\n \n Hello, wörld!\n\n',
u"".join(t2.serialize(encoding=unicode)))
def test_xsl(self):
t = utils.getHTMLSerializer(self.create_iterable(body='
'))
transform = lxml.etree.XSLT(lxml.etree.XML(b'''
'''))
t.tree = transform(t.tree)
self.failUnless('
' in str(t))
def test_replace_doctype(self):
t = utils.getHTMLSerializer(self.create_iterable(preamble='\n', body=''), pretty_print=True, doctype="")
self.failUnless(isinstance(t, serializer.XMLSerializer))
t2 = utils.getXMLSerializer(t)
self.failUnless(t2 is t)
self.assertEqual(
b'\n\n \n \n My homepage\n \n Hello, wörld!\n\n',
b"".join(t2))
self.assertEqual(
u'\n\n \n \n My homepage\n \n Hello, wörld!\n\n',
u"".join(t2.serialize(encoding=unicode)))
def test_replace_doctype_blank(self):
t = utils.getHTMLSerializer(self.create_iterable(preamble='\n', body=''), pretty_print=True, doctype="")
self.failUnless(isinstance(t, serializer.XMLSerializer))
t2 = utils.getXMLSerializer(t)
self.failUnless(t2 is t)
self.assertEqual(
b'\n \n \n My homepage\n \n Hello, wörld!\n\n',
b"".join(t2))
self.assertEqual(
u'\n \n \n My homepage\n \n Hello, wörld!\n\n',
u"".join(t2.serialize(encoding=unicode)))
def test_suite():
return unittest.defaultTestLoader.loadTestsFromName(__name__)