58ff43bdee
Swap Nous Portal for the newsPipeline local_llm generate path (POST /api/generate, GET /api/tags). No API key. Settings model selector lists Ollama tags; idle if the host is down.
147 lines
4.2 KiB
Python
147 lines
4.2 KiB
Python
"""API tests for GET/PUT /api/settings and GET /api/news/models."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import os
|
|
|
|
import asyncpg
|
|
import httpx
|
|
import pytest
|
|
|
|
from conftest import requires_db
|
|
|
|
from main import app
|
|
|
|
BASE = "http://test"
|
|
|
|
|
|
def _conn_kwargs() -> dict:
|
|
return {
|
|
"host": os.environ["DB_HOST"],
|
|
"port": int(os.environ["DB_PORT"]),
|
|
"user": os.environ["DB_USER"],
|
|
"password": os.environ["DB_PASSWORD"],
|
|
"database": os.environ["DB_NAME"],
|
|
}
|
|
|
|
|
|
def _truncate_settings() -> None:
|
|
async def run():
|
|
conn = await asyncpg.connect(**_conn_kwargs())
|
|
try:
|
|
await conn.execute("DROP TABLE IF EXISTS app_settings")
|
|
finally:
|
|
await conn.close()
|
|
|
|
asyncio.run(run())
|
|
|
|
|
|
@pytest.fixture()
|
|
def clean_settings():
|
|
import settings_store
|
|
settings_store._ensured = False
|
|
_truncate_settings()
|
|
yield
|
|
settings_store._ensured = False
|
|
_truncate_settings()
|
|
|
|
|
|
def _request(method: str, path: str, json: dict | None = None) -> httpx.Response:
|
|
async def _run() -> httpx.Response:
|
|
transport = httpx.ASGITransport(app=app)
|
|
async with httpx.AsyncClient(transport=transport, base_url=BASE) as client:
|
|
return await client.request(method, path, json=json)
|
|
|
|
return asyncio.run(_run())
|
|
|
|
|
|
def _get(path: str) -> httpx.Response:
|
|
return _request("GET", path)
|
|
|
|
|
|
def _put(path: str, json: dict) -> httpx.Response:
|
|
return _request("PUT", path, json=json)
|
|
|
|
|
|
def test_get_news_models_ollama_down_returns_fallback(monkeypatch):
|
|
async def boom_http_get(url, *, headers, timeout):
|
|
raise httpx.ConnectError("ollama down")
|
|
|
|
import settings_store
|
|
monkeypatch.setattr(settings_store, "_http_get", boom_http_get)
|
|
settings_store._models_cache = None
|
|
|
|
resp = _get("/api/news/models")
|
|
assert resp.status_code == 200
|
|
body = resp.json()
|
|
assert body["source"] == "fallback"
|
|
ids = [m["id"] for m in body["models"]]
|
|
from settings_store import FALLBACK_MODELS
|
|
assert ids == FALLBACK_MODELS
|
|
assert "qwen3:30b-a3b" in ids
|
|
|
|
|
|
def test_get_news_models_live_from_ollama_tags(monkeypatch):
|
|
monkeypatch.setenv("LLM_URL", "http://ollama:11434")
|
|
monkeypatch.delenv("OSINT_USER_AGENT", raising=False)
|
|
|
|
captured: dict = {}
|
|
|
|
class FakeResponse:
|
|
status_code = 200
|
|
|
|
def raise_for_status(self):
|
|
return None
|
|
|
|
def json(self):
|
|
return {"models": [{"name": "qwen3:30b-a3b"}, {"name": "llama3.2:latest"}]}
|
|
|
|
async def fake_http_get(url, *, headers, timeout):
|
|
captured["url"] = url
|
|
captured["headers"] = headers
|
|
captured["timeout"] = timeout
|
|
return FakeResponse()
|
|
|
|
import settings_store
|
|
monkeypatch.setattr(settings_store, "_http_get", fake_http_get)
|
|
settings_store._models_cache = None
|
|
|
|
resp = _get("/api/news/models")
|
|
assert resp.status_code == 200
|
|
body = resp.json()
|
|
assert body["source"] == "live"
|
|
ids = [m["id"] for m in body["models"]]
|
|
assert ids == ["qwen3:30b-a3b", "llama3.2:latest"]
|
|
assert captured["url"] == "http://ollama:11434/api/tags"
|
|
assert captured["headers"]["User-Agent"] == "osint-dashboard-news-summarizer"
|
|
assert "Authorization" not in captured["headers"]
|
|
|
|
|
|
def test_put_settings_empty_returns_422():
|
|
resp = _put("/api/settings", {"summary_model": ""})
|
|
assert resp.status_code == 422
|
|
|
|
|
|
def test_put_settings_whitespace_only_returns_422():
|
|
resp = _put("/api/settings", {"summary_model": " "})
|
|
assert resp.status_code == 422
|
|
|
|
|
|
@requires_db
|
|
def test_put_settings_round_trip(clean_settings, monkeypatch):
|
|
monkeypatch.delenv("SUMMARY_MODEL", raising=False)
|
|
monkeypatch.delenv("LLM_URL", raising=False)
|
|
|
|
put_resp = _put("/api/settings", {"summary_model": "qwen3:30b-a3b"})
|
|
assert put_resp.status_code == 200
|
|
put_body = put_resp.json()
|
|
assert put_body["summary_model"] == "qwen3:30b-a3b"
|
|
assert put_body["llm_url"] == "http://127.0.0.1:11434"
|
|
|
|
get_resp = _get("/api/settings")
|
|
assert get_resp.status_code == 200
|
|
get_body = get_resp.json()
|
|
assert get_body["summary_model"] == "qwen3:30b-a3b"
|
|
assert get_body["llm_url"] == "http://127.0.0.1:11434"
|
|
assert set(get_body.keys()) == {"summary_model", "llm_url"}
|