import initial

2026-06-10 10:21:18 +02:00
commit 5a03f8a38d
59 changed files with 4777 additions and 0 deletions
@@ -0,0 +1,80 @@
+import io
+import tempfile
+from pathlib import Path
+from PIL import Image as PilImage
+from logimage.infrastructure.pdf.reportlab_exporter import ReportLabPdfExporter
+from logimage.domain.entities.puzzle import NonogramPuzzle
+from logimage.domain.value_objects.grid import Grid
+
+
+def _make_puzzle(rows: int = 10, cols: int = 10, title: str = "Test") -> NonogramPuzzle:
+    cells = [[i % 3 == 0] * cols for i in range(rows)]
+    grid = Grid.from_list(cells)
+    return NonogramPuzzle.from_grid(grid, title)
+
+
+def test_export_creates_pdf_file() -> None:
+    exporter = ReportLabPdfExporter()
+    with tempfile.TemporaryDirectory() as tmp:
+        path = Path(tmp) / "output.pdf"
+        exporter.export([_make_puzzle()], path)
+        assert path.exists()
+        assert path.stat().st_size > 1000
+
+
+def test_export_multiple_puzzles_creates_larger_file() -> None:
+    exporter = ReportLabPdfExporter()
+    with tempfile.TemporaryDirectory() as tmp:
+        path_single = Path(tmp) / "single.pdf"
+        path_multi = Path(tmp) / "multi.pdf"
+        exporter.export([_make_puzzle()], path_single)
+        exporter.export([_make_puzzle(), _make_puzzle(), _make_puzzle()], path_multi)
+        assert path_multi.stat().st_size > path_single.stat().st_size
+
+
+def test_export_with_solution_creates_larger_file() -> None:
+    exporter = ReportLabPdfExporter()
+    with tempfile.TemporaryDirectory() as tmp:
+        path_no_sol = Path(tmp) / "no_solution.pdf"
+        path_with_sol = Path(tmp) / "with_solution.pdf"
+        puzzle = _make_puzzle()
+        exporter.export([puzzle], path_no_sol, with_solution=False)
+        exporter.export([puzzle], path_with_sol, with_solution=True)
+        assert path_with_sol.stat().st_size > path_no_sol.stat().st_size
+
+
+def test_export_starts_with_pdf_magic_bytes() -> None:
+    exporter = ReportLabPdfExporter()
+    with tempfile.TemporaryDirectory() as tmp:
+        path = Path(tmp) / "output.pdf"
+        exporter.export([_make_puzzle()], path)
+        assert path.read_bytes()[:4] == b"%PDF"
+
+
+def _make_puzzle_with_image(rows: int = 10, cols: int = 10) -> NonogramPuzzle:
+    buf = io.BytesIO()
+    PilImage.new("RGB", (40, 40), color=(100, 150, 200)).save(buf, format="JPEG")
+    cells = [[i % 3 == 0] * cols for i in range(rows)]
+    grid = Grid.from_list(cells)
+    return NonogramPuzzle.from_grid(grid, "Test With Image", buf.getvalue())
+
+
+def test_solution_with_image_bytes_creates_larger_file() -> None:
+    exporter = ReportLabPdfExporter()
+    puzzle_no_img = _make_puzzle()
+    puzzle_with_img = _make_puzzle_with_image()
+    with tempfile.TemporaryDirectory() as tmp:
+        path_no_img = Path(tmp) / "no_img.pdf"
+        path_with_img = Path(tmp) / "with_img.pdf"
+        exporter.export([puzzle_no_img], path_no_img, with_solution=True)
+        exporter.export([puzzle_with_img], path_with_img, with_solution=True)
+        assert path_with_img.stat().st_size > path_no_img.stat().st_size
+
+
+def test_solution_without_image_bytes_still_produces_valid_pdf() -> None:
+    exporter = ReportLabPdfExporter()
+    puzzle = _make_puzzle()  # image_bytes is None
+    with tempfile.TemporaryDirectory() as tmp:
+        path = Path(tmp) / "sol.pdf"
+        exporter.export([puzzle], path, with_solution=True)
+        assert path.read_bytes()[:4] == b"%PDF"