fix(gateway): scope session lists before limiting

2026-07-16 10:43:24 +02:00
parent 75aeba09e0
commit fa9e967a2d
8 changed files with 573 additions and 24 deletions
@@ -56,3 +56,82 @@ class TestQuerySessionListingSearch:
        finally:
            db.close()

+
+class TestQuerySessionListingLaneScope:
+    @pytest.fixture
+    def db(self, tmp_path):
+        from hermes_state import SessionDB
+
+        db = SessionDB(db_path=tmp_path / "state.db")
+        lane_key = "agent:main:telegram:dm:lane"
+        db.create_session(
+            "lane_current", "telegram", session_key=lane_key,
+            user_id="lane-user", chat_id="lane",
+        )
+        db.set_session_title("lane_current", "Current lane")
+        db.create_session(
+            "lane_named", "telegram", session_key=lane_key,
+            user_id="lane-user", chat_id="lane",
+        )
+        db.set_session_title("lane_named", "Needle lane")
+        db.create_session(
+            "lane_unnamed", "telegram", session_key=lane_key,
+            user_id="lane-user", chat_id="lane",
+        )
+        for i in range(60):
+            db.create_session(
+                f"foreign_{i}", "telegram",
+                session_key=f"agent:main:telegram:dm:foreign-{i}",
+                user_id=f"foreign-user-{i}", chat_id=f"foreign-{i}",
+            )
+            db.set_session_title(f"foreign_{i}", f"Needle foreign {i}")
+        yield db, lane_key
+        db.close()
+
+    def test_exact_lane_precedes_limit_and_current_session_exclusion(self, db):
+        session_db, lane_key = db
+
+        rows = query_session_listing(
+            session_db,
+            source="telegram",
+            session_key=lane_key,
+            current_session_id="lane_current",
+            limit=1,
+        )
+
+        assert [row["id"] for row in rows] == ["lane_named"]
+
+    def test_exact_lane_preserves_full_and_search_modes(self, db):
+        session_db, lane_key = db
+
+        full_rows = query_session_listing(
+            session_db,
+            source="telegram",
+            session_key=lane_key,
+            include_unnamed=True,
+            limit=10,
+        )
+        search_rows = query_session_listing(
+            session_db,
+            source="telegram",
+            session_key=lane_key,
+            search_query="needle",
+            limit=10,
+        )
+
+        assert {row["id"] for row in full_rows} == {
+            "lane_current", "lane_named", "lane_unnamed",
+        }
+        assert [row["id"] for row in search_rows] == ["lane_named"]
+
+    def test_omitted_session_key_keeps_source_scope(self, db):
+        session_db, _lane_key = db
+
+        rows = query_session_listing(
+            session_db,
+            source="telegram",
+            search_query="needle foreign 59",
+            limit=10,
+        )
+
+        assert [row["id"] for row in rows] == ["foreign_59"]