119 lines
5.0 KiB
Python
119 lines
5.0 KiB
Python
"""Issue #76423 — SessionDB: telegram topic tables namespace by profile."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sqlite3
|
|
from pathlib import Path
|
|
|
|
from hermes_state import SessionDB
|
|
|
|
|
|
CHAT = "208214988"
|
|
|
|
|
|
def _session(db, sid, profile_name=None):
|
|
db.create_session(session_id=sid, source="telegram", user_id=CHAT, profile_name=profile_name)
|
|
|
|
|
|
def test_legacy_rows_migrate_only_to_default(tmp_path: Path):
|
|
"""v1 shape (no CASCADE, old user index) → v3: rows land in 'default' only."""
|
|
db_path = tmp_path / "legacy.db"
|
|
conn = sqlite3.connect(str(db_path))
|
|
conn.executescript(
|
|
f"""
|
|
CREATE TABLE state_meta (key TEXT PRIMARY KEY, value TEXT);
|
|
INSERT INTO state_meta(key, value) VALUES ('telegram_dm_topic_schema_version', '1');
|
|
CREATE TABLE sessions (
|
|
id TEXT PRIMARY KEY, source TEXT, user_id TEXT, model TEXT,
|
|
model_config TEXT, system_prompt TEXT, parent_session_id TEXT,
|
|
started_at REAL, ended_at REAL, end_reason TEXT,
|
|
message_count INTEGER DEFAULT 0, tool_call_count INTEGER DEFAULT 0,
|
|
input_tokens INTEGER DEFAULT 0, output_tokens INTEGER DEFAULT 0
|
|
);
|
|
INSERT INTO sessions(id, source, user_id, started_at)
|
|
VALUES ('legacy-sess', 'telegram', '{CHAT}', 1.0);
|
|
CREATE TABLE telegram_dm_topic_mode (
|
|
chat_id TEXT PRIMARY KEY, user_id TEXT NOT NULL,
|
|
enabled INTEGER NOT NULL DEFAULT 1,
|
|
activated_at REAL NOT NULL, updated_at REAL NOT NULL,
|
|
has_topics_enabled INTEGER, allows_users_to_create_topics INTEGER,
|
|
capability_checked_at REAL, intro_message_id TEXT, pinned_message_id TEXT
|
|
);
|
|
INSERT INTO telegram_dm_topic_mode(chat_id, user_id, enabled, activated_at, updated_at)
|
|
VALUES ('{CHAT}', '{CHAT}', 1, 1.0, 1.0);
|
|
CREATE TABLE telegram_dm_topic_bindings (
|
|
chat_id TEXT NOT NULL, thread_id TEXT NOT NULL, user_id TEXT NOT NULL,
|
|
session_key TEXT NOT NULL,
|
|
session_id TEXT NOT NULL REFERENCES sessions(id),
|
|
managed_mode TEXT NOT NULL DEFAULT 'auto',
|
|
linked_at REAL NOT NULL, updated_at REAL NOT NULL,
|
|
PRIMARY KEY (chat_id, thread_id)
|
|
);
|
|
CREATE INDEX idx_telegram_dm_topic_bindings_user
|
|
ON telegram_dm_topic_bindings(user_id, chat_id);
|
|
INSERT INTO telegram_dm_topic_bindings
|
|
VALUES ('{CHAT}', '99', '{CHAT}', 'k', 'legacy-sess', 'auto', 1.0, 1.0);
|
|
"""
|
|
)
|
|
conn.close()
|
|
|
|
db = SessionDB(db_path=db_path)
|
|
db.apply_telegram_topic_migration()
|
|
assert db.get_meta("telegram_dm_topic_schema_version") == "3"
|
|
assert db.is_telegram_topic_mode_enabled(
|
|
chat_id=CHAT, user_id=CHAT, profile_name="default",
|
|
)
|
|
assert not db.is_telegram_topic_mode_enabled(
|
|
chat_id=CHAT, user_id=CHAT, profile_name="coder",
|
|
)
|
|
assert db.get_telegram_topic_binding(
|
|
chat_id=CHAT, thread_id="99", profile_name="default",
|
|
)["session_id"] == "legacy-sess"
|
|
assert db.get_telegram_topic_binding(
|
|
chat_id=CHAT, thread_id="99", profile_name="coder",
|
|
) is None
|
|
fk = db._conn.execute("PRAGMA foreign_key_list('telegram_dm_topic_bindings')").fetchall()
|
|
assert any(row[2] == "sessions" and row[6] == "CASCADE" for row in fk)
|
|
db.close()
|
|
|
|
|
|
def test_mode_and_bindings_isolated_across_profiles(tmp_path: Path):
|
|
db = SessionDB(db_path=tmp_path / "state.db")
|
|
_session(db, "sess-a", "alpha")
|
|
_session(db, "sess-b", "beta")
|
|
|
|
db.enable_telegram_topic_mode(chat_id=CHAT, user_id=CHAT, profile_name="alpha")
|
|
db.enable_telegram_topic_mode(chat_id=CHAT, user_id=CHAT, profile_name="beta")
|
|
db.disable_telegram_topic_mode(chat_id=CHAT, profile_name="alpha")
|
|
assert not db.is_telegram_topic_mode_enabled(chat_id=CHAT, user_id=CHAT, profile_name="alpha")
|
|
assert db.is_telegram_topic_mode_enabled(chat_id=CHAT, user_id=CHAT, profile_name="beta")
|
|
|
|
db.bind_telegram_topic(
|
|
chat_id=CHAT, thread_id="77", user_id=CHAT,
|
|
session_key="ka", session_id="sess-a", profile_name="alpha",
|
|
)
|
|
db.bind_telegram_topic(
|
|
chat_id=CHAT, thread_id="77", user_id=CHAT,
|
|
session_key="kb", session_id="sess-b", profile_name="beta",
|
|
)
|
|
assert db.get_telegram_topic_binding(
|
|
chat_id=CHAT, thread_id="77", profile_name="alpha",
|
|
)["session_id"] == "sess-a"
|
|
assert db.get_telegram_topic_binding(
|
|
chat_id=CHAT, thread_id="77", profile_name="beta",
|
|
)["session_id"] == "sess-b"
|
|
|
|
assert db.delete_telegram_topic_binding(
|
|
chat_id=CHAT, thread_id="77", profile_name="alpha",
|
|
) == 1
|
|
assert db.get_telegram_topic_binding(
|
|
chat_id=CHAT, thread_id="77", profile_name="alpha",
|
|
) is None
|
|
assert db.get_telegram_topic_binding(
|
|
chat_id=CHAT, thread_id="77", profile_name="beta",
|
|
) is not None
|
|
# Omitted kwarg == the single-profile "default" namespace, not a wildcard.
|
|
assert not db.is_telegram_topic_mode_enabled(chat_id=CHAT, user_id=CHAT)
|
|
assert db.get_telegram_topic_binding(chat_id=CHAT, thread_id="77") is None
|
|
db.close()
|