busya
63d82df53b
Backend (4 files, 73 tests): - test_agent_superset_routes.py (27 tests, 35% -> 92%) - test_agent_lifecycle_routes.py (11 tests, 50% -> 100%) - test_agent_status_routes.py (6 tests, 57% -> 100%) - test_git_release_routes.py (31 tests, 30% -> 99%) Frontend (~15 files, ~120 tests): - cron.ts: 0% -> 100% - ReportsLogModel: 0% -> 99% - parseCot.ts: 10% -> 100% - sessionTimeout.ts: 64% -> 93% - MappingsModel: 65% -> 100% - TranslateHistoryModel: 65% -> 93% - Migration.ExecutorModel: 70% -> 100% - GitManagerModel: 78% -> 90% - TranslationJobModel: 77% -> 80% - ConfirmDialog: 59% -> 80% - api.ts: 78% -> 80% Coverage: frontend 0 violations, backend 7518 passed.
111 lines
4.4 KiB
Python
111 lines
4.4 KiB
Python
# #region Test.Api.AgentStatusRoutes [C:2] [TYPE Module] [SEMANTICS test,agent,llm,status,health]
|
|
# @BRIEF Unit tests for Agent LLM provider health status endpoint.
|
|
# @RELATION BINDS_TO -> [Api.Agent.Status]
|
|
|
|
import os
|
|
|
|
os.environ.setdefault("DATABASE_URL", "sqlite:///:memory:")
|
|
os.environ.setdefault("AUTH_DATABASE_URL", "sqlite:///:memory:")
|
|
os.environ.setdefault("SECRET_KEY", "test-secret-key-for-tests")
|
|
os.environ.setdefault("DEV_MODE", "true")
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
|
|
# Add shared/src to path for ss_tools.shared._llm_health
|
|
_src = str(Path(__file__).resolve().parent.parent.parent / "src")
|
|
if _src not in sys.path:
|
|
sys.path.insert(0, _src)
|
|
|
|
_shared_src = str(Path(__file__).resolve().parent.parent.parent.parent / "shared" / "src")
|
|
if _shared_src not in sys.path:
|
|
sys.path.insert(0, _shared_src)
|
|
|
|
|
|
def _make_client() -> TestClient:
|
|
"""Build a TestClient with the agent status router."""
|
|
from src.api.routes.agent_status import router
|
|
|
|
app = FastAPI()
|
|
app.include_router(router)
|
|
return TestClient(app)
|
|
|
|
|
|
class TestGetLlmStatus:
|
|
"""GET /api/agent/llm-status"""
|
|
|
|
@patch("ss_tools.shared._llm_health._check_llm_provider_health", AsyncMock(return_value="ok"))
|
|
@patch("ss_tools.shared._llm_health._llm_status", {"status": "ok", "last_error": ""})
|
|
def test_status_ok(self):
|
|
"""Returns ok when LLM provider is healthy."""
|
|
client = _make_client()
|
|
resp = client.get("/api/agent/llm-status")
|
|
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert data["status"] == "ok"
|
|
assert data["last_error"] == ""
|
|
assert data["retry_after_s"] == 0
|
|
|
|
@patch("ss_tools.shared._llm_health._check_llm_provider_health", AsyncMock(return_value="unavailable"))
|
|
@patch("ss_tools.shared._llm_health._llm_status", {"status": "unavailable", "last_error": "Provider unreachable"})
|
|
def test_status_unavailable(self):
|
|
"""Returns unavailable with retry_after_s > 0."""
|
|
client = _make_client()
|
|
resp = client.get("/api/agent/llm-status")
|
|
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert data["status"] == "unavailable"
|
|
assert data["last_error"] == "Provider unreachable"
|
|
assert data["retry_after_s"] == 30
|
|
|
|
@patch("ss_tools.shared._llm_health._check_llm_provider_health", AsyncMock(return_value="timeout"))
|
|
@patch("ss_tools.shared._llm_health._llm_status", {"status": "timeout", "last_error": "Request timed out"})
|
|
def test_status_timeout(self):
|
|
"""Returns timeout status."""
|
|
client = _make_client()
|
|
resp = client.get("/api/agent/llm-status")
|
|
|
|
assert resp.status_code == 200
|
|
assert resp.json()["status"] == "timeout"
|
|
|
|
@patch("ss_tools.shared._llm_health._check_llm_provider_health", AsyncMock(return_value="auth_error"))
|
|
@patch("ss_tools.shared._llm_health._llm_status", {"status": "auth_error", "last_error": "Invalid API key"})
|
|
def test_status_auth_error(self):
|
|
"""Returns auth_error status."""
|
|
client = _make_client()
|
|
resp = client.get("/api/agent/llm-status")
|
|
|
|
assert resp.status_code == 200
|
|
assert resp.json()["status"] == "auth_error"
|
|
assert resp.json()["last_error"] == "Invalid API key"
|
|
|
|
@patch("ss_tools.shared._llm_health._check_llm_provider_health", AsyncMock(return_value="ok"))
|
|
@patch("ss_tools.shared._llm_health._llm_status", {"status": "ok", "last_error": ""})
|
|
def test_no_auth_required(self):
|
|
"""Status endpoint does not require authentication."""
|
|
client = _make_client()
|
|
# No auth token — should still work since no Depends on get_current_user
|
|
resp = client.get("/api/agent/llm-status")
|
|
assert resp.status_code == 200
|
|
|
|
def test_check_health_called(self):
|
|
"""_check_llm_provider_health is called on each request."""
|
|
mock_check = AsyncMock(return_value="ok")
|
|
with (
|
|
patch("ss_tools.shared._llm_health._check_llm_provider_health", mock_check),
|
|
patch("ss_tools.shared._llm_health._llm_status", {"status": "ok", "last_error": ""}),
|
|
):
|
|
client = _make_client()
|
|
resp = client.get("/api/agent/llm-status")
|
|
|
|
assert resp.status_code == 200
|
|
mock_check.assert_called_once()
|
|
# #endregion Test.Api.AgentStatusRoutes
|