47 lines
1.0 KiB
Python
47 lines
1.0 KiB
Python
|
import json
|
||
|
|
import os
|
||
|
|
import unittest
|
||
|
|
from io import BytesIO
|
||
|
|
|
||
|
|
from pdfjinja import PdfJinja
|
||
|
|
|
||
|
|
|
||
|
|
class insertTemplateTestCase(unittest.TestCase):
|
||
|
|
|
||
|
|
datadir = os.path.join(os.path.dirname(__file__))
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
pdffile = os.path.join(self.datadir, "sample.pdf")
|
||
|
|
self.data = {
|
||
|
|
'firstName': 'Renan',
|
||
|
|
'lastName': 'Husson',
|
||
|
|
'address': {
|
||
|
|
'street': '24 rue de la pommes',
|
||
|
|
'apt': 'C317',
|
||
|
|
'city': 'TOULOUSE',
|
||
|
|
'zipcode': 31000
|
||
|
|
},
|
||
|
|
'universite':'Jean Jaures',
|
||
|
|
'spirit': 'Panda',
|
||
|
|
'evil': True,
|
||
|
|
'language': {
|
||
|
|
'french': True,
|
||
|
|
'esperento': True
|
||
|
|
}
|
||
|
|
}
|
||
|
|
self.pdfjinja = PdfJinja(pdffile)
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
del self.data
|
||
|
|
del self.pdfjinja
|
||
|
|
|
||
|
|
def test_render(self):
|
||
|
|
output = self.pdfjinja(self.data)
|
||
|
|
outfile = BytesIO()
|
||
|
|
output.write(outfile)
|
||
|
|
outfile.seek(0)
|
||
|
|
self.assertTrue(len(outfile.read()) > 0, "Output PDF is not empty.")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|