Skip to content

Commit ba2a51a

Browse files
committed
Added test case for bad/incomplete pdf files
1 parent 68b54a1 commit ba2a51a

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

tests/files.enc

80 KB
Binary file not shown.

tests/test_mupdf.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
import re
2+
13
from click.testing import CliRunner
24
import pytest
35

@@ -24,17 +26,21 @@ def test_cli(self, tmpdir):
2426
assert result.exit_code == 0
2527
assert "File saved" in result.output
2628

27-
fpath = tmpdir.join("output.txt")
29+
fpath = tmpdir.join("output.html")
2830
result = runner.invoke(
29-
cli, [self.cams_file_name, "-p", self.cams_password, "-o", fpath.strpath]
31+
cli, [self.cams_file_name, "-p", self.cams_password, "-o", fpath.strpath, "-s", "html"]
3032
)
3133
assert result.exit_code != 1
32-
assert "Output filename should end" in result.output
34+
assert "File saved" in result.output
3335

3436
result = runner.invoke(cli, [self.kfintech_file_name, "-p", self.cams_password])
3537
assert result.exit_code != 0
3638
assert "Incorrect PDF password!" in result.output
3739

40+
result = runner.invoke(cli, [self.bad_file_name, "-p", ""])
41+
assert result.exit_code == 0
42+
assert re.search(r"Error\s+:\s+1\s+schemes", result.output) is not None
43+
3844
def test_bad_investor_info(self):
3945
from casparser.parsers.mupdf import parse_investor_info
4046

0 commit comments

Comments
 (0)