217 lines
8.0 KiB
Python
217 lines
8.0 KiB
Python
"""
|
|
Sprint 12 Tests: settings panel, session pinning, session import, SSE reconnect.
|
|
"""
|
|
import json, pathlib, urllib.error, urllib.request, urllib.parse
|
|
|
|
from tests._pytest_port import BASE, TEST_DEFAULT_MODEL
|
|
|
|
|
|
def get(path):
|
|
with urllib.request.urlopen(BASE + path, timeout=10) as r:
|
|
return json.loads(r.read()), r.status
|
|
|
|
|
|
def post(path, body=None):
|
|
data = json.dumps(body or {}).encode()
|
|
req = urllib.request.Request(BASE + path, data=data,
|
|
headers={"Content-Type": "application/json"})
|
|
try:
|
|
with urllib.request.urlopen(req, timeout=10) as r:
|
|
return json.loads(r.read()), r.status
|
|
except urllib.error.HTTPError as e:
|
|
return json.loads(e.read()), e.code
|
|
|
|
|
|
def make_session(created_list):
|
|
d, _ = post("/api/session/new", {})
|
|
sid = d["session"]["session_id"]
|
|
created_list.append(sid)
|
|
return sid
|
|
|
|
|
|
def _make_session_visible(sid):
|
|
from api.models import Session
|
|
|
|
session = Session.load(sid)
|
|
assert session is not None
|
|
session.messages = [{"role": "user", "content": "visible row", "timestamp": 1.0}]
|
|
session.save()
|
|
|
|
|
|
# ── Settings API ──────────────────────────────────────────────────────────
|
|
|
|
def test_settings_get_returns_defaults():
|
|
"""GET /api/settings returns default settings."""
|
|
d, status = get("/api/settings")
|
|
assert status == 200
|
|
assert 'default_model' in d
|
|
assert 'default_workspace' in d
|
|
|
|
def test_default_model_updates_hermes_config():
|
|
"""POST /api/default-model updates the effective Hermes default model.
|
|
|
|
As of #895 the endpoint returns a lightweight ack {ok, model} rather than
|
|
the full model catalog, to avoid triggering a blocking live-provider fetch
|
|
on every Settings save. The default model is verified via /api/settings.
|
|
"""
|
|
try:
|
|
d, status = post("/api/default-model", {"model": "anthropic/claude-sonnet-4.6"})
|
|
assert status == 200
|
|
# Lightweight ack — no longer the full catalog
|
|
assert d.get("ok") is True, f"expected ok=True, got {d}"
|
|
assert 'claude-sonnet-4.6' in d.get("model", ""), (
|
|
f"response model field should echo the saved model: {d}"
|
|
)
|
|
# Verify the setting actually persisted
|
|
d2, _ = get("/api/settings")
|
|
assert 'claude-sonnet-4.6' in d2['default_model']
|
|
finally:
|
|
post("/api/default-model", {"model": TEST_DEFAULT_MODEL})
|
|
|
|
|
|
def test_settings_does_not_persist_default_model():
|
|
"""POST /api/settings with default_model in body is silently ignored."""
|
|
d1, _ = get("/api/settings")
|
|
original_model = d1['default_model']
|
|
# Send default_model via /api/settings — it must be dropped (not persisted)
|
|
post("/api/settings", {"default_model": "openai/fake-model-xyz"})
|
|
d2, _ = get("/api/settings")
|
|
assert d2['default_model'] == original_model, (
|
|
"POST /api/settings must not persist default_model — use /api/default-model instead"
|
|
)
|
|
|
|
|
|
def test_default_model_empty_returns_400():
|
|
"""POST /api/default-model with empty model returns 400."""
|
|
d, status = post("/api/default-model", {"model": ""})
|
|
assert status == 400
|
|
|
|
def test_settings_partial_update():
|
|
"""POST /api/settings with partial data doesn't clobber other fields."""
|
|
d1, _ = get("/api/settings")
|
|
original_ws = d1['default_workspace']
|
|
post("/api/settings", {"send_key": "ctrl+enter"})
|
|
d2, _ = get("/api/settings")
|
|
assert d2['send_key'] == 'ctrl+enter'
|
|
assert d2['default_workspace'] == original_ws
|
|
post("/api/settings", {"send_key": "enter"})
|
|
|
|
|
|
# ── Session Pinning ───────────────────────────────────────────────────────
|
|
|
|
def test_pin_session():
|
|
"""POST /api/session/pin sets pinned=true."""
|
|
created = []
|
|
try:
|
|
sid = make_session(created)
|
|
d, status = post("/api/session/pin", {"session_id": sid, "pinned": True})
|
|
assert status == 200
|
|
assert d['ok'] is True
|
|
assert d['session']['pinned'] is True
|
|
finally:
|
|
for sid in created:
|
|
post("/api/session/delete", {"session_id": sid})
|
|
|
|
def test_unpin_session():
|
|
"""POST /api/session/pin with pinned=false unpins."""
|
|
created = []
|
|
try:
|
|
sid = make_session(created)
|
|
post("/api/session/pin", {"session_id": sid, "pinned": True})
|
|
d, status = post("/api/session/pin", {"session_id": sid, "pinned": False})
|
|
assert status == 200
|
|
assert d['session']['pinned'] is False
|
|
finally:
|
|
for sid in created:
|
|
post("/api/session/delete", {"session_id": sid})
|
|
|
|
def test_pinned_in_session_list():
|
|
"""Pinned sessions include pinned field in session list."""
|
|
created = []
|
|
try:
|
|
sid = make_session(created)
|
|
# Give it one real message so the default sidebar route keeps it visible.
|
|
post("/api/session/rename", {"session_id": sid, "title": "Pinned Test"})
|
|
_make_session_visible(sid)
|
|
post("/api/session/pin", {"session_id": sid, "pinned": True})
|
|
d, _ = get("/api/sessions")
|
|
match = [s for s in d['sessions'] if s['session_id'] == sid]
|
|
assert len(match) == 1
|
|
assert match[0]['pinned'] is True
|
|
finally:
|
|
for sid in created:
|
|
post("/api/session/delete", {"session_id": sid})
|
|
|
|
def test_pinned_persists_on_reload():
|
|
"""Pin status survives session reload from disk."""
|
|
created = []
|
|
try:
|
|
sid = make_session(created)
|
|
post("/api/session/pin", {"session_id": sid, "pinned": True})
|
|
d, _ = get(f"/api/session?session_id={sid}")
|
|
assert d['session']['pinned'] is True
|
|
finally:
|
|
for sid in created:
|
|
post("/api/session/delete", {"session_id": sid})
|
|
|
|
|
|
# ── Session Import ────────────────────────────────────────────────────────
|
|
|
|
def test_import_session_basic():
|
|
"""POST /api/session/import creates a new session from JSON."""
|
|
payload = {
|
|
"title": "Imported Test",
|
|
"messages": [
|
|
{"role": "user", "content": "Hello from import"},
|
|
{"role": "assistant", "content": "Hi there!"},
|
|
],
|
|
"model": "test/import-model",
|
|
}
|
|
d, status = post("/api/session/import", payload)
|
|
assert status == 200
|
|
assert d['ok'] is True
|
|
sid = d['session']['session_id']
|
|
try:
|
|
assert d['session']['title'] == 'Imported Test'
|
|
assert len(d['session']['messages']) == 2
|
|
# Verify it loads correctly
|
|
d2, _ = get(f"/api/session?session_id={sid}")
|
|
assert d2['session']['model'] == 'test/import-model'
|
|
finally:
|
|
post("/api/session/delete", {"session_id": sid})
|
|
|
|
def test_import_requires_messages():
|
|
"""Import fails without a messages array."""
|
|
d, status = post("/api/session/import", {"title": "No messages"})
|
|
assert status == 400
|
|
|
|
def test_import_creates_new_id():
|
|
"""Imported session gets a new session_id, not reusing any from the payload."""
|
|
payload = {
|
|
"session_id": "should_be_ignored",
|
|
"title": "ID Test",
|
|
"messages": [{"role": "user", "content": "test"}],
|
|
}
|
|
d, _ = post("/api/session/import", payload)
|
|
sid = d['session']['session_id']
|
|
try:
|
|
# The import should create a new ID, not use the one from the payload
|
|
assert sid != "should_be_ignored"
|
|
finally:
|
|
post("/api/session/delete", {"session_id": sid})
|
|
|
|
def test_import_with_pinned():
|
|
"""Imported session can be pinned."""
|
|
payload = {
|
|
"title": "Pinned Import",
|
|
"messages": [{"role": "user", "content": "test"}],
|
|
"pinned": True,
|
|
}
|
|
d, _ = post("/api/session/import", payload)
|
|
sid = d['session']['session_id']
|
|
try:
|
|
d2, _ = get(f"/api/session?session_id={sid}")
|
|
assert d2['session']['pinned'] is True
|
|
finally:
|
|
post("/api/session/delete", {"session_id": sid})
|