65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
import zipfile
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from backend.app.services.archive_extractor import extract_archive, is_archive
|
|
|
|
|
|
def test_is_archive():
|
|
assert is_archive(Path("a.zip"))
|
|
assert is_archive(Path("a.CBZ"))
|
|
assert is_archive(Path("a.rar"))
|
|
assert is_archive(Path("a.7z"))
|
|
assert not is_archive(Path("a.jpg"))
|
|
|
|
|
|
def _zip(path, entries):
|
|
with zipfile.ZipFile(path, "w") as zf:
|
|
for name, data in entries.items():
|
|
zf.writestr(name, data)
|
|
|
|
|
|
def test_zip_members_enumerated(tmp_path):
|
|
z = tmp_path / "set.zip"
|
|
_zip(z, {"a.jpg": b"img", "sub/b.png": b"img2", "notes.txt": b"hi"})
|
|
with extract_archive(z) as members:
|
|
names = sorted(n for n, _ in members)
|
|
assert names == ["a.jpg", "notes.txt", "sub/b.png"]
|
|
for _, p in members:
|
|
assert p.is_file()
|
|
for _, p in members:
|
|
assert not p.exists() # temp cleaned on exit
|
|
|
|
|
|
def test_cbz_alias(tmp_path):
|
|
z = tmp_path / "set.cbz"
|
|
_zip(z, {"p1.jpg": b"i"})
|
|
with extract_archive(z) as members:
|
|
assert [n for n, _ in members] == ["p1.jpg"]
|
|
|
|
|
|
def test_corrupt_archive_yields_nothing(tmp_path):
|
|
bad = tmp_path / "broken.zip"
|
|
bad.write_bytes(b"not a zip at all")
|
|
with extract_archive(bad) as members:
|
|
assert members == []
|
|
|
|
|
|
def test_7z_roundtrip(tmp_path):
|
|
py7zr = pytest.importorskip("py7zr")
|
|
a = tmp_path / "s.7z"
|
|
(tmp_path / "x.jpg").write_bytes(b"img")
|
|
with py7zr.SevenZipFile(a, "w") as zf:
|
|
zf.write(tmp_path / "x.jpg", "x.jpg")
|
|
with extract_archive(a) as members:
|
|
assert [n for n, _ in members] == ["x.jpg"]
|
|
|
|
|
|
def test_rar_corrupt_is_failsoft(tmp_path):
|
|
pytest.importorskip("rarfile")
|
|
bad = tmp_path / "b.rar"
|
|
bad.write_bytes(b"Rar! not really")
|
|
with extract_archive(bad) as members:
|
|
assert members == [] # invalid / no unrar → fail-soft, nothing lost
|