TST: Regression test for #327

Credits to Denis Osipov: #359 (comment) Co-authored-by: Denis Osipov <[email protected]>
py-pdf · Apr 16, 2022 · d58a849 · d58a849
1 parent bd7500d
commit d58a849
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 4 deletions.
diff --git a/Resources/encrypted-file.pdf b/Resources/encrypted-file.pdf
diff --git a/Tests/test_page.py b/Tests/test_page.py
@@ -30,7 +30,7 @@
         "libreoffice-writer-password",
         "imagemagick-images",
         "imagemagick-lzw",
-        "reportlab-inline-image"
+        "reportlab-inline-image",
     ],
 )
 def test_page_operations(pdf_path, password):

diff --git a/Tests/test_pagerange.py b/Tests/test_pagerange.py
@@ -62,6 +62,7 @@ def test_parse_filename_page_ranges_err():
 
 def test_page_range_help():
     from PyPDF2.pagerange import PAGE_RANGE_HELP
+
     assert len(PAGE_RANGE_HELP) > 20
     assert "0:3" in PAGE_RANGE_HELP
     assert PAGE_RANGE_HELP.endswith("\n")
diff --git a/Tests/test_reader.py b/Tests/test_reader.py
@@ -51,8 +51,8 @@ def test_read_metadata(pdf_path, expected):
         docinfo = reader.getDocumentInfo()
         metadict = dict(docinfo)
         assert metadict == expected
-        if '/Title' in metadict:
-            assert metadict['/Title'] == docinfo.title
+        if "/Title" in metadict:
+            assert metadict["/Title"] == docinfo.title
 
 
 @pytest.mark.parametrize(
@@ -120,7 +120,7 @@ def test_get_outlines(src, outline_elements):
     ],
 )
 def test_get_images(src, nb_images):
-    src =os.path.join(RESOURCE_ROOT, src)
+    src = os.path.join(RESOURCE_ROOT, src)
     reader = PdfFileReader(src)
 
     with pytest.raises(TypeError):
@@ -207,3 +207,19 @@ def test_issue297():
     path = os.path.join(RESOURCE_ROOT, "issue-297.pdf")
     reader = PdfFileReader(path, "rb")
     reader.getPage(0)
+
+
+def test_get_page_of_encrypted_file():
+    """
+    Check if we can read a page of an encrypted file.
+
+    This is a regression test for issue 327:
+    IndexError for getPage() of decrypted file
+    """
+    path = os.path.join(RESOURCE_ROOT, "encrypted-file.pdf")
+    reader = PdfFileReader(path)
+
+    # Password is correct:)
+    reader.decrypt("test")
+
+    reader.getPage(0)