fz0x1/HPI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

my.core.structure: add support for .tar.gz archives

Browse source 
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
This commit is contained in:
Dima Gerasimov 2024-09-16 23:41:58 +01:00 • committed by karlicoss
parent 27178c0939
commit 201ddd4d7c
3 changed files with 33 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

7
my/core/tests/structure.py
View file

@ -14,8 +14,9 @@ def test_gdpr_structure_exists() -> None:
assert results == (structure_data / "gdpr_subdirs" / "gdpr_export",)
def test_gdpr_unzip() -> None:
with match_structure(structure_data / "gdpr_export.zip", expected=gdpr_expected) as results:
@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"
@ -32,6 +33,6 @@ def test_match_partial() -> None:
def test_not_directory() -> None:
with pytest.raises(NotADirectoryError, match=r"Expected either a zipfile or a directory"):
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