49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import io
|
|
import json
|
|
import tempfile
|
|
import zipfile
|
|
from pathlib import Path
|
|
|
|
from app.config_upload import install_config_files, normalize_json, validate_xlsx
|
|
|
|
|
|
def _minimal_xlsx() -> bytes:
|
|
buffer = io.BytesIO()
|
|
with zipfile.ZipFile(buffer, "w") as archive:
|
|
archive.writestr("[Content_Types].xml", "<Types/>")
|
|
archive.writestr("xl/workbook.xml", "<workbook/>")
|
|
return buffer.getvalue()
|
|
|
|
|
|
def test_validate_xlsx_accepts_workbook_zip():
|
|
assert validate_xlsx(_minimal_xlsx()) == _minimal_xlsx()
|
|
|
|
|
|
def test_validate_xlsx_rejects_arbitrary_pk_data():
|
|
try:
|
|
validate_xlsx(b"PK-not-a-real-xlsx")
|
|
except ValueError as exc:
|
|
assert "XLSX" in str(exc)
|
|
else:
|
|
raise AssertionError("invalid XLSX was accepted")
|
|
|
|
|
|
def test_normalize_json_rejects_scalar_and_formats_object():
|
|
assert json.loads(normalize_json(b'{"a":1}')) == {"a": 1}
|
|
try:
|
|
normalize_json(b'"scalar"')
|
|
except ValueError as exc:
|
|
assert "object" in str(exc)
|
|
else:
|
|
raise AssertionError("scalar JSON was accepted")
|
|
|
|
|
|
def test_install_config_files_replaces_and_backs_up_existing_file():
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
target = root / "rules.json"
|
|
target.write_bytes(b"old")
|
|
install_config_files(root, [("rules.json", b"new")])
|
|
assert target.read_bytes() == b"new"
|
|
assert (root / "rules.json.bak").read_bytes() == b"old"
|