forked from reingart/pyfpdf
-
Notifications
You must be signed in to change notification settings - Fork 231
/
test_embed_file.py
65 lines (50 loc) · 1.76 KB
/
test_embed_file.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from pathlib import Path
from fpdf import FPDF
from test.conftest import assert_pdf_equal, EPOCH
import pytest
HERE = Path(__file__).resolve().parent
EMBEDDED_FILE = HERE / "requirements.txt"
def test_embed_file_self(tmp_path):
pdf = FPDF()
pdf.add_page()
pdf.embed_file(EMBEDDED_FILE, modification_date=False)
assert_pdf_equal(pdf, HERE / "embed_file_self.pdf", tmp_path)
def test_embed_file_all_optionals(tmp_path):
pdf = FPDF()
pdf.add_page()
pdf.embed_file(
str(EMBEDDED_FILE),
desc="Source Python code",
creation_date=EPOCH,
modification_date=EPOCH,
compress=True,
checksum=True,
)
assert_pdf_equal(pdf, HERE / "embed_file_all_optionals.pdf", tmp_path)
def test_embed_file_from_bytes(tmp_path):
pdf = FPDF()
pdf.add_page()
pdf.embed_file(bytes=b"Hello world!", basename="hello_world.txt")
assert_pdf_equal(pdf, HERE / "embed_file_from_bytes.pdf", tmp_path)
def test_file_attachment_annotation(tmp_path):
pdf = FPDF()
pdf.add_page()
pdf.file_attachment_annotation(EMBEDDED_FILE, modification_date=False, x=50, y=50)
assert_pdf_equal(pdf, HERE / "file_attachment_annotation.pdf", tmp_path)
def test_embed_file_invalid_params():
pdf = FPDF()
pdf.add_page()
with pytest.raises(ValueError):
pdf.embed_file(__file__, bytes=b"...")
with pytest.raises(ValueError):
pdf.embed_file(__file__, basename="file.txt")
with pytest.raises(ValueError):
pdf.embed_file(bytes=b"...")
with pytest.raises(ValueError):
pdf.embed_file(basename="file.txt")
def test_embed_file_duplicate():
pdf = FPDF()
pdf.add_page()
pdf.embed_file(EMBEDDED_FILE)
with pytest.raises(ValueError):
pdf.embed_file(EMBEDDED_FILE)