107 lines
3.8 KiB
TypeScript
107 lines
3.8 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { buildTelegramMessageContextForTest } from "./bot-message-context.test-harness.js";
|
|
|
|
describe("buildTelegramMessageContext dm thread sessions", () => {
|
|
const buildContext = async (message: Record<string, unknown>) =>
|
|
await buildTelegramMessageContextForTest({
|
|
message,
|
|
});
|
|
|
|
it("uses thread session key for dm topics", async () => {
|
|
const ctx = await buildContext({
|
|
message_id: 1,
|
|
chat: { id: 1234, type: "private" },
|
|
date: 1700000000,
|
|
text: "hello",
|
|
message_thread_id: 42,
|
|
from: { id: 42, first_name: "Alice" },
|
|
});
|
|
|
|
expect(ctx).not.toBeNull();
|
|
expect(ctx?.ctxPayload?.MessageThreadId).toBe(42);
|
|
expect(ctx?.ctxPayload?.SessionKey).toBe("agent:main:main:thread:42");
|
|
});
|
|
|
|
it("keeps legacy dm session key when no thread id", async () => {
|
|
const ctx = await buildContext({
|
|
message_id: 2,
|
|
chat: { id: 1234, type: "private" },
|
|
date: 1700000001,
|
|
text: "hello",
|
|
from: { id: 42, first_name: "Alice" },
|
|
});
|
|
|
|
expect(ctx).not.toBeNull();
|
|
expect(ctx?.ctxPayload?.MessageThreadId).toBeUndefined();
|
|
expect(ctx?.ctxPayload?.SessionKey).toBe("agent:main:main");
|
|
});
|
|
});
|
|
|
|
describe("buildTelegramMessageContext group sessions without forum", () => {
|
|
const buildContext = async (message: Record<string, unknown>) =>
|
|
await buildTelegramMessageContextForTest({
|
|
message,
|
|
options: { forceWasMentioned: true },
|
|
resolveGroupActivation: () => true,
|
|
});
|
|
|
|
it("ignores message_thread_id for regular groups (not forums)", async () => {
|
|
// When someone replies to a message in a non-forum group, Telegram sends
|
|
// message_thread_id but this should NOT create a separate session
|
|
const ctx = await buildContext({
|
|
message_id: 1,
|
|
chat: { id: -1001234567890, type: "supergroup", title: "Test Group" },
|
|
date: 1700000000,
|
|
text: "@bot hello",
|
|
message_thread_id: 42, // This is a reply thread, NOT a forum topic
|
|
from: { id: 42, first_name: "Alice" },
|
|
});
|
|
|
|
expect(ctx).not.toBeNull();
|
|
// Session key should NOT include :topic:42
|
|
expect(ctx?.ctxPayload?.SessionKey).toBe("agent:main:telegram:group:-1001234567890");
|
|
// MessageThreadId should be undefined (not a forum)
|
|
expect(ctx?.ctxPayload?.MessageThreadId).toBeUndefined();
|
|
});
|
|
|
|
it("keeps same session for regular group with and without message_thread_id", async () => {
|
|
const ctxWithThread = await buildContext({
|
|
message_id: 1,
|
|
chat: { id: -1001234567890, type: "supergroup", title: "Test Group" },
|
|
date: 1700000000,
|
|
text: "@bot hello",
|
|
message_thread_id: 42,
|
|
from: { id: 42, first_name: "Alice" },
|
|
});
|
|
|
|
const ctxWithoutThread = await buildContext({
|
|
message_id: 2,
|
|
chat: { id: -1001234567890, type: "supergroup", title: "Test Group" },
|
|
date: 1700000001,
|
|
text: "@bot world",
|
|
from: { id: 42, first_name: "Alice" },
|
|
});
|
|
|
|
expect(ctxWithThread).not.toBeNull();
|
|
expect(ctxWithoutThread).not.toBeNull();
|
|
// Both messages should use the same session key
|
|
expect(ctxWithThread?.ctxPayload?.SessionKey).toBe(ctxWithoutThread?.ctxPayload?.SessionKey);
|
|
});
|
|
|
|
it("uses topic session for forum groups with message_thread_id", async () => {
|
|
const ctx = await buildContext({
|
|
message_id: 1,
|
|
chat: { id: -1001234567890, type: "supergroup", title: "Test Forum", is_forum: true },
|
|
date: 1700000000,
|
|
text: "@bot hello",
|
|
message_thread_id: 99,
|
|
from: { id: 42, first_name: "Alice" },
|
|
});
|
|
|
|
expect(ctx).not.toBeNull();
|
|
// Session key SHOULD include :topic:99 for forums
|
|
expect(ctx?.ctxPayload?.SessionKey).toBe("agent:main:telegram:group:-1001234567890:topic:99");
|
|
expect(ctx?.ctxPayload?.MessageThreadId).toBe(99);
|
|
});
|
|
});
|