recipe: harden auth/metrics, add rate limiting and tests
This commit is contained in:
@@ -28,4 +28,19 @@ def test_missing_key_returns_service_unavailable(monkeypatch) -> None:
|
||||
files={"image": ("input.png", b"not-an-image", "image/png")},
|
||||
data={"prompt": "test", "generation_config": "{}"},
|
||||
)
|
||||
assert response.status_code == 503
|
||||
|
||||
|
||||
def test_rejects_when_only_override_provided(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": "{}",
|
||||
"api_key_override": "manual-key",
|
||||
},
|
||||
)
|
||||
assert response.status_code == 503
|
||||
Reference in New Issue
Block a user