imageforge/backend/tests/test_dummy_provider.py

from pathlib import Path

from backend.app.providers.dummy_provider import DummyProvider
from backend.app.providers.interface import ProviderRequest


def test_dummy_provider_generates_images(tmp_path: Path) -> None:
    provider = DummyProvider()
    req = ProviderRequest(
        prompt="test prompt",
        negative_prompt="",
        count=2,
        width=256,
        height=256,
        seed=123,
        steps=10,
        guidance=7.5,
    )

    events: list[tuple[int, str]] = []

    result = provider.generate(
        req,
        tmp_path,
        lambda p, m: events.append((p, m)),
        lambda: False,
    )

    assert len(result.image_paths) == 2
    assert all(path.exists() for path in result.image_paths)
    assert events
    assert events[-1][0] == 100