feat(emailgen): global persona pool + Date-stamped EML mtimes
Two changes that unwind earlier MazeNET-only assumptions and fix a
realism tell:
1. Persona resolution is now per-decky-source, not topology-only. The
scheduler walks the union view (list_running_deckies, including
fleet MACVLAN/IPVLAN + SWARM shards) and picks the right persona
list for each source:
* topology decky -> Topology.email_personas (per-topology richness
preserved)
* fleet / shard -> a single host-wide pool loaded from disk
(DECNET_EMAILGEN_PERSONAS, /etc/decnet/email_personas.json, or
~/.decnet/email_personas.json)
Operators install the global pool via 'decnet emailgen
import-personas <file>' which validates with the same Pydantic
schema the worker uses.
2. The driver now runs 'touch -d <Date>' inside the docker exec right
after the EML write so file mtime matches the email's RFC 2822
Date: header. Without this an attacker 'ls -lt'ing the spool sees
every email clustered inside the worker's tick window — the
cluster itself was a stylometric tell.
CLI now exposes 'decnet emailgen' as a sub-app with 'run' (default,
backwards-compatible with bare 'decnet emailgen') and 'import-personas'.
list_running_deckies carries topology_id through so consumers can resolve
the parent topology without a second round-trip.
This commit is contained in:
129
tests/cli/test_emailgen_import_personas.py
Normal file
129
tests/cli/test_emailgen_import_personas.py
Normal file
@@ -0,0 +1,129 @@
|
||||
"""``decnet emailgen import-personas`` CLI command."""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
|
||||
import pytest
|
||||
from typer.testing import CliRunner
|
||||
|
||||
from decnet.cli import app
|
||||
from decnet.orchestrator.emailgen import global_pool
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def _reset_pool():
|
||||
global_pool.reset_cache()
|
||||
yield
|
||||
global_pool.reset_cache()
|
||||
|
||||
|
||||
_TWO = [
|
||||
{
|
||||
"name": "John Smith",
|
||||
"email": "john@corp.com",
|
||||
"role": "COO",
|
||||
"tone": "formal",
|
||||
"mannerisms": ["uses 'Best regards'"],
|
||||
},
|
||||
{
|
||||
"name": "Sarah Johnson",
|
||||
"email": "sarah@corp.com",
|
||||
"role": "PM",
|
||||
"tone": "direct",
|
||||
"mannerisms": ["uses bullets"],
|
||||
},
|
||||
]
|
||||
|
||||
|
||||
def test_import_personas_writes_canonical_file(tmp_path, monkeypatch):
|
||||
src = tmp_path / "src.json"
|
||||
src.write_text(json.dumps(_TWO))
|
||||
dest = tmp_path / "global_pool.json"
|
||||
monkeypatch.setenv("DECNET_EMAILGEN_PERSONAS", str(dest))
|
||||
|
||||
result = CliRunner().invoke(
|
||||
app, ["emailgen", "import-personas", str(src)]
|
||||
)
|
||||
assert result.exit_code == 0, result.stdout
|
||||
assert dest.exists()
|
||||
written = json.loads(dest.read_text())
|
||||
assert {p["email"] for p in written} == {"john@corp.com", "sarah@corp.com"}
|
||||
|
||||
|
||||
def test_import_personas_explicit_output_overrides_env(tmp_path, monkeypatch):
|
||||
src = tmp_path / "src.json"
|
||||
src.write_text(json.dumps(_TWO))
|
||||
env_dest = tmp_path / "env.json"
|
||||
explicit = tmp_path / "explicit.json"
|
||||
monkeypatch.setenv("DECNET_EMAILGEN_PERSONAS", str(env_dest))
|
||||
|
||||
result = CliRunner().invoke(
|
||||
app,
|
||||
["emailgen", "import-personas", str(src), "--output", str(explicit)],
|
||||
)
|
||||
assert result.exit_code == 0, result.stdout
|
||||
assert explicit.exists()
|
||||
assert not env_dest.exists()
|
||||
|
||||
|
||||
def test_import_personas_rejects_invalid_json(tmp_path):
|
||||
src = tmp_path / "src.json"
|
||||
src.write_text("{not valid")
|
||||
result = CliRunner().invoke(
|
||||
app, ["emailgen", "import-personas", str(src)]
|
||||
)
|
||||
assert result.exit_code != 0
|
||||
assert "Invalid JSON" in result.stdout
|
||||
|
||||
|
||||
def test_import_personas_rejects_non_list(tmp_path, monkeypatch):
|
||||
src = tmp_path / "src.json"
|
||||
src.write_text(json.dumps({"not": "a list"}))
|
||||
monkeypatch.setenv("DECNET_EMAILGEN_PERSONAS", str(tmp_path / "out.json"))
|
||||
result = CliRunner().invoke(
|
||||
app, ["emailgen", "import-personas", str(src)]
|
||||
)
|
||||
assert result.exit_code != 0
|
||||
assert "list" in result.stdout.lower()
|
||||
|
||||
|
||||
def test_import_personas_rejects_all_invalid_entries(tmp_path, monkeypatch):
|
||||
src = tmp_path / "src.json"
|
||||
src.write_text(json.dumps([
|
||||
{"name": "broken", "email": "no-at-symbol"},
|
||||
]))
|
||||
monkeypatch.setenv("DECNET_EMAILGEN_PERSONAS", str(tmp_path / "out.json"))
|
||||
result = CliRunner().invoke(
|
||||
app, ["emailgen", "import-personas", str(src)]
|
||||
)
|
||||
assert result.exit_code != 0
|
||||
assert "No valid personas" in result.stdout
|
||||
|
||||
|
||||
def test_import_personas_warns_on_single_persona(tmp_path, monkeypatch):
|
||||
src = tmp_path / "src.json"
|
||||
src.write_text(json.dumps(_TWO[:1]))
|
||||
dest = tmp_path / "out.json"
|
||||
monkeypatch.setenv("DECNET_EMAILGEN_PERSONAS", str(dest))
|
||||
result = CliRunner().invoke(
|
||||
app, ["emailgen", "import-personas", str(src)]
|
||||
)
|
||||
assert result.exit_code == 0, result.stdout
|
||||
assert "Warning" in result.stdout
|
||||
assert dest.exists()
|
||||
|
||||
|
||||
def test_imported_personas_load_via_global_pool(tmp_path, monkeypatch):
|
||||
src = tmp_path / "src.json"
|
||||
src.write_text(json.dumps(_TWO))
|
||||
dest = tmp_path / "out.json"
|
||||
monkeypatch.setenv("DECNET_EMAILGEN_PERSONAS", str(dest))
|
||||
|
||||
result = CliRunner().invoke(
|
||||
app, ["emailgen", "import-personas", str(src)]
|
||||
)
|
||||
assert result.exit_code == 0, result.stdout
|
||||
|
||||
personas = global_pool.load()
|
||||
assert len(personas) == 2
|
||||
assert {p.email for p in personas} == {"john@corp.com", "sarah@corp.com"}
|
||||
Reference in New Issue
Block a user