forked from retoor/devplacepy
35 lines
1018 B
Python
35 lines
1018 B
Python
 
|
# 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"]
|