23 lines
752 B
Python
23 lines
752 B
Python
import zipfile
|
|
|
|
from review_agent.regulatory_info_package.schemas import GeneratedFileResult
|
|
from review_agent.regulatory_info_package.services.zip_export import create_zip_package
|
|
|
|
|
|
def test_create_zip_package_includes_only_success_files(tmp_path):
|
|
success = tmp_path / "ok.docx"
|
|
failed = tmp_path / "bad.docx"
|
|
success.write_bytes(b"ok")
|
|
failed.write_bytes(b"bad")
|
|
|
|
zip_path = create_zip_package(
|
|
tmp_path,
|
|
[
|
|
GeneratedFileResult("ok", "ok.docx", "docx", "docx", "success", path=str(success)),
|
|
GeneratedFileResult("bad", "bad.docx", "docx", "docx", "failed", path=str(failed)),
|
|
],
|
|
)
|
|
|
|
with zipfile.ZipFile(zip_path) as archive:
|
|
assert archive.namelist() == ["ok.docx"]
|