AIOSAIandClaude Opus 4.8
f460cd577e
One live Claude runtime per branch. presence.py manages .ai_central/PRESENCE.central.json (claim/release/refresh, PID + /proc/cwd liveness, stale-reclaim, PID-guarded release so a non-holder can never release the holder). presence_gate.py: UserPromptSubmit blocks a duplicate (exit 2 + decision:block), Stop releases; skips sub-agents + dispatched/daemon. SessionStart can't block in Claude Code (inject-only) → gate is UserPromptSubmit, like the edit/git gates. NOT wired into hooks.json yet — dormant, zero behavior change until enabled. 705 tests pass, seedgo 100%. Live cross-process block + PID-guard verified (devpulse). Design: DPLAN-0225. Next: P2 telegram relay follows the pointer (@skills). Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01GFihce1oLtp6UDAPGryYSv
405 lines
15 KiB
Python
405 lines
15 KiB
Python
"""Tests for the presence service module."""
|
|
|
|
import json
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from aipass.hooks.apps.modules import presence
|
|
|
|
|
|
@pytest.fixture
|
|
def presence_dir(tmp_path):
|
|
"""Create a temporary .ai_central directory with PRESENCE.central.json."""
|
|
ai_central = tmp_path / ".ai_central"
|
|
ai_central.mkdir()
|
|
return ai_central
|
|
|
|
|
|
@pytest.fixture
|
|
def presence_file(presence_dir):
|
|
"""Return path to the presence file."""
|
|
return presence_dir / "PRESENCE.central.json"
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_paths(presence_dir):
|
|
"""Patch _find_ai_central to use the temp directory."""
|
|
with patch.object(presence, "_find_ai_central", return_value=presence_dir):
|
|
yield presence_dir
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_flock():
|
|
"""Patch flock to be a no-op (avoid real file locking in tests)."""
|
|
with patch("aipass.hooks.apps.modules.presence.fcntl") as mock_fcntl:
|
|
mock_fcntl.LOCK_EX = 2
|
|
mock_fcntl.LOCK_UN = 8
|
|
yield mock_fcntl
|
|
|
|
|
|
# ── claim tests ──────────────────────────────────────────────────────────
|
|
|
|
|
|
class TestClaim:
|
|
def test_claim_empty_file(self, patch_paths, patch_flock, presence_file):
|
|
with patch("os.getpid", return_value=1000), patch("os.getcwd", return_value="/tmp/branch"):
|
|
result = presence.claim("devpulse", session_id="abc")
|
|
assert result["status"] == "ACQUIRED"
|
|
data = json.loads(presence_file.read_text())
|
|
assert data["devpulse"]["pid"] == 1000
|
|
assert data["devpulse"]["session_id"] == "abc"
|
|
assert data["devpulse"]["work_dir"] == "/tmp/branch"
|
|
|
|
def test_claim_reentry_same_pid(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"devpulse": {
|
|
"pid": 1000,
|
|
"session_id": "old",
|
|
"work_dir": "/w",
|
|
"session_type": "interactive",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
}
|
|
}
|
|
)
|
|
)
|
|
with patch("os.getpid", return_value=1000), patch("os.getcwd", return_value="/w"):
|
|
result = presence.claim("devpulse", session_id="new-id")
|
|
assert result["status"] == "ACQUIRED"
|
|
data = json.loads(presence_file.read_text())
|
|
assert data["devpulse"]["session_id"] == "new-id"
|
|
|
|
def test_claim_stale_dead_pid(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"devpulse": {
|
|
"pid": 9999,
|
|
"session_id": "old",
|
|
"work_dir": "/w",
|
|
"session_type": "interactive",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
}
|
|
}
|
|
)
|
|
)
|
|
with (
|
|
patch("os.getpid", return_value=2000),
|
|
patch("os.getcwd", return_value="/w2"),
|
|
patch.object(presence, "_is_pid_alive", return_value=False),
|
|
):
|
|
result = presence.claim("devpulse", session_id="new")
|
|
assert result["status"] == "ACQUIRED"
|
|
data = json.loads(presence_file.read_text())
|
|
assert data["devpulse"]["pid"] == 2000
|
|
|
|
def test_claim_occupied_live_pid(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"devpulse": {
|
|
"pid": 5000,
|
|
"session_id": "live",
|
|
"work_dir": "/w",
|
|
"session_type": "interactive",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
}
|
|
}
|
|
)
|
|
)
|
|
with (
|
|
patch("os.getpid", return_value=6000),
|
|
patch("os.getcwd", return_value="/w2"),
|
|
patch.object(presence, "_is_pid_alive", return_value=True),
|
|
patch.object(presence, "_cwd_matches", return_value=True),
|
|
):
|
|
result = presence.claim("devpulse", session_id="new")
|
|
assert result["status"] == "OCCUPIED"
|
|
assert result["pid"] == 5000
|
|
assert result["session_type"] == "interactive"
|
|
|
|
def test_claim_stale_cwd_mismatch(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"devpulse": {
|
|
"pid": 5000,
|
|
"session_id": "old",
|
|
"work_dir": "/original",
|
|
"session_type": "interactive",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
}
|
|
}
|
|
)
|
|
)
|
|
with (
|
|
patch("os.getpid", return_value=6000),
|
|
patch("os.getcwd", return_value="/new"),
|
|
patch.object(presence, "_is_pid_alive", return_value=True),
|
|
patch.object(presence, "_cwd_matches", return_value=False),
|
|
):
|
|
result = presence.claim("devpulse", session_id="new")
|
|
assert result["status"] == "ACQUIRED"
|
|
|
|
def test_claim_no_existing_file(self, patch_paths, patch_flock):
|
|
with patch("os.getpid", return_value=1000), patch("os.getcwd", return_value="/w"):
|
|
result = presence.claim("hooks")
|
|
assert result["status"] == "ACQUIRED"
|
|
|
|
def test_claim_multiple_branches(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"api": {
|
|
"pid": 3000,
|
|
"session_id": "a",
|
|
"work_dir": "/api",
|
|
"session_type": "interactive-mirror",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
}
|
|
}
|
|
)
|
|
)
|
|
with patch("os.getpid", return_value=4000), patch("os.getcwd", return_value="/hooks"):
|
|
result = presence.claim("hooks", session_id="h1")
|
|
assert result["status"] == "ACQUIRED"
|
|
data = json.loads(presence_file.read_text())
|
|
assert "api" in data
|
|
assert "hooks" in data
|
|
|
|
|
|
# ── release tests ────────────────────────────────────────────────────────
|
|
|
|
|
|
class TestRelease:
|
|
def test_release_existing(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"devpulse": {
|
|
"pid": 1000,
|
|
"session_id": "a",
|
|
"work_dir": "/w",
|
|
"session_type": "interactive",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
}
|
|
}
|
|
)
|
|
)
|
|
with patch("os.getpid", return_value=1000):
|
|
result = presence.release("devpulse")
|
|
assert result is True
|
|
data = json.loads(presence_file.read_text())
|
|
assert "devpulse" not in data
|
|
|
|
def test_release_not_claimed(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(json.dumps({}))
|
|
result = presence.release("devpulse")
|
|
assert result is False
|
|
|
|
def test_release_wrong_pid_refused(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"devpulse": {
|
|
"pid": 1000,
|
|
"session_id": "a",
|
|
"work_dir": "/w",
|
|
"session_type": "interactive",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
}
|
|
}
|
|
)
|
|
)
|
|
with patch("os.getpid", return_value=9999):
|
|
result = presence.release("devpulse")
|
|
assert result is False
|
|
data = json.loads(presence_file.read_text())
|
|
assert "devpulse" in data
|
|
assert data["devpulse"]["pid"] == 1000
|
|
|
|
def test_release_preserves_others(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"devpulse": {
|
|
"pid": 1000,
|
|
"session_id": "a",
|
|
"work_dir": "/w",
|
|
"session_type": "interactive",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
},
|
|
"api": {
|
|
"pid": 2000,
|
|
"session_id": "b",
|
|
"work_dir": "/api",
|
|
"session_type": "interactive-mirror",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
},
|
|
}
|
|
)
|
|
)
|
|
with patch("os.getpid", return_value=1000):
|
|
presence.release("devpulse")
|
|
data = json.loads(presence_file.read_text())
|
|
assert "api" in data
|
|
assert "devpulse" not in data
|
|
|
|
|
|
# ── refresh tests ────────────────────────────────────────────────────────
|
|
|
|
|
|
class TestRefresh:
|
|
def test_refresh_updates_last_seen(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(
|
|
json.dumps(
|
|
{
|
|
"hooks": {
|
|
"pid": 1000,
|
|
"session_id": "a",
|
|
"work_dir": "/w",
|
|
"session_type": "interactive",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
}
|
|
}
|
|
)
|
|
)
|
|
presence.refresh("hooks")
|
|
data = json.loads(presence_file.read_text())
|
|
assert data["hooks"]["last_seen"] != "2026-01-01T00:00:00"
|
|
|
|
def test_refresh_nonexistent_noop(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text(json.dumps({}))
|
|
presence.refresh("hooks")
|
|
data = json.loads(presence_file.read_text())
|
|
assert data == {}
|
|
|
|
|
|
# ── read_all tests ───────────────────────────────────────────────────────
|
|
|
|
|
|
class TestReadAll:
|
|
def test_read_all_returns_data(self, patch_paths, patch_flock, presence_file):
|
|
expected = {
|
|
"hooks": {
|
|
"pid": 1000,
|
|
"session_id": "a",
|
|
"work_dir": "/w",
|
|
"session_type": "interactive",
|
|
"attach_handle": "",
|
|
"started": "2026-01-01T00:00:00",
|
|
"last_seen": "2026-01-01T00:00:00",
|
|
}
|
|
}
|
|
presence_file.write_text(json.dumps(expected))
|
|
result = presence.read_all()
|
|
assert result == expected
|
|
|
|
def test_read_all_empty(self, patch_paths, patch_flock):
|
|
result = presence.read_all()
|
|
assert result == {}
|
|
|
|
|
|
# ── liveness tests ───────────────────────────────────────────────────────
|
|
|
|
|
|
class TestLiveness:
|
|
def test_is_pid_alive_true(self):
|
|
with patch("os.kill") as mock_kill:
|
|
assert presence._is_pid_alive(1234) is True
|
|
mock_kill.assert_called_once_with(1234, 0)
|
|
|
|
def test_is_pid_alive_dead(self):
|
|
with patch("os.kill", side_effect=ProcessLookupError):
|
|
assert presence._is_pid_alive(1234) is False
|
|
|
|
def test_is_pid_alive_permission_error(self):
|
|
with patch("os.kill", side_effect=PermissionError):
|
|
assert presence._is_pid_alive(1234) is True
|
|
|
|
def test_cwd_matches_linux(self):
|
|
with (
|
|
patch("sys.platform", "linux"),
|
|
patch("os.readlink", return_value="/tmp/project/src/aipass/hooks"),
|
|
):
|
|
assert presence._cwd_matches(1234, "/tmp/project/src/aipass/hooks") is True
|
|
|
|
def test_cwd_mismatch_linux(self):
|
|
with (
|
|
patch("sys.platform", "linux"),
|
|
patch("os.readlink", return_value="/tmp/project/src/aipass/api"),
|
|
):
|
|
assert presence._cwd_matches(1234, "/tmp/project/src/aipass/hooks") is False
|
|
|
|
def test_cwd_matches_non_linux_skips(self):
|
|
with patch("sys.platform", "win32"):
|
|
assert presence._cwd_matches(1234, "/anything") is True
|
|
|
|
def test_cwd_read_fails_treats_as_stale(self):
|
|
with (
|
|
patch("sys.platform", "linux"),
|
|
patch("os.readlink", side_effect=OSError("no proc")),
|
|
):
|
|
assert presence._cwd_matches(1234, "/w") is False
|
|
|
|
def test_is_holder_alive_full_check(self):
|
|
entry = {"pid": 1234, "work_dir": "/w"}
|
|
with (
|
|
patch.object(presence, "_is_pid_alive", return_value=True),
|
|
patch.object(presence, "_cwd_matches", return_value=True),
|
|
):
|
|
assert presence._is_holder_alive(entry) is True
|
|
|
|
def test_is_holder_alive_dead_pid(self):
|
|
entry = {"pid": 1234, "work_dir": "/w"}
|
|
with patch.object(presence, "_is_pid_alive", return_value=False):
|
|
assert presence._is_holder_alive(entry) is False
|
|
|
|
def test_is_holder_alive_no_pid(self):
|
|
assert presence._is_holder_alive({}) is False
|
|
|
|
def test_is_holder_alive_no_work_dir(self):
|
|
entry = {"pid": 1234, "work_dir": ""}
|
|
with patch.object(presence, "_is_pid_alive", return_value=True):
|
|
assert presence._is_holder_alive(entry) is False
|
|
|
|
|
|
# ── file locking tests ──────────────────────────────────────────────────
|
|
|
|
|
|
class TestFileLocking:
|
|
def test_presence_lock_acquires_flock(self, patch_paths):
|
|
with patch("aipass.hooks.apps.modules.presence.fcntl") as mock_fcntl:
|
|
mock_fcntl.LOCK_EX = 2
|
|
mock_fcntl.LOCK_UN = 8
|
|
with presence._presence_lock():
|
|
pass
|
|
mock_fcntl.flock.assert_called()
|
|
|
|
def test_corrupt_json_returns_empty(self, patch_paths, patch_flock, presence_file):
|
|
presence_file.write_text("not json {{{")
|
|
result = presence._read_presence()
|
|
assert result == {}
|