Skip to content

Commit 3a9a2ad

Browse files
mezkaPetrDlouhy
authored andcommitted
fix: PdfFileReader deprecated for PdfReader, and test was failing because of wrongly encoded string in assertTrue
1 parent 936ac50 commit 3a9a2ad

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

tests/test_pdf.py

+10-10
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from InvoiceGenerator.api import Client, Creator, Invoice, Item, Provider
88
from InvoiceGenerator.pdf import CorrectingInvoice, ProformaInvoice, SimpleInvoice
99

10-
from PyPDF2 import PdfFileReader
10+
from PyPDF2 import PdfReader
1111

1212

1313
class TestBaseInvoice(unittest.TestCase):
@@ -77,10 +77,10 @@ def test_generate(self):
7777
pdf = CorrectingInvoice(invoice)
7878
pdf.gen(tmp_file1.name)
7979

80-
pdf = PdfFileReader(tmp_file1)
81-
pdf_string = pdf.pages[1].extractText()
82-
self.assertTrue(u"Celkem s DPH: 32 255,- K…" in pdf_string)
83-
self.assertTrue(u"Vytvo⁄il: blah" in pdf_string)
80+
pdf = PdfReader(tmp_file1)
81+
pdf_string = pdf.pages[1].extract_text()
82+
self.assertTrue(u"Celkem s DPH: 32 255,- " in pdf_string)
83+
self.assertTrue(u"Vytvořil: blah" in pdf_string)
8484

8585
def test_generate_proforma(self):
8686
provider = Provider('Pupik')
@@ -136,9 +136,9 @@ def test_generate_proforma(self):
136136
pdf = ProformaInvoice(invoice)
137137
pdf.gen(tmp_file.name, True)
138138

139-
pdf = PdfFileReader(tmp_file)
140-
pdf_string = pdf.pages[1].extractText()
141-
self.assertTrue(u"Celkem s DPH: 32 255,- K…" in pdf_string)
139+
pdf = PdfReader(tmp_file)
140+
pdf_string = pdf.pages[1].extract_text()
141+
self.assertTrue(u"Celkem s DPH: 32 255,- " in pdf_string)
142142

143143
def test_generate_with_vat(self):
144144
os.environ["INVOICE_LANG"] = "en"
@@ -157,8 +157,8 @@ def test_generate_with_vat(self):
157157
pdf = SimpleInvoice(invoice)
158158
pdf.gen(tmp_file.name)
159159

160-
pdf = PdfFileReader(tmp_file)
161-
pdf_string = pdf.pages[0].extractText()
160+
pdf = PdfReader(tmp_file)
161+
pdf_string = pdf.pages[0].extract_text()
162162
self.assertTrue(u"$3,000.00" in pdf_string)
163163
self.assertTrue(u"Total with tax: $30,150.00" in pdf_string)
164164
self.assertTrue(u"Creator: blah" in pdf_string)

0 commit comments

Comments
 (0)