openclaw/extensions/telegram/src/bot-handlers.message-session.runtime.test.ts

// Telegram tests cover message-session routing and persisted model inheritance.
import type { SessionEntry } from "openclaw/plugin-sdk/session-store-runtime";
import { describe, expect, it, vi } from "vitest";
import type { TelegramBotDeps } from "./bot-deps.js";
import { createTelegramMessageSessionRuntime } from "./bot-handlers.message-context.js";

describe("createTelegramMessageSessionRuntime", () => {
  it("inherits a DM topic model override through keyed session loads", () => {
    const storePath = "/tmp/telegram-sessions.sqlite";
    const childSessionKey = "agent:main:main:thread:12345:99";
    const parentSessionKey = "agent:main:main";
    const entries: Record<string, SessionEntry> = {
      [childSessionKey]: { sessionId: "child", updatedAt: 2 },
      [parentSessionKey]: {
        sessionId: "parent",
        updatedAt: 1,
        providerOverride: "anthropic",
        modelOverride: "claude-opus-4-7",
        modelOverrideSource: "user",
      },
    };
    const getSessionEntry = vi.fn<NonNullable<TelegramBotDeps["getSessionEntry"]>>(
      ({ sessionKey }) => entries[sessionKey],
    );
    const telegramDeps = {
      resolveStorePath: vi.fn(() => storePath),
      getSessionEntry,
    } as Pick<TelegramBotDeps, "resolveStorePath" | "getSessionEntry"> as TelegramBotDeps;
    const { resolveTelegramSessionState } = createTelegramMessageSessionRuntime({
      accountId: "default",
      resolveTelegramGroupConfig: () => ({}),
      telegramDeps,
    });

    const state = resolveTelegramSessionState({
      chatId: 12345,
      isGroup: false,
      threadSpec: { id: 99, scope: "dm" },
      botHasTopicsEnabled: true,
      senderId: 12345,
      runtimeCfg: {},
    });

    expect(state.sessionKey).toBe(childSessionKey);
    expect(state.model).toBe("anthropic/claude-opus-4-7");
    expect(getSessionEntry.mock.calls.map(([params]) => params)).toEqual([
      { storePath, sessionKey: childSessionKey },
      { storePath, sessionKey: parentSessionKey },
    ]);
  });
});