120 lines
3.8 KiB
TypeScript
120 lines
3.8 KiB
TypeScript
import { describe, expect, it, vi, beforeEach } from "vitest";
|
|
import { buildTelegramMessageContext } from "./bot-message-context.js";
|
|
|
|
// Mock recordInboundSession to capture updateLastRoute parameter
|
|
const recordInboundSessionMock = vi.fn().mockResolvedValue(undefined);
|
|
vi.mock("../channels/session.js", () => ({
|
|
recordInboundSession: (...args: unknown[]) => recordInboundSessionMock(...args),
|
|
}));
|
|
|
|
describe("buildTelegramMessageContext DM topic threadId in deliveryContext (#8891)", () => {
|
|
const baseConfig = {
|
|
agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/openclaw" } },
|
|
channels: { telegram: {} },
|
|
messages: { groupChat: { mentionPatterns: [] } },
|
|
} as never;
|
|
|
|
async function buildCtx(params: {
|
|
message: Record<string, unknown>;
|
|
options?: Record<string, unknown>;
|
|
resolveGroupActivation?: () => unknown;
|
|
}): Promise<Awaited<ReturnType<typeof buildTelegramMessageContext>>> {
|
|
return await buildTelegramMessageContext({
|
|
primaryCtx: {
|
|
message: {
|
|
message_id: 1,
|
|
date: 1700000000,
|
|
text: "hello",
|
|
from: { id: 42, first_name: "Alice" },
|
|
...params.message,
|
|
},
|
|
me: { id: 7, username: "bot" },
|
|
} as never,
|
|
allMedia: [],
|
|
storeAllowFrom: [],
|
|
options: params.options ?? {},
|
|
bot: {
|
|
api: {
|
|
sendChatAction: vi.fn(),
|
|
setMessageReaction: vi.fn(),
|
|
},
|
|
} as never,
|
|
cfg: baseConfig,
|
|
account: { accountId: "default" } as never,
|
|
historyLimit: 0,
|
|
groupHistories: new Map(),
|
|
dmPolicy: "open",
|
|
allowFrom: [],
|
|
groupAllowFrom: [],
|
|
ackReactionScope: "off",
|
|
logger: { info: vi.fn() },
|
|
resolveGroupActivation: params.resolveGroupActivation ?? (() => undefined),
|
|
resolveGroupRequireMention: () => false,
|
|
resolveTelegramGroupConfig: () => ({
|
|
groupConfig: { requireMention: false },
|
|
topicConfig: undefined,
|
|
}),
|
|
});
|
|
}
|
|
|
|
function getUpdateLastRoute(): unknown {
|
|
const callArgs = recordInboundSessionMock.mock.calls[0]?.[0] as { updateLastRoute?: unknown };
|
|
return callArgs?.updateLastRoute;
|
|
}
|
|
|
|
beforeEach(() => {
|
|
recordInboundSessionMock.mockClear();
|
|
});
|
|
|
|
it("passes threadId to updateLastRoute for DM topics", async () => {
|
|
const ctx = await buildCtx({
|
|
message: {
|
|
chat: { id: 1234, type: "private" },
|
|
message_thread_id: 42, // DM Topic ID
|
|
},
|
|
});
|
|
|
|
expect(ctx).not.toBeNull();
|
|
expect(recordInboundSessionMock).toHaveBeenCalled();
|
|
|
|
// Check that updateLastRoute includes threadId
|
|
const updateLastRoute = getUpdateLastRoute() as { threadId?: string } | undefined;
|
|
expect(updateLastRoute).toBeDefined();
|
|
expect(updateLastRoute?.threadId).toBe("42");
|
|
});
|
|
|
|
it("does not pass threadId for regular DM without topic", async () => {
|
|
const ctx = await buildCtx({
|
|
message: {
|
|
chat: { id: 1234, type: "private" },
|
|
},
|
|
});
|
|
|
|
expect(ctx).not.toBeNull();
|
|
expect(recordInboundSessionMock).toHaveBeenCalled();
|
|
|
|
// Check that updateLastRoute does NOT include threadId
|
|
const updateLastRoute = getUpdateLastRoute() as { threadId?: string } | undefined;
|
|
expect(updateLastRoute).toBeDefined();
|
|
expect(updateLastRoute?.threadId).toBeUndefined();
|
|
});
|
|
|
|
it("does not set updateLastRoute for group messages", async () => {
|
|
const ctx = await buildCtx({
|
|
message: {
|
|
chat: { id: -1001234567890, type: "supergroup", title: "Test Group" },
|
|
text: "@bot hello",
|
|
message_thread_id: 99,
|
|
},
|
|
options: { forceWasMentioned: true },
|
|
resolveGroupActivation: () => true,
|
|
});
|
|
|
|
expect(ctx).not.toBeNull();
|
|
expect(recordInboundSessionMock).toHaveBeenCalled();
|
|
|
|
// Check that updateLastRoute is undefined for groups
|
|
expect(getUpdateLastRoute()).toBeUndefined();
|
|
});
|
|
});
|