theirongolddev/amc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(state): add same-pane session deduplication

Browse Source 
Handle edge case where Claude's --resume creates an orphan session file
before resuming the original session, leaving two session files pointing
to the same Zellij pane.

The deduplication algorithm (_dedupe_same_pane_sessions) resolves
conflicts by preferring:
1. Sessions with context_usage (indicates actual conversation occurred)
2. Higher conversation_mtime_ns (more recent file activity)

When an orphan is identified, its session file is deleted from disk to
prevent re-discovery on subsequent state collection cycles.

Test coverage includes:
- Keeping session with context_usage over one without
- Keeping higher mtime when both have context_usage
- Keeping higher mtime when neither has context_usage
- Preserving sessions on different panes (no false positives)
- Single session per pane unchanged
- Sessions without pane info unchanged
- Handling non-numeric mtime values defensively

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
teernisse
2026-02-28 00:47:15 -05:00
parent 1fb4a82b39
commit 19a31a4620
2 changed files with 288 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

228
tests/test_state.py
View File

@@ -416,5 +416,233 @@ class TestCollectSessions(unittest.TestCase):
self.assertEqual(ids, ["alpha", "middle", "zebra"])
class TestDedupeSamePaneSessions(unittest.TestCase):
"""Tests for _dedupe_same_pane_sessions (--resume orphan handling)."""
def setUp(self):
self.handler = DummyStateHandler()
def test_keeps_session_with_context_usage(self):
"""When two sessions share a pane, keep the one with context_usage."""
sessions = [
{
"session_id": "orphan",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": None,
"conversation_mtime_ns": 1000,
},
{
"session_id": "real",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": {"current_tokens": 5000},
"conversation_mtime_ns": 900, # older but has context
},
]
with tempfile.TemporaryDirectory() as tmpdir:
sessions_dir = Path(tmpdir)
# Create dummy session files
for s in sessions:
(sessions_dir / f"{s['session_id']}.json").write_text("{}")
with patch.object(state_mod, "SESSIONS_DIR", sessions_dir):
result = self.handler._dedupe_same_pane_sessions(sessions)
self.assertEqual(len(result), 1)
self.assertEqual(result[0]["session_id"], "real")
def test_keeps_higher_mtime_when_both_have_context(self):
"""When both have context_usage, keep the one with higher mtime."""
sessions = [
{
"session_id": "older",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": {"current_tokens": 5000},
"conversation_mtime_ns": 1000,
},
{
"session_id": "newer",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": {"current_tokens": 6000},
"conversation_mtime_ns": 2000,
},
]
with tempfile.TemporaryDirectory() as tmpdir:
sessions_dir = Path(tmpdir)
for s in sessions:
(sessions_dir / f"{s['session_id']}.json").write_text("{}")
with patch.object(state_mod, "SESSIONS_DIR", sessions_dir):
result = self.handler._dedupe_same_pane_sessions(sessions)
self.assertEqual(len(result), 1)
self.assertEqual(result[0]["session_id"], "newer")
def test_no_dedup_when_different_panes(self):
"""Sessions in different panes should not be deduped."""
sessions = [
{
"session_id": "session1",
"zellij_session": "infra",
"zellij_pane": "42",
},
{
"session_id": "session2",
"zellij_session": "infra",
"zellij_pane": "43", # different pane
},
]
with tempfile.TemporaryDirectory() as tmpdir:
with patch.object(state_mod, "SESSIONS_DIR", Path(tmpdir)):
result = self.handler._dedupe_same_pane_sessions(sessions)
self.assertEqual(len(result), 2)
def test_no_dedup_when_empty_pane_info(self):
"""Sessions without pane info should not be deduped."""
sessions = [
{"session_id": "session1", "zellij_session": "", "zellij_pane": ""},
{"session_id": "session2", "zellij_session": "", "zellij_pane": ""},
]
with tempfile.TemporaryDirectory() as tmpdir:
with patch.object(state_mod, "SESSIONS_DIR", Path(tmpdir)):
result = self.handler._dedupe_same_pane_sessions(sessions)
self.assertEqual(len(result), 2)
def test_deletes_orphan_session_file(self):
"""Orphan session file should be deleted from disk."""
sessions = [
{
"session_id": "orphan",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": None,
},
{
"session_id": "real",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": {"current_tokens": 5000},
},
]
with tempfile.TemporaryDirectory() as tmpdir:
sessions_dir = Path(tmpdir)
orphan_file = sessions_dir / "orphan.json"
real_file = sessions_dir / "real.json"
orphan_file.write_text("{}")
real_file.write_text("{}")
with patch.object(state_mod, "SESSIONS_DIR", sessions_dir):
self.handler._dedupe_same_pane_sessions(sessions)
self.assertFalse(orphan_file.exists())
self.assertTrue(real_file.exists())
def test_handles_session_without_id(self):
"""Sessions without session_id should be skipped, not cause errors."""
sessions = [
{
"session_id": None, # Missing ID
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": None,
},
{
"session_id": "real",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": {"current_tokens": 5000},
},
]
with tempfile.TemporaryDirectory() as tmpdir:
sessions_dir = Path(tmpdir)
(sessions_dir / "real.json").write_text("{}")
with patch.object(state_mod, "SESSIONS_DIR", sessions_dir):
result = self.handler._dedupe_same_pane_sessions(sessions)
# Should keep real, skip the None-id session (not filter it out)
self.assertEqual(len(result), 2) # Both still in list
self.assertIn("real", [s.get("session_id") for s in result])
def test_keeps_only_best_with_three_sessions(self):
"""When 3+ sessions share a pane, keep only the best one."""
sessions = [
{
"session_id": "worst",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": None,
"conversation_mtime_ns": 1000,
},
{
"session_id": "middle",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": None,
"conversation_mtime_ns": 2000,
},
{
"session_id": "best",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": {"current_tokens": 100},
"conversation_mtime_ns": 500, # Oldest but has context
},
]
with tempfile.TemporaryDirectory() as tmpdir:
sessions_dir = Path(tmpdir)
for s in sessions:
(sessions_dir / f"{s['session_id']}.json").write_text("{}")
with patch.object(state_mod, "SESSIONS_DIR", sessions_dir):
result = self.handler._dedupe_same_pane_sessions(sessions)
self.assertEqual(len(result), 1)
self.assertEqual(result[0]["session_id"], "best")
def test_handles_non_numeric_mtime(self):
"""Non-numeric mtime values should be treated as 0."""
sessions = [
{
"session_id": "bad_mtime",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": None,
"conversation_mtime_ns": "not a number", # Invalid
},
{
"session_id": "good_mtime",
"zellij_session": "infra",
"zellij_pane": "42",
"context_usage": None,
"conversation_mtime_ns": 1000,
},
]
with tempfile.TemporaryDirectory() as tmpdir:
sessions_dir = Path(tmpdir)
for s in sessions:
(sessions_dir / f"{s['session_id']}.json").write_text("{}")
with patch.object(state_mod, "SESSIONS_DIR", sessions_dir):
# Should not raise, should keep session with valid mtime
result = self.handler._dedupe_same_pane_sessions(sessions)
self.assertEqual(len(result), 1)
self.assertEqual(result[0]["session_id"], "good_mtime")
if __name__ == "__main__":
unittest.main()