import initial

tests/application/test_generate_puzzles.py

@@ -0,0 +1,113 @@
+import pytest
+from pathlib import Path
+from logimage.application.use_cases.generate_puzzles import (
+    GeneratePuzzlesRequest,
+    GeneratePuzzlesUseCase,
+)
+from tests.fakes.fakes import FakeImageSource, FakeImageConverter, FakePdfExporter
+
+
+def _make_use_case() -> tuple[GeneratePuzzlesUseCase, FakeImageSource, FakeImageConverter, FakePdfExporter]:
+    source = FakeImageSource()
+    converter = FakeImageConverter()
+    exporter = FakePdfExporter()
+    use_case = GeneratePuzzlesUseCase(source, converter, exporter)
+    return use_case, source, converter, exporter
+
+
+def test_generate_single_puzzle_calls_fetch_once() -> None:
+    use_case, source, _, _ = _make_use_case()
+    use_case.execute(GeneratePuzzlesRequest())
+    assert len(source.fetch_calls) == 1
+
+
+def test_generate_count_5_calls_fetch_5_times() -> None:
+    use_case, source, _, _ = _make_use_case()
+    use_case.execute(GeneratePuzzlesRequest(count=5))
+    assert len(source.fetch_calls) == 5
+
+
+def test_theme_forwarded_to_source() -> None:
+    use_case, source, _, _ = _make_use_case()
+    use_case.execute(GeneratePuzzlesRequest(theme="cats"))
+    assert source.fetch_calls[0] == "cats"
+
+
+def test_no_theme_passes_none_to_source() -> None:
+    use_case, source, _, _ = _make_use_case()
+    use_case.execute(GeneratePuzzlesRequest())
+    assert source.fetch_calls[0] is None
+
+
+def test_solution_flag_forwarded_to_exporter() -> None:
+    use_case, _, _, exporter = _make_use_case()
+    use_case.execute(GeneratePuzzlesRequest(with_solution=True))
+    assert exporter.last_with_solution is True
+
+
+def test_output_path_forwarded_to_exporter() -> None:
+    use_case, _, _, exporter = _make_use_case()
+    path = Path("my_output.pdf")
+    use_case.execute(GeneratePuzzlesRequest(output_path=path))
+    assert exporter.last_path == path
+
+
+def test_difficulty_easy_uses_10x10(monkeypatch: pytest.MonkeyPatch) -> None:
+    source = FakeImageSource()
+    exporter = FakePdfExporter()
+    sizes_used: list[tuple[int, int]] = []
+
+    class CapturingConverter(FakeImageConverter):
+        def to_grid(self, image_bytes: bytes, width: int, height: int):  # type: ignore[override]
+            sizes_used.append((width, height))
+            return super().to_grid(image_bytes, width, height)
+
+    use_case = GeneratePuzzlesUseCase(source, CapturingConverter(), exporter)
+    use_case.execute(GeneratePuzzlesRequest(difficulty="easy"))
+    assert sizes_used[0] == (10, 10)
+
+
+def test_difficulty_hard_uses_20x20(monkeypatch: pytest.MonkeyPatch) -> None:
+    source = FakeImageSource()
+    exporter = FakePdfExporter()
+    sizes_used: list[tuple[int, int]] = []
+
+    class CapturingConverter(FakeImageConverter):
+        def to_grid(self, image_bytes: bytes, width: int, height: int):  # type: ignore[override]
+            sizes_used.append((width, height))
+            return super().to_grid(image_bytes, width, height)
+
+    use_case = GeneratePuzzlesUseCase(source, CapturingConverter(), exporter)
+    use_case.execute(GeneratePuzzlesRequest(difficulty="hard"))
+    assert sizes_used[0] == (20, 20)
+
+
+def test_custom_size_overrides_difficulty(monkeypatch: pytest.MonkeyPatch) -> None:
+    source = FakeImageSource()
+    exporter = FakePdfExporter()
+    sizes_used: list[tuple[int, int]] = []
+
+    class CapturingConverter(FakeImageConverter):
+        def to_grid(self, image_bytes: bytes, width: int, height: int):  # type: ignore[override]
+            sizes_used.append((width, height))
+            return super().to_grid(image_bytes, width, height)
+
+    use_case = GeneratePuzzlesUseCase(source, CapturingConverter(), exporter)
+    use_case.execute(GeneratePuzzlesRequest(difficulty="easy", size=(25, 30)))
+    assert sizes_used[0] == (25, 30)
+
+
+def test_puzzle_title_comes_from_image_source() -> None:
+    source = FakeImageSource(title="Beautiful Cat")
+    exporter = FakePdfExporter()
+    use_case = GeneratePuzzlesUseCase(source, FakeImageConverter(), exporter)
+    use_case.execute(GeneratePuzzlesRequest())
+    assert exporter.exported_puzzles[0].title == "Beautiful Cat"
+
+
+def test_puzzle_image_bytes_comes_from_image_source() -> None:
+    source = FakeImageSource(content=b"real-image-bytes")
+    exporter = FakePdfExporter()
+    use_case = GeneratePuzzlesUseCase(source, FakeImageConverter(), exporter)
+    use_case.execute(GeneratePuzzlesRequest())
+    assert exporter.exported_puzzles[0].image_bytes == b"real-image-bytes"