mirror of
https://github.com/freedomofpress/dangerzone.git
synced 2025-04-29 02:12:36 +02:00
2bed3c10e4
Let the Document class suggest the default filename for the safe PDF, based on the provided input filename, appended with the extension `-safe.pdf`. Previously, this logic was copy-pasted throughout the code, which made it difficult to maintain.
44 lines
994 B
Python
44 lines
994 B
Python
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
sys.dangerzone_dev = True # type: ignore[attr-defined]
|
|
|
|
from dangerzone.document import SAFE_EXTENSION
|
|
|
|
SAMPLE_DIRECTORY = "test_docs"
|
|
BASIC_SAMPLE = "sample.pdf"
|
|
test_docs_dir = Path(__file__).parent.joinpath(SAMPLE_DIRECTORY)
|
|
test_docs = [
|
|
p
|
|
for p in test_docs_dir.rglob("*")
|
|
if p.is_file() and not p.name.endswith(SAFE_EXTENSION)
|
|
]
|
|
|
|
# Pytest parameter decorators
|
|
for_each_doc = pytest.mark.parametrize("doc", test_docs)
|
|
|
|
|
|
class TestBase:
|
|
sample_doc = str(test_docs_dir.joinpath(BASIC_SAMPLE))
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_doc() -> str:
|
|
return str(test_docs_dir.joinpath(BASIC_SAMPLE))
|
|
|
|
|
|
@pytest.fixture
|
|
def unwriteable_pdf(tmp_path: Path) -> str:
|
|
file_path = tmp_path / "document.pdf"
|
|
file_path.touch(mode=0o400)
|
|
return str(file_path)
|
|
|
|
|
|
@pytest.fixture
|
|
def unreadable_pdf(tmp_path: Path) -> str:
|
|
file_path = tmp_path / "document.pdf"
|
|
file_path.touch(mode=0o000)
|
|
return str(file_path)
|