def test_pdf_generation(self): filename = '/tmp/test.pdf' url = 'http://www.google.com' save_pdf(url, filename=filename) self.assertTrue(os.path.exists(filename)) self.assertTrue(os.path.getsize(filename) > 0) os.remove(filename) obj = PdfDownloadMixin() obj.pdf_authenticate = False data = obj._return_pdf(url) # Check if PDF contains data self.assertTrue(data.len > 0)
def test_get_pdf_url(self): obj = PdfDownloadMixin() rf = RequestFactory() req = rf.get('/test/invoice?pdf') self.assertEqual(obj.get_pdf_location(), '/tmp/download.pdf') obj.get_pdf_url(req)
def test_get_pdf_cookie(self): obj = PdfDownloadMixin() obj.pdf_authenticate = False rf = RequestFactory() req = rf.get('/test/invoice?pdf') self.assertEqual(obj.get_pdf_cookie(req), None)
def test_methods(self): obj = PdfDownloadMixin() self.assertTrue(obj.pdf_authenticate) self.assertTrue(obj.pdf_remove_file) self.assertEqual(obj.get_pdf_attachment_name(), 'download.pdf') self.assertEqual(obj.get_pdf_location(), '/tmp/download.pdf')