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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
from pathlib import Path
from test.conftest import assert_pdf_equal, EPOCH, LOREM_IPSUM
from fpdf import FPDF
from fpdf.recorder import FPDFRecorder
HERE = Path(__file__).resolve().parent
def init_pdf():
pdf = FPDF()
pdf.set_creation_date(EPOCH)
pdf.set_font("helvetica", size=24)
pdf.add_page()
pdf.cell(w=pdf.epw, h=10, text="Hello fpdf2!", align="C")
return pdf
def test_recorder_rewind_ok(tmp_path):
pdf = init_pdf()
recorder = FPDFRecorder(pdf)
expected = recorder.output() # close the document as a side-effect
recorder.rewind() # in order to un-close the document
recorder.add_page()
recorder.cell(w=recorder.epw, h=10, text="Hello again!", align="C")
recorder.rewind()
assert_pdf_equal(recorder, expected, tmp_path)
def test_recorder_rewind_twice_ok(tmp_path):
pdf = init_pdf()
recorder = FPDFRecorder(pdf)
expected = recorder.output() # close the document as a side-effect
recorder.rewind() # in order to un-close the document
recorder.add_page()
recorder.cell(w=recorder.epw, h=10, text="Hello again!", align="C")
recorder.rewind()
assert_pdf_equal(recorder, expected, tmp_path)
def test_recorder_replay_ok(tmp_path):
recorder = FPDFRecorder(init_pdf())
recorder.add_page()
recorder.cell(w=recorder.epw, h=10, text="Hello again!", align="C")
expected = recorder.output()
recorder.rewind()
recorder.replay()
assert_pdf_equal(recorder, expected, tmp_path)
def test_recorder_override_accept_page_break_ok():
recorder = FPDFRecorder(init_pdf(), accept_page_break=False)
assert recorder.accept_page_break is False
def test_recorder_preserve_pages_count():
pdf = init_pdf()
pdf.set_y(250)
assert pdf.pages_count == 1
with pdf.offset_rendering() as recorder:
pdf.multi_cell(text=LOREM_IPSUM, w=pdf.epw)
assert pdf.pages_count == 2
assert recorder.page_break_triggered
assert pdf.pages_count == 1
def test_recorder_with_ttf_font(tmp_path):
pdf = init_pdf()
pdf.add_font(fname=str(HERE / "fonts" / "Roboto-Regular.ttf"))
pdf.set_font("Roboto-Regular", size=64)
pdf.add_page()
pdf.cell(text="Hello!", align="C")
recorder = FPDFRecorder(pdf)
expected = recorder.output() # close the document as a side-effect
recorder.rewind() # in order to un-close the document
recorder.add_page()
recorder.cell(w=recorder.epw, h=10, text="Hello again!", align="C")
recorder.rewind()
assert_pdf_equal(recorder, expected, tmp_path)
|
