hermes-agent/tests/gateway/test_telegram_topic_profile_isolation.py

"""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()