Vincent Bourdon
114 lines
4.3 KiB
Python
114 lines
4.3 KiB
Python
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"
|