Joseph Doherty
211 lines
6.3 KiB
Python
211 lines
6.3 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from chat.app import app
|
|
from chat.db.connection import open_db
|
|
from chat.eventlog.log import append_event
|
|
from chat.eventlog.projector import project
|
|
|
|
|
|
@pytest.fixture
|
|
def client(tmp_path, monkeypatch):
|
|
cfg = tmp_path / "config.toml"
|
|
cfg.write_text('featherless_api_key = "test"\n')
|
|
monkeypatch.setenv("CHAT_CONFIG_PATH", str(cfg))
|
|
db = tmp_path / "test.db"
|
|
monkeypatch.setenv("CHAT_DB_PATH", str(db))
|
|
with TestClient(app) as c:
|
|
# Disable background worker (we won't drive turns).
|
|
if hasattr(app.state, "background_worker"):
|
|
app.state.background_worker.enabled = False
|
|
yield c
|
|
|
|
|
|
def _seed(db: Path) -> None:
|
|
with open_db(db) as conn:
|
|
append_event(
|
|
conn,
|
|
kind="bot_authored",
|
|
payload={
|
|
"id": "bot_a",
|
|
"name": "BotA",
|
|
"persona": "...",
|
|
"voice_samples": [],
|
|
"traits": ["shy"],
|
|
"backstory": "",
|
|
"initial_relationship_to_you": "",
|
|
"kickoff_prose": "",
|
|
},
|
|
)
|
|
append_event(
|
|
conn,
|
|
kind="you_authored",
|
|
payload={
|
|
"name": "Me",
|
|
"pronouns": "they/them",
|
|
"persona": "engineer",
|
|
},
|
|
)
|
|
append_event(
|
|
conn,
|
|
kind="chat_created",
|
|
payload={
|
|
"id": "chat_bot_a",
|
|
"host_bot_id": "bot_a",
|
|
"initial_time": "2026-04-26T20:00:00+00:00",
|
|
"narrative_anchor": "Day 1",
|
|
"weather": "",
|
|
},
|
|
)
|
|
# Activity for both you and host bot.
|
|
append_event(
|
|
conn,
|
|
kind="activity_change",
|
|
payload={
|
|
"entity_id": "you",
|
|
"posture": "sitting",
|
|
"action": {"verb": "thinking"},
|
|
"attention": "the screen",
|
|
},
|
|
)
|
|
append_event(
|
|
conn,
|
|
kind="activity_change",
|
|
payload={
|
|
"entity_id": "bot_a",
|
|
"posture": "standing",
|
|
"action": {"verb": "looking out the window"},
|
|
},
|
|
)
|
|
# Edge host -> you.
|
|
append_event(
|
|
conn,
|
|
kind="edge_update",
|
|
payload={
|
|
"source_id": "bot_a",
|
|
"target_id": "you",
|
|
"chat_id": "chat_bot_a",
|
|
"affinity_delta": 5,
|
|
"trust_delta": 2,
|
|
"knowledge_facts": ["Me likes coffee"],
|
|
},
|
|
)
|
|
# A regular memory.
|
|
append_event(
|
|
conn,
|
|
kind="memory_written",
|
|
payload={
|
|
"owner_id": "bot_a",
|
|
"chat_id": "chat_bot_a",
|
|
"pov_summary": "Talked about her sister",
|
|
"witness_you": 1,
|
|
"witness_host": 1,
|
|
"witness_guest": 0,
|
|
"significance": 2,
|
|
},
|
|
)
|
|
# A pinned memory with significance 3 (★★).
|
|
append_event(
|
|
conn,
|
|
kind="memory_written",
|
|
payload={
|
|
"owner_id": "bot_a",
|
|
"chat_id": "chat_bot_a",
|
|
"pov_summary": "First kiss",
|
|
"witness_you": 1,
|
|
"witness_host": 1,
|
|
"witness_guest": 0,
|
|
"significance": 3,
|
|
"pinned": 1,
|
|
"auto_pinned": 1,
|
|
},
|
|
)
|
|
project(conn)
|
|
|
|
|
|
def test_drawer_404_when_chat_missing(client):
|
|
response = client.get("/chats/no_such/drawer")
|
|
assert response.status_code == 404
|
|
|
|
|
|
def test_drawer_renders_scene_and_activity(client, tmp_path):
|
|
_seed(tmp_path / "test.db")
|
|
response = client.get("/chats/chat_bot_a/drawer")
|
|
assert response.status_code == 200
|
|
body = response.text
|
|
# Scene/time anchor.
|
|
assert "2026-04-26" in body
|
|
# Activity verbs from both entities.
|
|
assert "thinking" in body
|
|
assert "looking out the window" in body
|
|
# Activity attention.
|
|
assert "the screen" in body
|
|
|
|
|
|
def test_drawer_renders_edges(client, tmp_path):
|
|
_seed(tmp_path / "test.db")
|
|
response = client.get("/chats/chat_bot_a/drawer")
|
|
assert response.status_code == 200
|
|
body = response.text
|
|
assert "BotA" in body
|
|
assert "you" in body
|
|
# Default affinity 50 + delta 5 = 55.
|
|
assert "55" in body
|
|
# Knowledge fact appears.
|
|
assert "Me likes coffee" in body
|
|
|
|
|
|
def test_drawer_renders_memories_with_significance_markers(client, tmp_path):
|
|
_seed(tmp_path / "test.db")
|
|
response = client.get("/chats/chat_bot_a/drawer")
|
|
assert response.status_code == 200
|
|
body = response.text
|
|
assert "Talked about her sister" in body
|
|
assert "First kiss" in body
|
|
# Pinned counter shows 1 / 8 (or 1/8).
|
|
assert "1 / 8" in body or "1/8" in body
|
|
# Significance star marker for the pinned, score-3 memory.
|
|
assert "★" in body
|
|
|
|
|
|
def test_drawer_handles_no_state_gracefully(client, tmp_path):
|
|
db = tmp_path / "test.db"
|
|
with open_db(db) as conn:
|
|
# Just enough state for the chat to exist.
|
|
append_event(
|
|
conn,
|
|
kind="bot_authored",
|
|
payload={
|
|
"id": "bot_a",
|
|
"name": "BotA",
|
|
"persona": "...",
|
|
"voice_samples": [],
|
|
"traits": [],
|
|
"backstory": "",
|
|
"initial_relationship_to_you": "",
|
|
"kickoff_prose": "",
|
|
},
|
|
)
|
|
append_event(
|
|
conn,
|
|
kind="chat_created",
|
|
payload={
|
|
"id": "chat_bot_a",
|
|
"host_bot_id": "bot_a",
|
|
"initial_time": "2026-04-26T20:00:00+00:00",
|
|
"narrative_anchor": "Day 1",
|
|
"weather": "",
|
|
},
|
|
)
|
|
project(conn)
|
|
response = client.get("/chats/chat_bot_a/drawer")
|
|
assert response.status_code == 200
|
|
body = response.text
|
|
# Drawer renders gracefully with empty placeholders.
|
|
assert "No active container" in body or "Container:" not in body
|
|
assert "No edges yet" in body or "Edges" in body
|