Thinh Lam
100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
"""Unit tests for backup ZIP helpers (no database)."""
|
|
|
|
import unittest
|
|
|
|
from src.initiative_db.backup_naming import backup_zip_attachment_filename, official_form_pdf_backup_zip_path
|
|
from src.initiative_db.application_storage import (
|
|
EVIDENCE_ROLE_RESEARCH,
|
|
STORAGE_FILESYSTEM,
|
|
STORAGE_MINIO_ATTACHMENTS,
|
|
STORAGE_MINIO_EXPORTS,
|
|
effective_storage_kind,
|
|
)
|
|
|
|
|
|
class OfficialFormPdfZipPathTests(unittest.TestCase):
|
|
def test_trang_bia_fields(self) -> None:
|
|
obm = {
|
|
"TRANG BÌA": {
|
|
"Tên sáng kiến (Tiếng Việt)": " Khảo sát thảo dược ",
|
|
"Tác giả/nhóm tác giả sáng kiến": "Lê Thị A",
|
|
"Thông tin liên hệ (Điện thoại, Email)": "0909, a.b@ump.edu.vn",
|
|
}
|
|
}
|
|
p = official_form_pdf_backup_zip_path(obm)
|
|
self.assertEqual(p, "submitted/Khảo_sát_thảo_dược_Lê_Thị_A_a.b@ump.edu.vn.pdf")
|
|
|
|
def test_no_trang_bia_returns_none(self) -> None:
|
|
self.assertIsNone(official_form_pdf_backup_zip_path({}))
|
|
self.assertIsNone(official_form_pdf_backup_zip_path({"OTHER": {}}))
|
|
|
|
def test_empty_cover_returns_none(self) -> None:
|
|
self.assertIsNone(
|
|
official_form_pdf_backup_zip_path({"TRANG BÌA": {"Tên sáng kiến (Tiếng Việt)": " "}})
|
|
)
|
|
|
|
|
|
class BackupZipFilenameTests(unittest.TestCase):
|
|
def test_email_local_part_and_sub_id(self) -> None:
|
|
self.assertEqual(
|
|
backup_zip_attachment_filename(
|
|
owner_email=" nguyen.van.a@ump.edu.vn ",
|
|
owner_full_name="Nguyễn Văn A",
|
|
public_application_id="sub-deadbeef",
|
|
),
|
|
"nguyen.van.a_sub-deadbeef.zip",
|
|
)
|
|
|
|
def test_fallback_name_when_no_email(self) -> None:
|
|
fn = backup_zip_attachment_filename(
|
|
owner_email=None,
|
|
owner_full_name=" Lê Thị B ",
|
|
public_application_id="sub-001",
|
|
)
|
|
self.assertTrue(fn.endswith("_sub-001.zip"))
|
|
self.assertIn("Lê", fn)
|
|
|
|
def test_applicant_fallback(self) -> None:
|
|
self.assertEqual(
|
|
backup_zip_attachment_filename(
|
|
owner_email="",
|
|
owner_full_name="",
|
|
public_application_id="sub-x",
|
|
),
|
|
"applicant_sub-x.zip",
|
|
)
|
|
|
|
|
|
class EffectiveStorageKindTests(unittest.TestCase):
|
|
def test_full_pdf_minio_key(self) -> None:
|
|
self.assertEqual(
|
|
effective_storage_kind("full_pdf", "initiatives/abcd/2025/01/x-file.pdf", None),
|
|
STORAGE_MINIO_EXPORTS,
|
|
)
|
|
|
|
def test_full_pdf_filesystem(self) -> None:
|
|
self.assertEqual(
|
|
effective_storage_kind("full_pdf", "/submitted-initiatives/sub-abc.pdf", None),
|
|
STORAGE_FILESYSTEM,
|
|
)
|
|
|
|
def test_evidence_attachments(self) -> None:
|
|
self.assertEqual(
|
|
effective_storage_kind(
|
|
EVIDENCE_ROLE_RESEARCH,
|
|
"initiatives/abcd/2025/01/x-file.pdf",
|
|
None,
|
|
),
|
|
STORAGE_MINIO_ATTACHMENTS,
|
|
)
|
|
|
|
def test_respects_declared(self) -> None:
|
|
self.assertEqual(
|
|
effective_storage_kind("full_pdf", "/any", STORAGE_MINIO_EXPORTS),
|
|
STORAGE_MINIO_EXPORTS,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|