from pathlib import Path

from aistudio_api.domain.models import parse_image_output


ROOT = Path(__file__).resolve().parents[1]


def test_parse_image_output_keeps_only_final_images_in_images_field():
    raw = (ROOT / "test-image-output.json").read_text()
    output = parse_image_output(raw)

    assert len(output.images) == 1
    assert len(output.reasoning_images) == 1
    assert output.images[0].mime == "image/jpeg"
    assert output.reasoning_images[0].mime == "image/jpeg"
    assert output.images[0].data != output.reasoning_images[0].data
    assert output.thinking.startswith("**Envisioning a Kitty Scene**")