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_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