import os
import unittest
from xhtml2pdf.parser import pisaParser
from xhtml2pdf.context import pisaContext
_data = b"""
TITLE
BODY
"""
class TestCase(unittest.TestCase):
def testParser(self):
c = pisaContext(".")
r = pisaParser(_data, c)
self.assertEqual(c, r)
self.assertEqual(r.err, 0)
self.assertEqual(r.warn, 0)
def test_getFile(self):
c = pisaContext(".")
r = pisaParser(_data, c)
self.assertEqual(c.getFile(None), None)
self.assertEqual(r.err, 0)
self.assertEqual(r.warn, 0)
def test_height_as_list(self):
"""Asserts attributes like 'height: 10px !important" are parsed"""
c = pisaContext(".")
data = b"test
"
r = pisaParser(data, c)
self.assertEqual(c, r)
self.assertEqual(r.err, 0)
self.assertEqual(r.warn, 0)
def test_image_os_path(self):
c = pisaContext(".")
tests_folder = os.path.dirname(os.path.realpath(__file__))
img_path = os.path.join(tests_folder, 'samples', 'img', 'denker.png')
data = '
'.format(img_path).encode('utf-8')
r = pisaParser(data, c)
self.assertEqual(c, r)
self.assertEqual(r.err, 0)
self.assertEqual(r.warn, 0)
def test_image_base64(self):
c = pisaContext(".")
data = b'
'
r = pisaParser(data, c)
self.assertEqual(r.warn, 0)
def buildTestSuite():
return unittest.defaultTestLoader.loadTestsFromName(__name__)
def main():
buildTestSuite()
unittest.main()
if __name__ == "__main__":
main()