|
@@ -1,7 +1,10 @@
|
|
|
|
|
+import importlib
|
|
|
|
|
+import os
|
|
|
import sys
|
|
import sys
|
|
|
import unittest
|
|
import unittest
|
|
|
from pathlib import Path
|
|
from pathlib import Path
|
|
|
from tempfile import TemporaryDirectory
|
|
from tempfile import TemporaryDirectory
|
|
|
|
|
+from unittest.mock import patch
|
|
|
|
|
|
|
|
from openpyxl import load_workbook
|
|
from openpyxl import load_workbook
|
|
|
|
|
|
|
@@ -13,6 +16,7 @@ from tools.im_approval_creation import ( # noqa: E402
|
|
|
DECISION_COL_LETTER,
|
|
DECISION_COL_LETTER,
|
|
|
generate_approval_xlsx,
|
|
generate_approval_xlsx,
|
|
|
)
|
|
)
|
|
|
|
|
+import tools.im_approval_creation as approval_module # noqa: E402
|
|
|
|
|
|
|
|
|
|
|
|
|
def _sample_record() -> dict:
|
|
def _sample_record() -> dict:
|
|
@@ -39,6 +43,13 @@ def _sample_record() -> dict:
|
|
|
|
|
|
|
|
|
|
|
|
|
class CreativeApprovalSheetTest(unittest.TestCase):
|
|
class CreativeApprovalSheetTest(unittest.TestCase):
|
|
|
|
|
+ def test_embeds_material_preview_images_by_default(self):
|
|
|
|
|
+ with patch.dict(os.environ, {}, clear=False):
|
|
|
|
|
+ os.environ.pop("CREATION_APPROVAL_EMBED_IMAGES", None)
|
|
|
|
|
+ reloaded = importlib.reload(approval_module)
|
|
|
|
|
+
|
|
|
|
|
+ self.assertTrue(reloaded.EMBED_MATERIAL_PREVIEW_IMAGES)
|
|
|
|
|
+
|
|
|
def test_keeps_dedicated_material_link_column(self):
|
|
def test_keeps_dedicated_material_link_column(self):
|
|
|
with TemporaryDirectory() as tmp:
|
|
with TemporaryDirectory() as tmp:
|
|
|
output = Path(tmp) / "approval.xlsx"
|
|
output = Path(tmp) / "approval.xlsx"
|