almet/dangerzone
1
0
Fork 0
mirror of https://github.com/freedomofpress/dangerzone.git synced 2025-04-29 10:12:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

Tests: add cli --archive param test

Browse source
This commit is contained in:
deeplow 2022-11-24 10:50:06 +00:00
parent b4849995e3
commit c36f73ac8d
No known key found for this signature in database
GPG key ID: 577982871529A52A
1 changed files with 17 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/test_cli.py
View file

@ -15,6 +15,7 @@ from click.testing import CliRunner, Result
from strip_ansi import strip_ansi from strip_ansi import strip_ansi
from dangerzone.cli import cli_main, display_banner from dangerzone.cli import cli_main, display_banner
from dangerzone.document import ARCHIVE_SUBDIR, SAFE_EXTENSION
from . import TestBase, for_each_doc from . import TestBase, for_each_doc
@ -237,6 +238,22 @@ class TestCliConversion(TestCliBasic):
result = self.run_cli(['--output-filename="output.pdf"'] + file_paths) result = self.run_cli(['--output-filename="output.pdf"'] + file_paths)
result.assert_failure() result.assert_failure()
def test_archive(self, tmp_path: Path) -> None:
test_string = "original file"
original_doc_path = str(tmp_path / "doc.pdf")
safe_doc_path = str(tmp_path / f"doc{SAFE_EXTENSION}")
archived_doc_path = str(tmp_path / ARCHIVE_SUBDIR / "doc.pdf")
shutil.copyfile(self.sample_doc, original_doc_path)
result = self.run_cli(["--archive", original_doc_path])
result.assert_success()
# original document has been moved to unsafe/doc.pdf
assert not os.path.exists(original_doc_path)
assert os.path.exists(archived_doc_path)
assert os.path.exists(safe_doc_path)
class TestSecurity(TestCli): class TestSecurity(TestCli):
def test_suspicious_double_dash_file(self, tmp_path: Path) -> None: def test_suspicious_double_dash_file(self, tmp_path: Path) -> None: