from spriteforge.studio.models import GenerationRecipe, GenerationRequest
from spriteforge.studio.providers import create_provider, provider_names


def test_diagnostic_provider_exercises_batch_without_gpu():
    request=GenerationRequest(project_id="p",asset_id="agent",shot_id="shot_1",count=3,
        recipe=GenerationRecipe(provider="diagnostic",model="test",seed=9,prompt="x",width=96,height=80))
    output=create_provider("diagnostic").generate(request,lambda _: b"")
    assert len(output)==3 and all(blob.startswith(b"\x89PNG") for blob in output)
    assert "diagnostic" in provider_names()