mirror of
https://github.com/ocrmypdf/OCRmyPDF.git
synced 2025-08-09 09:12:50 +00:00
101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
# © 2015-17 James R. Barlow: github.com/jbarlow83
|
|
#
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
|
|
from os import fspath
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from ocrmypdf._plugin_manager import get_parser_options_plugins
|
|
from ocrmypdf._validation import check_options
|
|
from ocrmypdf.exceptions import ExitCode, MissingDependencyError
|
|
|
|
from .conftest import check_ocrmypdf, have_unpaper, run_ocrmypdf
|
|
|
|
# pylint: disable=redefined-outer-name
|
|
|
|
|
|
def test_no_unpaper(resources, no_outpdf):
|
|
input_ = fspath(resources / "c02-22.pdf")
|
|
output = fspath(no_outpdf)
|
|
|
|
_parser, options, pm = get_parser_options_plugins(["--clean", input_, output])
|
|
with patch("ocrmypdf._exec.unpaper.version") as mock:
|
|
mock.side_effect = FileNotFoundError("unpaper")
|
|
|
|
with pytest.raises(MissingDependencyError):
|
|
check_options(options, pm)
|
|
mock.assert_called()
|
|
|
|
|
|
def test_old_unpaper(resources, no_outpdf):
|
|
input_ = fspath(resources / "c02-22.pdf")
|
|
output = fspath(no_outpdf)
|
|
|
|
_parser, options, pm = get_parser_options_plugins(["--clean", input_, output])
|
|
with patch("ocrmypdf._exec.unpaper.version") as mock:
|
|
mock.return_value = '0.5'
|
|
|
|
with pytest.raises(MissingDependencyError):
|
|
check_options(options, pm)
|
|
mock.assert_called()
|
|
|
|
|
|
@pytest.mark.skipif(not have_unpaper(), reason="requires unpaper")
|
|
def test_clean(resources, outpdf):
|
|
check_ocrmypdf(
|
|
resources / "skew.pdf",
|
|
outpdf,
|
|
"-c",
|
|
'--plugin',
|
|
'tests/plugins/tesseract_noop.py',
|
|
)
|
|
|
|
|
|
@pytest.mark.skipif(not have_unpaper(), reason="requires unpaper")
|
|
def test_unpaper_args_valid(resources, outpdf):
|
|
check_ocrmypdf(
|
|
resources / "skew.pdf",
|
|
outpdf,
|
|
"-c",
|
|
"--unpaper-args",
|
|
"--layout double", # Spaces required here
|
|
'--plugin',
|
|
'tests/plugins/tesseract_noop.py',
|
|
)
|
|
|
|
|
|
@pytest.mark.skipif(not have_unpaper(), reason="requires unpaper")
|
|
def test_unpaper_args_invalid_filename(resources, outpdf):
|
|
p, _out, err = run_ocrmypdf(
|
|
resources / "skew.pdf",
|
|
outpdf,
|
|
"-c",
|
|
"--unpaper-args",
|
|
"/etc/passwd",
|
|
'--plugin',
|
|
'tests/plugins/tesseract_noop.py',
|
|
)
|
|
assert "No filenames allowed" in err
|
|
assert p.returncode == ExitCode.bad_args
|
|
|
|
|
|
@pytest.mark.skipif(not have_unpaper(), reason="requires unpaper")
|
|
def test_unpaper_args_invalid(resources, outpdf):
|
|
p, _out, _err = run_ocrmypdf(
|
|
resources / "skew.pdf",
|
|
outpdf,
|
|
"-c",
|
|
"--unpaper-args",
|
|
"unpaper is not going to like these arguments",
|
|
'--plugin',
|
|
'tests/plugins/tesseract_noop.py',
|
|
)
|
|
# Can't tell difference between unpaper choking on bad arguments or some
|
|
# other unpaper failure
|
|
assert p.returncode == ExitCode.child_process_error
|