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_cookie(self): obj = PdfDownloadMixin() obj.pdf_authenticate = False rf = RequestFactory() req = rf.get('/test/invoice?pdf') self.assertEqual(obj.get_pdf_cookie(req), None)