tests: preserve session-key exports in media-only mock

This commit is contained in:
Eva
2026-04-11 03:32:44 +07:00
committed by Peter Steinberger
parent 9f476107ea
commit 0ff47c8720

View File

@@ -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<typeof import("../../routing/session-key.js")>();
return {
...actual,
normalizeMainKey: () => "main",
normalizeAgentId: (id: string | undefined | null) => id ?? "default",
};
});
vi.mock("../../utils/provider-utils.js", () => ({
isReasoningTagProvider: vi.fn().mockReturnValue(false),