mirror of
https://github.com/freedomofpress/dangerzone.git
synced 2025-04-28 18:02:38 +02:00
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
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
|