77e770cdb2
Restructures persona storage from a flat personas/{name}/ layout to
home/{username}/persona/{name}/, mirroring Linux home directories.
Changes:
- persona.py: two ContextVars (user + persona), Linux-style name validation,
set_context(), get_user(), get_persona(), validate(), list_users(),
list_user_personas(); persona_path() takes (username, name)
- config.py: replaces personas_dir with home_dir + home_root()
- git mv personas/inara → home/scott/persona/inara (history preserved)
- home/holly/persona/tina/: Holly's persona stub added
- cron_runner.py: all storage functions take (username, persona) params
- tools/cron.py: stamps user + persona on jobs; APScheduler IDs are
{user}:{persona}:{job_id} to prevent collisions across users
- memory_distiller.py: distill_short/mid/long take (username, persona);
added missing Path + settings imports
- scheduler.py: _load_user_crons() iterates home/*/persona/* (two-level)
- routers/chat.py, orchestrator.py: user field added; set_context() called
- tests/conftest.py: home_root fixture with two-level structure;
patches home_dir instead of personas_dir
- tests/test_persona.py: fully rewritten for two-level API
- tests/test_api_files.py: updated fixture name and path
- .env.default: documents HOME_DIR setting; scrubs stale API key
- CLAUDE.md, README.md: directory maps updated for new layout
All 80 tests pass.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
"""
|
|
Tests for GET/PUT /files/* — allowed set enforcement, read/write, IDENTITY.md.
|
|
"""
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_files_list(client):
|
|
r = await client.get("/files")
|
|
assert r.status_code == 200
|
|
files = r.json()["files"]
|
|
names = [f["name"] for f in files]
|
|
assert "SOUL.md" in names
|
|
assert "IDENTITY.md" in names
|
|
assert "USER.md" in names
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_files_get_allowed(client):
|
|
r = await client.get("/files/IDENTITY.md")
|
|
assert r.status_code == 200
|
|
assert "content" in r.json()
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_files_get_not_in_allowed(client):
|
|
"""Files outside the ALLOWED set should return 404, not the file content."""
|
|
for name in ("TASKS.json", "CRONS.json", "SCRATCH.md", "../config.py", ".env"):
|
|
r = await client.get(f"/files/{name}")
|
|
assert r.status_code == 404, f"Expected 404 for {name}, got {r.status_code}"
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_files_put_and_get(client):
|
|
"""Write a new value and read it back."""
|
|
content = "# Updated PROTOCOLS\nTest content."
|
|
r = await client.put("/files/PROTOCOLS.md", json={"content": content})
|
|
assert r.status_code == 200
|
|
assert r.json()["ok"] is True
|
|
|
|
r2 = await client.get("/files/PROTOCOLS.md")
|
|
assert r2.status_code == 200
|
|
assert r2.json()["content"] == content
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_files_put_not_allowed(client):
|
|
r = await client.put("/files/../../etc/passwd", json={"content": "pwned"})
|
|
assert r.status_code == 404
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_files_get_missing_but_allowed(client, home_root):
|
|
"""An allowed file that doesn't exist yet returns 404."""
|
|
# Temporarily remove MEMORY_MID.md
|
|
f = home_root / "scott" / "persona" / "inara" / "MEMORY_MID.md"
|
|
existed = f.exists()
|
|
if existed:
|
|
backup = f.read_text()
|
|
f.unlink()
|
|
try:
|
|
r = await client.get("/files/MEMORY_MID.md")
|
|
assert r.status_code == 404
|
|
finally:
|
|
if existed:
|
|
f.write_text(backup)
|