Add AI timeout tests

This commit is contained in:
Edern Deneuville
2026-09-11 13:44:14 +02:00
parent 9cf2178991
commit c117ef5440
+17 -1
View File
@@ -1,4 +1,7 @@
from ai_typewriter.ai_client import ask_ai
import pytest
import requests
from ai_typewriter.ai_client import AIClientError, ask_ai
from ai_typewriter.config import AppConfig
@@ -27,6 +30,19 @@ def test_ollama_payload_contains_strict_system_prompt(monkeypatch):
assert "Uniquement la réponse brute" in seen["json"]["messages"][0]["content"]
def test_ollama_timeout_message_suggests_config_change(monkeypatch):
def fake_post(url, json, timeout, **kwargs):
raise requests.Timeout("too slow")
monkeypatch.setattr("ai_typewriter.ai_client.requests.post", fake_post)
with pytest.raises(AIClientError) as exc:
ask_ai("texte", AppConfig(provider="ollama", model="m", request_timeout_seconds=300))
assert "300 s" in str(exc.value)
assert "request_timeout_seconds" in str(exc.value)
assert "0 pour désactiver" in str(exc.value)
def test_gemini_payload(monkeypatch):
seen = {}
def fake_post(url, params, json, timeout):