From a3f34a8f776203df48d1343c63dd873205ac2caa Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Fri, 3 Apr 2026 13:19:40 +0100 Subject: [PATCH] test: reduce telegram context partial mocks --- .../bot-message-context.acp-bindings.test.ts | 10 ++++++++-- .../bot-message-context.dm-threads.test.ts | 4 ++-- .../bot-message-context.route-test-support.ts | 4 ++-- .../src/bot-message-context.test-harness.ts | 9 --------- ...bot-message-context.thread-binding.test.ts | 4 ++-- .../bot-native-commands.session-meta.test.ts | 20 ++----------------- 6 files changed, 16 insertions(+), 35 deletions(-) diff --git a/extensions/telegram/src/bot-message-context.acp-bindings.test.ts b/extensions/telegram/src/bot-message-context.acp-bindings.test.ts index 7e123d2d985..f07788c7261 100644 --- a/extensions/telegram/src/bot-message-context.acp-bindings.test.ts +++ b/extensions/telegram/src/bot-message-context.acp-bindings.test.ts @@ -4,12 +4,18 @@ const ensureConfiguredBindingRouteReadyMock = vi.hoisted(() => vi.fn()); const recordInboundSessionMock = vi.hoisted(() => vi.fn().mockResolvedValue(undefined)); const resolveTelegramConversationRouteMock = vi.hoisted(() => vi.fn()); -vi.mock("openclaw/plugin-sdk/conversation-runtime", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("./bot-message-context.runtime.js", async (importOriginal) => { + const actual = await importOriginal(); return { ...actual, ensureConfiguredBindingRouteReady: (...args: unknown[]) => ensureConfiguredBindingRouteReadyMock(...args), + }; +}); +vi.mock("./bot-message-context.session.runtime.js", async (importOriginal) => { + const actual = await importOriginal(); + return { + ...actual, recordInboundSession: (...args: unknown[]) => recordInboundSessionMock(...args), }; }); diff --git a/extensions/telegram/src/bot-message-context.dm-threads.test.ts b/extensions/telegram/src/bot-message-context.dm-threads.test.ts index cf477eaa5e4..aca9e8412aa 100644 --- a/extensions/telegram/src/bot-message-context.dm-threads.test.ts +++ b/extensions/telegram/src/bot-message-context.dm-threads.test.ts @@ -3,8 +3,8 @@ const { recordInboundSessionMock } = vi.hoisted(() => ({ recordInboundSessionMock: vi.fn().mockResolvedValue(undefined), })); -vi.mock("openclaw/plugin-sdk/conversation-runtime", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("./bot-message-context.session.runtime.js", async (importOriginal) => { + const actual = await importOriginal(); return { ...actual, recordInboundSession: (...args: unknown[]) => recordInboundSessionMock(...args), diff --git a/extensions/telegram/src/bot-message-context.route-test-support.ts b/extensions/telegram/src/bot-message-context.route-test-support.ts index 47716be0795..1172a605368 100644 --- a/extensions/telegram/src/bot-message-context.route-test-support.ts +++ b/extensions/telegram/src/bot-message-context.route-test-support.ts @@ -6,8 +6,8 @@ const hoisted = vi.hoisted(() => ({ export const recordInboundSessionMock = hoisted.recordInboundSessionMock; -vi.mock("openclaw/plugin-sdk/conversation-runtime", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("./bot-message-context.session.runtime.js", async (importOriginal) => { + const actual = await importOriginal(); return { ...actual, recordInboundSession: (...args: unknown[]) => recordInboundSessionMock(...args), diff --git a/extensions/telegram/src/bot-message-context.test-harness.ts b/extensions/telegram/src/bot-message-context.test-harness.ts index ef53b7a359c..fdb5ffa24cc 100644 --- a/extensions/telegram/src/bot-message-context.test-harness.ts +++ b/extensions/telegram/src/bot-message-context.test-harness.ts @@ -90,14 +90,5 @@ async function installMessageContextTestMocks() { if (messageContextMocksInstalled) { return; } - const { vi } = await loadVitestModule(); - vi.doMock("openclaw/plugin-sdk/config-runtime", async (importOriginal) => { - const actual = await importOriginal(); - return { - ...actual, - readSessionUpdatedAt: () => undefined, - resolveStorePath: (storePath?: string) => storePath ?? "/tmp/sessions.json", - }; - }); messageContextMocksInstalled = true; } diff --git a/extensions/telegram/src/bot-message-context.thread-binding.test.ts b/extensions/telegram/src/bot-message-context.thread-binding.test.ts index 790eda40ad3..cdff8c7ca76 100644 --- a/extensions/telegram/src/bot-message-context.thread-binding.test.ts +++ b/extensions/telegram/src/bot-message-context.thread-binding.test.ts @@ -4,8 +4,8 @@ import { buildTelegramMessageContextForTest } from "./bot-message-context.test-h const recordInboundSessionMock = vi.hoisted(() => vi.fn().mockResolvedValue(undefined)); const resolveTelegramConversationRouteMock = vi.hoisted(() => vi.fn()); -vi.mock("openclaw/plugin-sdk/conversation-runtime", async (importOriginal) => { - const actual = await importOriginal(); +vi.mock("./bot-message-context.session.runtime.js", async (importOriginal) => { + const actual = await importOriginal(); return { ...actual, recordInboundSession: (...args: unknown[]) => recordInboundSessionMock(...args), diff --git a/extensions/telegram/src/bot-native-commands.session-meta.test.ts b/extensions/telegram/src/bot-native-commands.session-meta.test.ts index 93d0997f3a4..6c9288e98a3 100644 --- a/extensions/telegram/src/bot-native-commands.session-meta.test.ts +++ b/extensions/telegram/src/bot-native-commands.session-meta.test.ts @@ -104,30 +104,14 @@ vi.mock("openclaw/plugin-sdk/conversation-runtime", async (importOriginal) => { }), }; }); -vi.mock("openclaw/plugin-sdk/command-auth", async (importOriginal) => { - const actual = await importOriginal(); - return { - ...actual, - listSkillCommandsForAgents: vi.fn(() => []), - }; -}); -vi.mock("openclaw/plugin-sdk/reply-dispatch-runtime", async (importOriginal) => { - const actual = - await importOriginal(); +vi.mock("./bot-native-commands.runtime.js", async (importOriginal) => { + const actual = await importOriginal(); return { ...actual, finalizeInboundContext: vi.fn((ctx: unknown) => ctx), dispatchReplyWithBufferedBlockDispatcher: replyMocks.dispatchReplyWithBufferedBlockDispatcher, }; }); -vi.mock("../../../src/config/sessions.js", async (importOriginal) => { - const actual = await importOriginal(); - return { - ...actual, - recordSessionMetaFromInbound: sessionMocks.recordSessionMetaFromInbound, - resolveStorePath: sessionMocks.resolveStorePath, - }; -}); vi.mock("../../../src/pairing/pairing-store.js", () => ({ readChannelAllowFromStore: vi.fn(async () => []), }));