def test_temporary_file_cleanup(self):
"""
Checks that all temporary files are cleaned up after generating the pdf
"""
temp_files_before = len(listdir(tempfile.gettempdir()))
transform_to_pdf(self.response)
temp_files_after = len(listdir(tempfile.gettempdir()))
assert temp_files_before == temp_files_after
def test_filename_handing(self):
""" Checks the filename handling in the Content-Disposition header """
# Check a default name is provided
response = transform_to_pdf(self.response)
assert 'page.pdf' in response['Content-Disposition']
# Check we can provide our own filename
response = transform_to_pdf(self.response, '', 'testfile')
assert 'testfile.pdf' in response['Content-Disposition']
def test_filename_handing(self):
""" Checks the filename handling in the Content-Disposition header """
# Check a default name is provided
response = transform_to_pdf(self.response)
assert 'page.pdf' in response['Content-Disposition']
# Check we can provide our own filename
response = transform_to_pdf(self.response, '', 'testfile')
assert 'testfile.pdf' in response['Content-Disposition']
def test_character_encodings(self):
"""
Tests that the pdf generation can support various character sets for
different alphabets
"""
html = u"""<html>
<body>
<p>Can the pdf generate these?</p>
ISO 8859-5 - чуоюющяа (Cyrillic ru)
ISO 8859-6 - ﻗﻠﯿﻼ (Arabic ar)
ISO 8859-7 - ΔΦψ (Greek el)
ISO 8859-8 - úŵËÕŷîâöòÆṫ (Hebrew he)
ISO 8859-14 - Ṫêċï (Welsh and Gaelic gd/ga)
ISO 639 - 電电 (Chinese simplified & traditional zh)
ISO-2022-JP - ゴシック (Japanese ja)
</body>
</html>"""
response = HttpResponse(html)
response = transform_to_pdf(response)
# Check the generated pdf contains our contents
pdf_content = get_content_from_pdf(response.content)
assert u'чуоюющяа' in pdf_content
assert u'ﻗﻠﯿﻼ' in pdf_content
assert u'ΔΦψ' in pdf_content
assert u'úŵËÕŷîâöòÆṫ' in pdf_content
assert u'Ṫêċï' in pdf_content
assert u'電电' in pdf_content
assert u'ゴシック' in pdf_content
assert u'\x00' not in pdf_content
def test_pdf_generation(self):
""" Check we can create a simple pdf file """
# create our pdf
response = transform_to_pdf(self.response)
# Check the generated pdf contains our contents
pdf_content = get_content_from_pdf(response.content)
assert self.string_contents in pdf_content
def test_temporary_file_cleanup(self):
"""
Checks that all temporary files are cleaned up after generating the pdf
"""
temp_files_before = len(listdir(tempfile.gettempdir()))
response = transform_to_pdf(self.response)
temp_files_after = len(listdir(tempfile.gettempdir()))
assert temp_files_before == temp_files_after
def test_pdf_generation(self):
""" Check we can create a simple pdf file """
# create our pdf
response = transform_to_pdf(self.response)
# Check the generated pdf contains our contents
pdf_content = get_content_from_pdf(response.content)
assert self.string_contents in pdf_content
