201ddd4d7c
this will be useful to migrate .tar.gz processing to kompress in a backwards compatible way, or to run them against unpacked folder structure if user prefers
38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from ..structure import match_structure
|
|
|
|
structure_data: Path = Path(__file__).parent / "structure_data"
|
|
|
|
gdpr_expected = ("comments", "messages/index.csv", "profile")
|
|
|
|
|
|
def test_gdpr_structure_exists() -> None:
|
|
with match_structure(structure_data, expected=gdpr_expected) as results:
|
|
assert results == (structure_data / "gdpr_subdirs" / "gdpr_export",)
|
|
|
|
|
|
@pytest.mark.parametrize("archive", ["gdpr_export.zip", "gdpr_export.tar.gz"])
|
|
def test_gdpr_unpack(archive: str) -> None:
|
|
with match_structure(structure_data / archive, expected=gdpr_expected) as results:
|
|
assert len(results) == 1
|
|
extracted = results[0]
|
|
index_file = extracted / "messages" / "index.csv"
|
|
assert index_file.read_text().strip() == "test message"
|
|
|
|
# make sure the temporary directory this created no longer exists
|
|
assert not extracted.exists()
|
|
|
|
|
|
def test_match_partial() -> None:
|
|
# a partial match should match both the 'broken' and 'gdpr_export' directories
|
|
with match_structure(structure_data / "gdpr_subdirs", expected=gdpr_expected, partial=True) as results:
|
|
assert len(results) == 2
|
|
|
|
|
|
def test_not_directory() -> None:
|
|
with pytest.raises(NotADirectoryError, match=r"Expected either a zip/tar.gz archive or a directory"):
|
|
with match_structure(structure_data / "messages/index.csv", expected=gdpr_expected):
|
|
pass
|