# retoor <retoor@molodetz.nl>

import requests
from tests.api.uploads.uploadurl import _key, _upload_settings  # noqa: F401
from tests.conftest import BASE_URL


def test_generate_image_requires_auth():
    response = requests.post(f"{BASE_URL}/uploads/generate-image", data={"prompt": "a cat"})
    assert response.status_code == 401


def test_generate_image_rejects_empty_prompt():
    response = requests.post(
        f"{BASE_URL}/uploads/generate-image",
        headers={"X-API-KEY": _key()},
        data={"prompt": ""},
        allow_redirects=False,
    )
    assert response.status_code != 201


def test_generate_image_upstream_failure_is_a_clean_json_error():
    response = requests.post(
        f"{BASE_URL}/uploads/generate-image",
        headers={"X-API-KEY": _key()},
        data={"prompt": "a lighthouse at dusk"},
    )
    assert response.status_code in (201, 502)
    body = response.json()
    if response.status_code == 502:
        assert body["error"]
    else:
        assert body["uid"]