From 0ff47c8720eb31e319602b04467113023a4cecd4 Mon Sep 17 00:00:00 2001 From: Eva Date: Sat, 11 Apr 2026 03:32:44 +0700 Subject: [PATCH] tests: preserve session-key exports in media-only mock --- .../reply/get-reply-run.media-only.test.ts | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/src/auto-reply/reply/get-reply-run.media-only.test.ts b/src/auto-reply/reply/get-reply-run.media-only.test.ts index 3a060f672d1..57cac677296 100644 --- a/src/auto-reply/reply/get-reply-run.media-only.test.ts +++ b/src/auto-reply/reply/get-reply-run.media-only.test.ts @@ -48,12 +48,14 @@ vi.mock("../../process/command-queue.js", () => ({ getQueueSize: vi.fn().mockReturnValue(0), })); -vi.mock("../../routing/session-key.js", () => ({ - DEFAULT_AGENT_ID: "main", - DEFAULT_MAIN_KEY: "main", - normalizeMainKey: vi.fn().mockReturnValue("main"), - normalizeAgentId: vi.fn((id?: string) => id ?? "default"), -})); +vi.mock(import("../../routing/session-key.js"), async (importOriginal) => { + const actual = await importOriginal(); + return { + ...actual, + normalizeMainKey: () => "main", + normalizeAgentId: (id: string | undefined | null) => id ?? "default", + }; +}); vi.mock("../../utils/provider-utils.js", () => ({ isReasoningTagProvider: vi.fn().mockReturnValue(false),