Ayuriel/openclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(telegram): align media harness with runtime seam

Browse Source
This commit is contained in:
Ayaan Zaidi 2026-03-18 09:48:06 +05:30 committed by Val Alexander
parent 97ae0158e8
commit 3a4ed87599
No known key found for this signature in database
2 changed files with 31 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
extensions/telegram/src/bot.media.e2e-harness.ts
View File

@ -56,12 +56,7 @@ const apiStub: ApiStub = {
setMyCommands: vi.fn(async () => undefined),
};
beforeEach(() => {
resetInboundDedupe();
resetSaveMediaBufferMock();
});
vi.mock("grammy", () => ({
export const telegramBotRuntimeForTest = {
Bot: class {
api = apiStub;
use = middlewareUseSpy;
@ -71,20 +66,25 @@ vi.mock("grammy", () => ({
catch = vi.fn();
constructor(public token: string) {}
},
InputFile: class {},
webhookCallback: vi.fn(),
}));
vi.mock("@grammyjs/runner", () => ({
sequentialize: () => vi.fn(),
}));
apiThrottler: () => throttlerSpy(),
loadConfig: () => ({
channels: { telegram: { dmPolicy: "open", allowFrom: ["*"] } },
}),
};
beforeEach(() => {
resetInboundDedupe();
resetSaveMediaBufferMock();
});
const throttlerSpy = vi.fn(() => "throttler");
vi.mock("@grammyjs/transformer-throttler", () => ({
apiThrottler: () => throttlerSpy(),
vi.doMock("./bot.runtime.js", () => ({
...telegramBotRuntimeForTest,
}));
vi.mock("undici", async (importOriginal) => {
vi.doMock("undici", async (importOriginal) => {
const actual = await importOriginal<typeof import("undici")>();
return {
...actual,
@ -92,7 +92,7 @@ vi.mock("undici", async (importOriginal) => {
};
});
vi.mock("openclaw/plugin-sdk/media-runtime", async (importOriginal) => {
vi.doMock("openclaw/plugin-sdk/media-runtime", async (importOriginal) => {
const actual = await importOriginal<typeof import("openclaw/plugin-sdk/media-runtime")>();
const mockModule = Object.create(null) as Record<string, unknown>;
Object.defineProperties(mockModule, Object.getOwnPropertyDescriptors(actual));
@ -105,7 +105,7 @@ vi.mock("openclaw/plugin-sdk/media-runtime", async (importOriginal) => {
return mockModule;
});
vi.mock("openclaw/plugin-sdk/config-runtime", async (importOriginal) => {
vi.doMock("openclaw/plugin-sdk/config-runtime", async (importOriginal) => {
const actual = await importOriginal<typeof import("openclaw/plugin-sdk/config-runtime")>();
return {
...actual,
@ -115,7 +115,7 @@ vi.mock("openclaw/plugin-sdk/config-runtime", async (importOriginal) => {
};
});
vi.mock("openclaw/plugin-sdk/config-runtime", async (importOriginal) => {
vi.doMock("openclaw/plugin-sdk/config-runtime", async (importOriginal) => {
const actual = await importOriginal<typeof import("openclaw/plugin-sdk/config-runtime")>();
return {
...actual,
@ -123,7 +123,7 @@ vi.mock("openclaw/plugin-sdk/config-runtime", async (importOriginal) => {
};
});
vi.mock("openclaw/plugin-sdk/conversation-runtime", () => ({
vi.doMock("openclaw/plugin-sdk/conversation-runtime", () => ({
readChannelAllowFromStore: vi.fn(async () => [] as string[]),
upsertChannelPairingRequest: vi.fn(async () => ({
code: "PAIRCODE",
@ -131,7 +131,7 @@ vi.mock("openclaw/plugin-sdk/conversation-runtime", () => ({
})),
}));
vi.mock("openclaw/plugin-sdk/reply-runtime", () => {
vi.doMock("openclaw/plugin-sdk/reply-runtime", () => {
const replySpy = vi.fn(async (_ctx, opts) => {
await opts?.onReplyStart?.();
return undefined;

16
extensions/telegram/src/bot.media.test-utils.ts
View File

@ -1,6 +1,5 @@
import * as ssrf from "openclaw/plugin-sdk/infra-runtime";
import { afterEach, beforeAll, beforeEach, expect, vi, type Mock } from "vitest";
import { onSpy, sendChatActionSpy } from "./bot.media.e2e-harness.js";
type StickerSpy = Mock<(...args: unknown[]) => unknown>;
@ -21,6 +20,8 @@ const TELEGRAM_BOT_IMPORT_TIMEOUT_MS = process.platform === "win32" ? 180_000 :
let createTelegramBotRef: typeof import("./bot.js").createTelegramBot;
let replySpyRef: ReturnType<typeof vi.fn>;
let onSpyRef: Mock;
let sendChatActionSpyRef: Mock;
export async function createBotHandler(): Promise<{
handler: (ctx: Record<string, unknown>) => Promise<void>;
@ -39,9 +40,9 @@ export async function createBotHandlerWithOptions(options: {
replySpy: ReturnType<typeof vi.fn>;
runtimeError: ReturnType<typeof vi.fn>;
}> {
onSpy.mockClear();
onSpyRef.mockClear();
replySpyRef.mockClear();
sendChatActionSpy.mockClear();
sendChatActionSpyRef.mockClear();
const runtimeError = options.runtimeError ?? vi.fn();
const runtimeLog = options.runtimeLog ?? vi.fn();
@ -57,7 +58,7 @@ export async function createBotHandlerWithOptions(options: {
},
},
});
const handler = onSpy.mock.calls.find((call) => call[0] === "message")?.[1] as (
const handler = onSpyRef.mock.calls.find((call) => call[0] === "message")?.[1] as (
ctx: Record<string, unknown>,
) => Promise<void>;
expect(handler).toBeDefined();
@ -102,7 +103,12 @@ afterEach(() => {
});
beforeAll(async () => {
({ createTelegramBot: createTelegramBotRef } = await import("./bot.js"));
const harness = await import("./bot.media.e2e-harness.js");
onSpyRef = harness.onSpy;
sendChatActionSpyRef = harness.sendChatActionSpy;
const botModule = await import("./bot.js");
botModule.setTelegramBotRuntimeForTest(harness.telegramBotRuntimeForTest);
({ createTelegramBot: createTelegramBotRef } = botModule);
const replyModule = await import("openclaw/plugin-sdk/reply-runtime");
replySpyRef = (replyModule as unknown as { __replySpy: ReturnType<typeof vi.fn> }).__replySpy;
}, TELEGRAM_BOT_IMPORT_TIMEOUT_MS);