Kevin Veen-BirkenbachandClaude Opus 5.5
995c7cb214
The Day/Night menu lists every claimed agent with a checkbox and an All box; Drain, Suspend and Resume act on the ticked agents only, are disabled when none is ticked, and the agent popup carries the same three buttons for that one agent. Each action is passkey-signed with its agent list, night.py takes --target per session and keeps a per-agent part, and each chosen agent gets a call naming the action and its skill. The night skills say how to read a per-agent night. 🎯 Fix: 90 % 💥 Break: 5 % 💸 Cost: 106 % ⏱️ Time: 100 % 🛡️ Security: 100 % 🏆 Score: 81 🚦 Gate: ⚠️ cost 106 % > 100: operator approved the commit Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
152 lines
8.3 KiB
Python
152 lines
8.3 KiB
Python
"""Exercise night.py: drain, suspend, status per session, resume, states from before the rename, and day on a bad state."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[1]
|
|
NIGHT = REPO_ROOT / "skills" / "kvb-org-suspend" / "scripts" / "night.py"
|
|
|
|
|
|
class TestNight(unittest.TestCase):
|
|
def setUp(self):
|
|
tmp = tempfile.TemporaryDirectory()
|
|
self.addCleanup(tmp.cleanup)
|
|
self.home = Path(tmp.name) / "home"
|
|
self.root = Path(tmp.name) / "council"
|
|
sessions = self.home / ".claude" / "sessions"
|
|
sessions.mkdir(parents=True)
|
|
for sid, name in (("s1", "One"), ("s2", "Two"), ("s3", "Three")):
|
|
(sessions / f"{sid}.json").write_text(json.dumps({"sessionId": sid, "name": name}), encoding="utf-8")
|
|
|
|
def _run(self, *args: str, sid: str | None = "s1") -> subprocess.CompletedProcess:
|
|
env = {"PATH": "/usr/bin:/bin", "HOME": str(self.home), "AGENTS_COUNCIL_ROOT": str(self.root)}
|
|
if sid:
|
|
env["CLAUDE_CODE_SESSION_ID"] = sid
|
|
return subprocess.run(
|
|
[sys.executable, str(NIGHT), *args], capture_output=True, text=True, env=env, check=False
|
|
)
|
|
|
|
def _out(self, *args: str, sid: str | None = "s1") -> list[str]:
|
|
result = self._run(*args, sid=sid)
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
return result.stdout.splitlines()
|
|
|
|
def _state(self, state: dict) -> None:
|
|
self.root.mkdir(exist_ok=True)
|
|
(self.root / "night.json").write_text(json.dumps(state), encoding="utf-8")
|
|
|
|
def test_day_by_default(self):
|
|
self.assertEqual(self._out("status", sid="s2"), ["mode: day"])
|
|
|
|
def test_suspend_keeps_the_caller_active_and_suspends_the_others(self):
|
|
self.assertEqual(self._out("suspend"), ["s3\tThree", "s2\tTwo"])
|
|
lines = self._out("status", sid="s2")
|
|
self.assertEqual(lines[:2], ["mode: suspend", "active: One s1"])
|
|
self.assertRegex(lines[2], r"^since: \d{4}-\d\d-\d\dT")
|
|
self.assertEqual(lines[3], "you: suspend (kvb-agent-suspend)")
|
|
self.assertEqual(self._out("status")[3], "you: active")
|
|
self.assertEqual(self._out("status", "--session", "s3", sid=None)[3], "you: suspend (kvb-agent-suspend)")
|
|
self.assertEqual(len(self._out("status", sid=None)), 3)
|
|
|
|
def test_drain_and_resume_return_to_day(self):
|
|
self._out("drain", sid="s2")
|
|
self.assertEqual(json.loads((self.root / "night.json").read_text(encoding="utf-8"))["mode"], "drain")
|
|
self.assertEqual(self._out("status")[3], "you: drain (kvb-agent-drain)")
|
|
self.assertEqual(self._out("resume", sid="s2"), ["s1\tOne", "s3\tThree"])
|
|
self.assertEqual(self._out("status"), ["mode: day"])
|
|
|
|
def test_resume_alone_lifts_one_session_until_the_next_suspend(self):
|
|
self.assertEqual(self._out("resume", "--alone", sid="s2"), ["mode: day"])
|
|
self._out("suspend")
|
|
self.assertEqual(self._out("resume", "--alone", sid="s2"), ["resumed: s2"])
|
|
self._out("resume", "--alone", sid="s2")
|
|
self.assertEqual(json.loads((self.root / "night.json").read_text(encoding="utf-8"))["resumed"], ["s2"])
|
|
self.assertEqual(self._out("status", sid="s2")[3], "you: resumed")
|
|
self.assertEqual(self._out("status", sid="s3")[3], "you: suspend (kvb-agent-suspend)")
|
|
self._out("suspend")
|
|
self.assertEqual(self._out("status", sid="s2")[3], "you: suspend (kvb-agent-suspend)")
|
|
|
|
def test_a_state_from_before_the_rename_keeps_working(self):
|
|
for old, new in (("evening", "drain"), ("sunset", "suspend")):
|
|
with self.subTest(old=old):
|
|
self._state({"mode": old, "active": "s1", "name": "One", "since": "2026-11-08T20:00:00+00:00", "awake": ["s3"]})
|
|
self.assertEqual(self._out("status", sid="s2"), ["mode: " + new, "active: One s1", "since: 2026-11-08T20:00:00+00:00", f"you: {new} (kvb-agent-{new})"])
|
|
self.assertEqual(self._out("status", sid="s3")[3], "you: resumed")
|
|
self.assertEqual(self._out("resume", "--alone", sid="s2"), ["resumed: s2"])
|
|
state = json.loads((self.root / "night.json").read_text(encoding="utf-8"))
|
|
self.assertEqual((state["mode"], state["resumed"], "awake" in state), (new, ["s3", "s2"], False))
|
|
self._state({"mode": "sunset", "active": "s1"})
|
|
self.assertEqual(self._out("resume", sid="s2"), ["s1\tOne", "s3\tThree"])
|
|
self.assertFalse((self.root / "night.json").exists())
|
|
|
|
def test_targets_take_a_part_alone_by_day_and_resume_lifts_only_theirs(self):
|
|
self.assertEqual(self._out("drain", "--target=s2", "--target=s3"), ["s3\tThree", "s2\tTwo"])
|
|
self._out("suspend", "--target=s3")
|
|
self.assertEqual(json.loads((self.root / "night.json").read_text(encoding="utf-8")), {"parts": {"s2": "drain", "s3": "suspend"}})
|
|
self.assertEqual(self._out("status", sid="s2"), ["mode: day", "you: drain (kvb-agent-drain)"])
|
|
self.assertEqual(self._out("status", sid="s3"), ["mode: day", "you: suspend (kvb-agent-suspend)"])
|
|
self.assertEqual(self._out("status"), ["mode: day"])
|
|
self.assertEqual(self._out("resume", "--target=s2"), ["s2\tTwo"])
|
|
self.assertEqual(self._out("status", sid="s2"), ["mode: day"])
|
|
self.assertEqual(self._out("resume", "--alone", sid="s3"), ["resumed: s3"])
|
|
self.assertFalse((self.root / "night.json").exists())
|
|
|
|
def test_a_part_outranks_the_mode_and_a_targeted_resume_joins_the_resumed(self):
|
|
self._out("drain")
|
|
self._out("suspend", "--target=s2")
|
|
state = json.loads((self.root / "night.json").read_text(encoding="utf-8"))
|
|
self.assertEqual((state["mode"], state["active"], state["parts"]), ("drain", "s1", {"s2": "suspend"}))
|
|
self.assertEqual(self._out("status", sid="s2")[3], "you: suspend (kvb-agent-suspend)")
|
|
self.assertEqual(self._out("status", sid="s3")[3], "you: drain (kvb-agent-drain)")
|
|
self._out("resume", "--target=s2", "--target=s3", "--target=s1")
|
|
state = json.loads((self.root / "night.json").read_text(encoding="utf-8"))
|
|
self.assertEqual((state["mode"], state["parts"], state["resumed"]), ("drain", {}, ["s2", "s3"]))
|
|
self.assertEqual([self._out("status", sid=sid)[3] for sid in ("s1", "s2", "s3")], ["you: active", "you: resumed", "you: resumed"])
|
|
self._out("suspend", "--target=s3")
|
|
self._out("resume", sid="s2")
|
|
self.assertFalse((self.root / "night.json").exists())
|
|
|
|
def test_a_target_never_parses_as_an_option(self):
|
|
self._out("drain", "--target=--alone")
|
|
self.assertEqual(json.loads((self.root / "night.json").read_text(encoding="utf-8")), {"parts": {"--alone": "drain"}})
|
|
self.assertEqual(self._run("resume", "--alone", "--target=s2").returncode, 2)
|
|
|
|
def test_bad_state_means_day(self):
|
|
self.root.mkdir()
|
|
for text in (
|
|
"{not json",
|
|
json.dumps({"mode": "noon", "active": "s1"}),
|
|
json.dumps({"mode": ["suspend"], "active": "s1"}),
|
|
json.dumps(["suspend"]),
|
|
json.dumps({"mode": "suspend", "active": "s1", "resumed": "s2"}),
|
|
json.dumps({"mode": "sunset", "active": "s1", "awake": "s2"}),
|
|
json.dumps({"parts": {}}),
|
|
json.dumps({"parts": {"s2": "noon"}}),
|
|
json.dumps({"parts": {"s2": ["drain"]}}),
|
|
json.dumps({"parts": ["s2"]}),
|
|
json.dumps({"mode": "drain", "active": "s1", "parts": {"s2": "evening"}}),
|
|
):
|
|
with self.subTest(text=text):
|
|
(self.root / "night.json").write_text(text, encoding="utf-8")
|
|
result = self._run("status", sid="s2")
|
|
self.assertEqual((result.returncode, result.stdout), (0, "mode: day\n"))
|
|
self.assertIn("treated as day", result.stderr)
|
|
|
|
def test_usage_errors_exit_2(self):
|
|
self.assertEqual(self._run("suspend", sid=None).returncode, 2)
|
|
self.assertEqual(self._run("resume", "--alone", sid=None).returncode, 2)
|
|
for old in (("set", "sunset"), ("sunset",), ("evening",), ("clear",), ("wake",), ("noon",)):
|
|
with self.subTest(command=old):
|
|
self.assertEqual(self._run(*old).returncode, 2)
|
|
self.assertFalse((self.root / "night.json").exists())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|