Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/hermes_state/test_resolve_resume_session_id.py

@@ -0,0 +1,104 @@
+"""Regression guard for #15000: --resume <id> after compression loses messages.
+
+Context compression ends the current session and forks a new child session
+(linked by ``parent_session_id``). The SQLite flush cursor is reset, so
+only the latest descendant ends up with rows in the ``messages`` table —
+the parent row has ``message_count = 0``. ``hermes --resume <parent_id>``
+used to load zero rows and show a blank chat.
+
+``SessionDB.resolve_resume_session_id()`` walks the parent → child chain
+and redirects to the first descendant that actually has messages. These
+tests pin that behaviour.
+"""
+import time
+
+import pytest
+
+from hermes_state import SessionDB
+
+
+@pytest.fixture
+def db(tmp_path):
+    return SessionDB(tmp_path / "state.db")
+
+
+def _make_chain(db: SessionDB, ids_with_parent):
+    """Create sessions in order, forcing started_at so ordering is deterministic."""
+    base = int(time.time()) - 10_000
+    for i, (sid, parent) in enumerate(ids_with_parent):
+        db.create_session(sid, source="cli", parent_session_id=parent)
+        db._conn.execute(
+            "UPDATE sessions SET started_at = ? WHERE id = ?",
+            (base + i * 100, sid),
+        )
+    db._conn.commit()
+
+
+def test_returns_self_when_only_parent_has_messages(db):
+    # When a session already has messages AND no descendant has messages,
+    # it should still be returned.  The chain walk finds no better candidate.
+    _make_chain(db, [("root", None), ("child", "root")])
+    db.append_message("root", role="user", content="hi")
+    assert db.resolve_resume_session_id("root") == "root"
+
+
+
+
+
+
+
+
+
+
+def test_walks_from_middle_of_chain(db):
+    # If the user happens to know an intermediate ID, we still find the msg-bearing descendant.
+    _make_chain(db, [("a", None), ("b", "a"), ("c", "b"), ("d", "c")])
+    db.append_message("d", role="user", content="x")
+    assert db.resolve_resume_session_id("b") == "d"
+    assert db.resolve_resume_session_id("c") == "d"
+
+
+def test_follows_compression_tip_when_parent_retains_messages(db):
+    # The bug behind the desktop "I came back and the reply isn't there" report
+    # on large sessions: auto-compression ends the live session and forks a
+    # continuation child, but a long parent keeps its own flushed message rows.
+    # The empty-head walk below never redirects a non-empty head, so resuming
+    # the parent id reloaded the pre-compression transcript and the response
+    # generated *after* compression (which lives in the continuation) was
+    # missing. resolve_resume_session_id must follow the compression-tip chain
+    # forward even when the parent still has messages.
+    base = int(time.time()) - 10_000
+    db.create_session("root", source="cli")
+    db.append_message("root", role="user", content="pre-compression turn")
+    db.end_session("root", "compression")
+    db.create_session("cont", source="cli", parent_session_id="root")
+    db.append_message("cont", role="assistant", content="post-compression reply")
+    # Force deterministic ordering so the continuation's started_at is clearly
+    # at/after the parent's ended_at (the get_compression_tip discriminator).
+    conn = db._conn
+    assert conn is not None
+    conn.execute("UPDATE sessions SET started_at = ?, ended_at = ? WHERE id = 'root'", (base, base + 50))
+    conn.execute("UPDATE sessions SET started_at = ? WHERE id = 'cont'", (base + 100,))
+    conn.commit()
+
+    assert db.resolve_resume_session_id("root") == "cont"
+
+
+
+
+def test_prefers_most_recent_child_when_fork_exists(db):
+    # If a session was somehow forked (two children), pick the latest one.
+    # In practice, compression only produces single-chain shape, but the helper
+    # should degrade gracefully.
+    _make_chain(db, [
+        ("parent", None),
+        ("older_fork", "parent"),
+        ("newer_fork", "parent"),
+    ])
+    db.append_message("newer_fork", role="user", content="x")
+    assert db.resolve_resume_session_id("parent") == "newer_fork"
+
+
+
+
+