718b5b8bee
* refactor: Private images rendering in PDFs Private images currently render fine if PDF is generated during a request as we pass the cookiejar to WKHTML. Background jobs however fail completeley because they can't retrieve private images without cookiejar. This PR converts all image types to base64 encoded sources in HTML itself, so wkhtmltopdf doesn't have to a fire a request. * test: private images in pdf
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
# Copyright (c) 2018, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
import io
|
|
|
|
from pypdf import PdfReader
|
|
|
|
import frappe
|
|
import frappe.utils.pdf as pdfgen
|
|
from frappe.core.doctype.file.test_file import make_test_image_file
|
|
from frappe.tests.utils import FrappeTestCase
|
|
|
|
|
|
class TestPdf(FrappeTestCase):
|
|
@property
|
|
def html(self):
|
|
return """<style>
|
|
.print-format {
|
|
margin-top: 0mm;
|
|
margin-left: 10mm;
|
|
margin-right: 0mm;
|
|
}
|
|
</style>
|
|
<p>This is a test html snippet</p>
|
|
<div class="more-info">
|
|
<a href="http://test.com">Test link 1</a>
|
|
<a href="/about">Test link 2</a>
|
|
<a href="login">Test link 3</a>
|
|
<img src="/assets/frappe/test.jpg">
|
|
</div>
|
|
<div style="background-image: url('/assets/frappe/bg.jpg')">
|
|
Please mail us at <a href="mailto:test@example.com">email</a>
|
|
</div>"""
|
|
|
|
def runTest(self):
|
|
self.test_read_options_from_html()
|
|
|
|
def test_read_options_from_html(self):
|
|
_, html_options = pdfgen.read_options_from_html(self.html)
|
|
self.assertTrue(html_options["margin-top"] == "0")
|
|
self.assertTrue(html_options["margin-left"] == "10")
|
|
self.assertTrue(html_options["margin-right"] == "0")
|
|
|
|
def test_pdf_encryption(self):
|
|
password = "qwe"
|
|
pdf = pdfgen.get_pdf(self.html, options={"password": password})
|
|
reader = PdfReader(io.BytesIO(pdf))
|
|
self.assertTrue(reader.is_encrypted)
|
|
self.assertTrue(reader.decrypt(password))
|
|
|
|
def test_pdf_generation_as_a_user(self):
|
|
frappe.set_user("Administrator")
|
|
pdf = pdfgen.get_pdf(self.html)
|
|
self.assertTrue(pdf)
|
|
|
|
def test_private_images_in_pdf(self):
|
|
with make_test_image_file(private=True) as file:
|
|
html = f""" <div>
|
|
<img src="{file.file_url}" class='responsive'>
|
|
<img src="{file.unique_url}" class='responsive'>
|
|
</div>
|
|
"""
|
|
|
|
pdf = pdfgen.get_pdf(html)
|
|
|
|
# If image was actually retrieved then size will be in few kbs, else bytes.
|
|
self.assertGreaterEqual(len(pdf), 10_000)
|