Stage 3 of the realism migration. Replaces orchestrator/scheduler.py's
hardcoded _FILE_TEMPLATES/_USERS (3 templates emitting epoch-suffixed
filenames like notes-1777315854.txt with identical bodies per
template) with a persona-driven realism engine.
New surface:
- SyntheticFile SQLModel (synthetic_files table, UNIQUE on
decky_uuid+path) — per-(decky, path) state for the future
edit-in-place flow. Pre-v1, no _migrate_* helper.
- BaseRepository methods: record_synthetic_file,
update_synthetic_file, list_synthetic_files,
pick_random_synthetic_file_for_edit (used by stage 3b).
- realism/naming.py: per-content-class filename templates,
persona-conditioned. /var/log/cron.log + logrotate skeleton for
system-class; /home/<persona>/TODO.md, scratch.md, etc. for
user-class. Anti-regression test pins "no 8+ digit decimals in
basenames" (the realism failure today).
- realism/bodies.py: deterministic body templates per content_class.
TODO body uses checkbox markdown, script body has a shebang, cron
body matches syslog cron shape ("CRON[PID]: (user) CMD (...)").
- realism/planner.py: pick(deckies, now, rng) returns a Plan.
Diurnal-gated, weighted user/system content split (70/30 user
bias). Create-only in stage 3; edit branch lands in stage 3b.
Scheduler split:
- scheduler.pick is now traffic-only (sync).
- scheduler.pick_file is async, takes a repo, resolves personas
(Topology.email_personas for topology-source deckies; global
realism.personas_pool otherwise), and maps Plan -> FileAction.
- FileAction gains persona/content_class/mtime fields.
Worker:
- _one_tick rolls 50/50 between traffic and file each tick. After a
successful FileAction plant, _record_synthetic_file persists or
patches the synthetic_files row (catching the unique-constraint
collision on re-plant of the same path).
- SSHDriver._run_file passes action.mtime through to plant_file so
files don't all stamp at wall-clock-now.
102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
"""Realism planner — picks (decky, persona, class, action, mtime).
|
|
|
|
Stage 3 ships create-only plans; the edit branch lands in 3b. Tests
|
|
pin the diurnal gate, the eligibility filter, and the create
|
|
contract.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import random
|
|
from datetime import datetime, timezone
|
|
|
|
import pytest
|
|
|
|
from decnet.realism.personas import EmailPersona
|
|
from decnet.realism.planner import pick
|
|
from decnet.realism.taxonomy import ContentClass
|
|
|
|
|
|
def _persona(name: str = "admin", window: str = "00:00-00:00") -> EmailPersona:
|
|
return EmailPersona(
|
|
name=name,
|
|
email=f"{name}@corp.com",
|
|
role="ops",
|
|
tone="direct",
|
|
active_hours=window,
|
|
)
|
|
|
|
|
|
def _decky(uuid: str = "u1", name: str = "decky-01", personas=None) -> dict:
|
|
return {
|
|
"uuid": uuid,
|
|
"name": name,
|
|
"_realism_personas": personas or [_persona()],
|
|
}
|
|
|
|
|
|
_NOW = datetime(2026, 4, 27, 14, 0, tzinfo=timezone.utc)
|
|
|
|
|
|
def test_pick_returns_none_when_no_deckies() -> None:
|
|
assert pick([], _NOW) is None
|
|
|
|
|
|
def test_pick_returns_none_when_decky_has_no_personas() -> None:
|
|
assert pick([{"uuid": "u1", "name": "d", "_realism_personas": []}], _NOW) is None
|
|
|
|
|
|
def test_pick_filters_personas_outside_window() -> None:
|
|
# A persona pegged to 01:00-02:00 with now=14:00 must not be picked.
|
|
out_of_hours = _persona(window="01:00-02:00")
|
|
deckies = [_decky(personas=[out_of_hours])]
|
|
assert pick(deckies, _NOW) is None
|
|
|
|
|
|
def test_pick_returns_create_plan_with_mtime_in_past() -> None:
|
|
deckies = [_decky()]
|
|
plan = pick(deckies, _NOW, rand=random.Random(0))
|
|
assert plan is not None
|
|
assert plan.action == "create"
|
|
assert plan.decky_uuid == "u1"
|
|
assert plan.persona == "admin"
|
|
assert plan.target_path.startswith("/")
|
|
assert plan.body_hint
|
|
assert plan.mtime < _NOW
|
|
|
|
|
|
def test_pick_distributes_across_user_and_system_classes() -> None:
|
|
deckies = [_decky()]
|
|
seen: set[ContentClass] = set()
|
|
for seed in range(80):
|
|
plan = pick(deckies, _NOW, rand=random.Random(seed))
|
|
if plan is not None:
|
|
seen.add(plan.content_class)
|
|
# Across 80 seeds we should hit both buckets — at least one user
|
|
# class and at least one system class — otherwise the weights or
|
|
# the 70/30 split is broken.
|
|
user_classes = {c for c in seen if c.is_user_class()}
|
|
system_classes = {c for c in seen if c.is_system_class()}
|
|
assert user_classes, f"no user-class plans in 80 trials: {seen}"
|
|
assert system_classes, f"no system-class plans in 80 trials: {seen}"
|
|
|
|
|
|
def test_pick_never_returns_canary_class_in_stage3() -> None:
|
|
deckies = [_decky()]
|
|
for seed in range(40):
|
|
plan = pick(deckies, _NOW, rand=random.Random(seed))
|
|
if plan is None:
|
|
continue
|
|
assert not plan.content_class.is_canary(), (
|
|
"canary class slipped into the realism planner; cultivator "
|
|
"lands in stage 7"
|
|
)
|
|
|
|
|
|
def test_pick_persists_persona_window_in_notes() -> None:
|
|
plan = pick([_decky()], _NOW, rand=random.Random(0))
|
|
assert plan is not None
|
|
# The plan's notes carry the persona name and the window — useful
|
|
# for the dashboard's "why this file" inspector.
|
|
assert any("persona=admin" in n for n in plan.notes)
|
|
assert any("window=" in n for n in plan.notes)
|