mirror of
https://github.com/freedomofpress/dangerzone.git
synced 2025-04-28 18:02:38 +02:00
0f4e6e9156
originally PDF files were included for these edge-cases but in reality all we want to test is the filename itself. So it reduces repo size if we have them generated dynamically.
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
|
|
import pytest
|
|
from click.testing import CliRunner, Result
|
|
|
|
from dangerzone.cli import cli_main
|
|
|
|
from . import TestBase, for_each_doc
|
|
|
|
# TODO --output-filename with spaces
|
|
# TODO explore any symlink edge cases
|
|
# TODO simulate ctrl-c, ctrl-d, SIGINT/SIGKILL/SIGTERM... (man 7 signal), etc?
|
|
# TODO validate output PDFs https://github.com/pdfminer/pdfminer.six
|
|
# TODO trigger "Invalid json returned from container"
|
|
# TODO trigger "pdf-to-pixels failed"
|
|
# TODO simulate container runtime missing
|
|
# TODO simulate container connection error
|
|
# TODO simulate container connection loss
|
|
# FIXME "/" path separator is platform-dependent, use pathlib instead
|
|
|
|
|
|
class TestCli(TestBase):
|
|
def run_cli(self, *args, **kwargs) -> Result:
|
|
return CliRunner().invoke(cli_main, *args, **kwargs)
|
|
|
|
|
|
class TestCliBasic(TestCli):
|
|
def test_no_args(self):
|
|
"""``$ dangerzone-cli``"""
|
|
result = self.run_cli()
|
|
assert result.exit_code != 0
|
|
|
|
def test_help(self):
|
|
"""``$ dangerzone-cli --help``"""
|
|
result = self.run_cli("--help")
|
|
assert result.exit_code == 0
|
|
|
|
|
|
class TestCliConversion(TestCliBasic):
|
|
def test_invalid_lang(self):
|
|
result = self.run_cli(f"{self.sample_doc} --ocr-lang piglatin")
|
|
assert result.exit_code != 0
|
|
|
|
@for_each_doc
|
|
def test_formats(self, doc):
|
|
result = self.run_cli(f'"{doc}"')
|
|
assert result.exit_code == 0
|
|
|
|
def test_output_filename(self):
|
|
temp_dir = tempfile.mkdtemp(prefix="dangerzone-")
|
|
result = self.run_cli(
|
|
f"{self.sample_doc} --output-filename {temp_dir}/safe.pdf"
|
|
)
|
|
assert result.exit_code == 0
|
|
|
|
def test_output_filename_new_dir(self):
|
|
result = self.run_cli(
|
|
f"{self.sample_doc} --output-filename fake-directory/my-output.pdf"
|
|
)
|
|
assert result.exit_code != 0
|
|
|
|
def test_sample_not_found(self):
|
|
result = self.run_cli("fake-directory/fake-file.pdf")
|
|
assert result.exit_code != 0
|
|
|
|
def test_lang_eng(self):
|
|
result = self.run_cli(f'"{self.sample_doc}" --ocr-lang eng')
|
|
assert result.exit_code == 0
|
|
|
|
@pytest.mark.parametrize(
|
|
"filename,",
|
|
[
|
|
"“Curly_Quotes”.pdf", # issue 144
|
|
"Оригинал.pdf",
|
|
],
|
|
)
|
|
def test_filenames(self, filename):
|
|
tempdir = tempfile.mkdtemp(prefix="dangerzone-")
|
|
doc_path = os.path.join(filename)
|
|
shutil.copyfile(self.sample_doc, doc_path)
|
|
result = self.run_cli(doc_path)
|
|
shutil.rmtree(tempdir)
|
|
assert result.exit_code == 0
|