Add isolated Gemini image proxy

This commit is contained in:
“Naeel”
2026-08-28 19:20:51 +03:00
parent ec57cf5d57
commit df5afcec8f
11 changed files with 849 additions and 0 deletions
+31
View File
@@ -0,0 +1,31 @@
from fastapi.testclient import TestClient
from app import app
def test_health() -> None:
client = TestClient(app)
response = client.get("/health")
assert response.status_code == 200
assert response.json() == {"status": "ok"}
def test_rejects_unsupported_type() -> None:
client = TestClient(app)
response = client.post(
"/gemini",
files={"image": ("input.txt", b"not-an-image", "text/plain")},
data={"prompt": "test", "generation_config": "{}"},
)
assert response.status_code == 415
def test_missing_key_returns_service_unavailable(monkeypatch) -> None:
monkeypatch.delenv("GEMINI_API_KEY", raising=False)
client = TestClient(app)
response = client.post(
"/gemini",
files={"image": ("input.png", b"not-an-image", "image/png")},
data={"prompt": "test", "generation_config": "{}"},
)
assert response.status_code == 503