gramps
90d2cf8326
- jarvisChat/JarvisChat/jarvischat → cAIc/cAIc/caic (branded/lower) - JARVISCHAT_ env vars → CAIC_ - jc- script/config prefix → caic- - jarvis_rag → caic_rag - jarvischat.db / volumes → caic.db / caic_* - AMQP vhost/user jarvischat → caic - Syslog, loggers, docstrings all updated - 47 files, zero stale references, 148 tests pass
129 lines
4.9 KiB
Python
129 lines
4.9 KiB
Python
import os
|
|
from pathlib import Path
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
import app
|
|
import db
|
|
from security import SESSIONS, PIN_ATTEMPTS, RATE_EVENTS
|
|
|
|
|
|
def make_client(tmp_path: Path) -> TestClient:
|
|
os.environ["CAIC_ADMIN_PIN"] = "1234"
|
|
db.DB_PATH = tmp_path / "caic-presets.db"
|
|
SESSIONS.clear()
|
|
PIN_ATTEMPTS.clear()
|
|
RATE_EVENTS.clear()
|
|
db.init_db()
|
|
return TestClient(app.app, raise_server_exceptions=False)
|
|
|
|
|
|
def _admin_headers(client: TestClient) -> dict:
|
|
login = client.post("/api/auth/login", json={"pin": "1234"}, headers={"Origin": "http://testserver"})
|
|
sid = login.json()["session_id"]
|
|
return {"X-Session-ID": sid, "Origin": "http://testserver"}
|
|
|
|
|
|
def _guest_headers(client: TestClient) -> dict:
|
|
sid = client.post("/api/auth/guest", headers={"Origin": "http://testserver"}).json()["session_id"]
|
|
return {"X-Session-ID": sid, "Origin": "http://testserver"}
|
|
|
|
|
|
def test_list_presets_returns_defaults(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
resp = client.get("/api/presets", headers=_guest_headers(client))
|
|
assert resp.status_code == 200
|
|
presets = resp.json()
|
|
assert len(presets) >= 3
|
|
names = [p["name"] for p in presets]
|
|
assert "Coding Companion" in names
|
|
|
|
|
|
def test_create_preset(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
headers = _admin_headers(client)
|
|
resp = client.post("/api/presets", json={"name": "My Preset", "prompt": "You are helpful."}, headers=headers)
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert data["name"] == "My Preset"
|
|
assert data["prompt"] == "You are helpful."
|
|
|
|
presets = client.get("/api/presets", headers=_guest_headers(client)).json()
|
|
assert any(p["name"] == "My Preset" for p in presets)
|
|
|
|
|
|
def test_create_preset_requires_name_and_prompt(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
headers = _admin_headers(client)
|
|
resp = client.post("/api/presets", json={"name": "", "prompt": ""}, headers=headers)
|
|
assert resp.status_code == 400
|
|
|
|
resp = client.post("/api/presets", json={"name": "Only Name"}, headers=headers)
|
|
assert resp.status_code == 400
|
|
|
|
|
|
def test_update_preset(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
headers = _admin_headers(client)
|
|
create = client.post("/api/presets", json={"name": "Old", "prompt": "Old prompt."}, headers=headers)
|
|
preset_id = create.json()["id"]
|
|
|
|
update = client.put(f"/api/presets/{preset_id}", json={"name": "New", "prompt": "New prompt."}, headers=headers)
|
|
assert update.status_code == 200
|
|
|
|
presets = client.get("/api/presets", headers=_guest_headers(client)).json()
|
|
updated = next(p for p in presets if p["id"] == preset_id)
|
|
assert updated["name"] == "New"
|
|
assert updated["prompt"] == "New prompt."
|
|
|
|
|
|
def test_update_preset_requires_fields(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
headers = _admin_headers(client)
|
|
resp = client.put("/api/presets/nope", json={"name": "", "prompt": ""}, headers=headers)
|
|
assert resp.status_code == 400
|
|
|
|
|
|
def test_delete_preset(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
headers = _admin_headers(client)
|
|
create = client.post("/api/presets", json={"name": "Temp", "prompt": "Temp."}, headers=headers)
|
|
preset_id = create.json()["id"]
|
|
|
|
delete = client.delete(f"/api/presets/{preset_id}", headers=headers)
|
|
assert delete.status_code == 200
|
|
|
|
presets = client.get("/api/presets", headers=_guest_headers(client)).json()
|
|
assert not any(p["id"] == preset_id for p in presets)
|
|
|
|
|
|
def test_delete_default_preset_is_noop(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
headers = _admin_headers(client)
|
|
presets_before = client.get("/api/presets", headers=_guest_headers(client)).json()
|
|
default = next(p for p in presets_before if p["is_default"])
|
|
|
|
delete = client.delete(f"/api/presets/{default['id']}", headers=headers)
|
|
assert delete.status_code == 200
|
|
|
|
presets_after = client.get("/api/presets", headers=_guest_headers(client)).json()
|
|
assert any(p["id"] == default["id"] for p in presets_after)
|
|
|
|
|
|
def test_guest_cannot_create_preset(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
resp = client.post("/api/presets", json={"name": "Hack", "prompt": "Hack"}, headers=_guest_headers(client))
|
|
assert resp.status_code == 403
|
|
|
|
|
|
def test_guest_cannot_update_preset(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
resp = client.put("/api/presets/some-id", json={"name": "Hack", "prompt": "Hack"}, headers=_guest_headers(client))
|
|
assert resp.status_code == 403
|
|
|
|
|
|
def test_guest_cannot_delete_preset(tmp_path: Path):
|
|
with make_client(tmp_path) as client:
|
|
resp = client.delete("/api/presets/some-id", headers=_guest_headers(client))
|
|
assert resp.status_code == 403
|