vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-06 10:30:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(discord): mock message process dependencies narrowly

Browse Source
This commit is contained in:
Peter Steinberger
2026-04-25 22:18:59 +01:00
parent 8a731c1ef7
commit 9d22061e3e
1 changed files with 93 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

173
extensions/discord/src/monitor/message-handler.process.test.ts
View File

@@ -36,11 +36,48 @@ const deliveryMocks = vi.hoisted(() => ({
) => Promise<import("discord-api-types/v10").APIMessage>
>(async () => ({ id: "m1" }) as import("discord-api-types/v10").APIMessage),
deliverDiscordReply: vi.fn<(params: unknown) => Promise<void>>(async () => {}),
createDiscordDraftStream: vi.fn(() => createMockDraftStream()),
createDiscordDraftStream: vi.fn<(params: unknown) => ReturnType<typeof createMockDraftStream>>(
() => createMockDraftStream(),
),
}));
const editMessageDiscord = deliveryMocks.editMessageDiscord;
const deliverDiscordReply = deliveryMocks.deliverDiscordReply;
const createDiscordDraftStream = deliveryMocks.createDiscordDraftStream;
vi.mock("../send.js", () => ({
reactMessageDiscord: async (
channelId: string,
messageId: string,
emoji: string,
opts?: unknown,
) => {
await sendMocks.reactMessageDiscord(channelId, messageId, emoji, opts);
return { ok: true };
},
removeReactionDiscord: async (
channelId: string,
messageId: string,
emoji: string,
opts?: unknown,
) => {
await sendMocks.removeReactionDiscord(channelId, messageId, emoji, opts);
return { ok: true };
},
}));
vi.mock("../send.messages.js", () => ({
editMessageDiscord: (channelId: string, messageId: string, payload: unknown, opts?: unknown) =>
deliveryMocks.editMessageDiscord(channelId, messageId, payload, opts),
}));
vi.mock("../draft-stream.js", () => ({
createDiscordDraftStream: (params: unknown) => deliveryMocks.createDiscordDraftStream(params),
}));
vi.mock("./reply-delivery.js", () => ({
deliverDiscordReply: (params: unknown) => deliveryMocks.deliverDiscordReply(params),
}));
type DispatchInboundParams = {
dispatcher: {
sendBlockReply: (payload: ReplyPayload) => boolean | Promise<boolean>;
@@ -101,94 +138,70 @@ const configSessionsMocks = vi.hoisted(() => ({
}));
const readSessionUpdatedAt = configSessionsMocks.readSessionUpdatedAt;
const resolveStorePath = configSessionsMocks.resolveStorePath;
const createDiscordRestClientSpy = vi.hoisted(() =>
vi.fn<
(params: unknown) => {
token: string;
rest: object;
account: { accountId: string; config: object };
}
>(() => ({
token: "token",
rest: {},
account: { accountId: "default", config: {} },
})),
);
let createBaseDiscordMessageContext: typeof import("./message-handler.test-harness.js").createBaseDiscordMessageContext;
let createDiscordDirectMessageContextOverrides: typeof import("./message-handler.test-harness.js").createDiscordDirectMessageContextOverrides;
let threadBindingTesting: typeof import("./thread-bindings.js").__testing;
let createThreadBindingManager: typeof import("./thread-bindings.js").createThreadBindingManager;
let processDiscordMessage: typeof import("./message-handler.process.js").processDiscordMessage;
const sendModule = await import("../send.js");
vi.spyOn(sendModule, "reactMessageDiscord").mockImplementation(
async (channelId, messageId, emoji, opts) => {
await sendMocks.reactMessageDiscord(channelId, messageId, emoji, opts);
return { ok: true };
},
);
vi.spyOn(sendModule, "removeReactionDiscord").mockImplementation(
async (channelId, messageId, emoji, opts) => {
await sendMocks.removeReactionDiscord(channelId, messageId, emoji, opts);
return { ok: true };
},
);
vi.mock("openclaw/plugin-sdk/reply-runtime", () => ({
dispatchInboundMessage: (params: DispatchInboundParams) => dispatchInboundMessage(params),
createReplyDispatcherWithTyping: (opts: {
deliver: (payload: unknown, info: { kind: string }) => Promise<void> | void;
}) => ({
dispatcher: {
sendToolResult: vi.fn(() => true),
sendBlockReply: vi.fn((payload: unknown) => {
void opts.deliver(payload, { kind: "block" });
return true;
}),
sendFinalReply: vi.fn((payload: unknown) => {
void opts.deliver(payload, { kind: "final" });
return true;
}),
waitForIdle: vi.fn(async () => {}),
getQueuedCounts: vi.fn(() => ({ tool: 0, block: 0, final: 0 })),
markComplete: vi.fn(),
},
replyOptions: {},
markDispatchIdle: vi.fn(),
markRunComplete: vi.fn(),
}),
}));
const sendMessagesModule = await import("../send.messages.js");
vi.spyOn(sendMessagesModule, "editMessageDiscord").mockImplementation(
((
channelId: Parameters<typeof sendMessagesModule.editMessageDiscord>[0],
messageId: Parameters<typeof sendMessagesModule.editMessageDiscord>[1],
payload: Parameters<typeof sendMessagesModule.editMessageDiscord>[2],
opts: Parameters<typeof sendMessagesModule.editMessageDiscord>[3],
) => deliveryMocks.editMessageDiscord(channelId, messageId, payload, opts) as never) as never,
);
vi.mock("openclaw/plugin-sdk/conversation-runtime", () => ({
recordInboundSession: (...args: unknown[]) => recordInboundSession(...args),
registerSessionBindingAdapter: vi.fn(),
unregisterSessionBindingAdapter: vi.fn(),
resolveThreadBindingConversationIdFromBindingId: (bindingId: string) =>
bindingId.split(":").at(-1) ?? bindingId,
}));
const draftStreamModule = await import("../draft-stream.js");
vi.spyOn(draftStreamModule, "createDiscordDraftStream").mockImplementation(
deliveryMocks.createDiscordDraftStream,
);
vi.mock("openclaw/plugin-sdk/session-store-runtime", () => ({
readSessionUpdatedAt: (...args: unknown[]) => configSessionsMocks.readSessionUpdatedAt(...args),
resolveStorePath: (...args: unknown[]) => configSessionsMocks.resolveStorePath(...args),
}));
const replyDeliveryModule = await import("./reply-delivery.js");
vi.spyOn(replyDeliveryModule, "deliverDiscordReply").mockImplementation(
((params: Parameters<typeof replyDeliveryModule.deliverDiscordReply>[0]) =>
deliveryMocks.deliverDiscordReply(params) as never) as never,
);
const replyRuntimeModule = await import("openclaw/plugin-sdk/reply-runtime");
vi.spyOn(replyRuntimeModule, "dispatchInboundMessage").mockImplementation(
((params: Parameters<typeof replyRuntimeModule.dispatchInboundMessage>[0]) =>
dispatchInboundMessage(params as DispatchInboundParams) as never) as never,
);
vi.spyOn(replyRuntimeModule, "createReplyDispatcherWithTyping").mockImplementation(((opts: {
deliver: (payload: unknown, info: { kind: string }) => Promise<void> | void;
}) => ({
dispatcher: {
sendToolResult: vi.fn(() => true),
sendBlockReply: vi.fn((payload: unknown) => {
void opts.deliver(payload as never, { kind: "block" });
return true;
}),
sendFinalReply: vi.fn((payload: unknown) => {
void opts.deliver(payload as never, { kind: "final" });
return true;
}),
waitForIdle: vi.fn(async () => {}),
getQueuedCounts: vi.fn(() => ({ tool: 0, block: 0, final: 0 })),
markComplete: vi.fn(),
},
replyOptions: {},
markDispatchIdle: vi.fn(),
markRunComplete: vi.fn(),
})) as never);
const conversationRuntimeModule = await import("openclaw/plugin-sdk/conversation-runtime");
vi.spyOn(conversationRuntimeModule, "recordInboundSession").mockImplementation(
((params: Parameters<typeof conversationRuntimeModule.recordInboundSession>[0]) =>
recordInboundSession(params) as never) as never,
);
const sessionStoreRuntimeModule = await import("openclaw/plugin-sdk/session-store-runtime");
vi.spyOn(sessionStoreRuntimeModule, "readSessionUpdatedAt").mockImplementation(
((params: Parameters<typeof sessionStoreRuntimeModule.readSessionUpdatedAt>[0]) =>
configSessionsMocks.readSessionUpdatedAt(params) as never) as never,
);
vi.spyOn(sessionStoreRuntimeModule, "resolveStorePath").mockImplementation(
((
path: Parameters<typeof sessionStoreRuntimeModule.resolveStorePath>[0],
opts: Parameters<typeof sessionStoreRuntimeModule.resolveStorePath>[1],
) => configSessionsMocks.resolveStorePath(path, opts) as never) as never,
);
const clientModule = await import("../client.js");
const createDiscordRestClientSpy = vi.spyOn(clientModule, "createDiscordRestClient");
vi.mock("../client.js", () => ({
createDiscordRuntimeAccountContext: (params: { cfg: unknown; accountId: string }) => ({
cfg: params.cfg,
accountId: params.accountId,
}),
createDiscordRestClient: (params: unknown) => createDiscordRestClientSpy(params),
}));
const BASE_CHANNEL_ROUTE = {
agentId: "main",