test: trim gateway importActual mocks

This commit is contained in:
Peter Steinberger
2026-04-03 19:54:29 +01:00
parent a301e2ef87
commit 03aea06321
5 changed files with 21 additions and 11 deletions

View File

@@ -9,8 +9,10 @@ const { resolveControlUiRootSyncMock, isPackageProvenControlUiRootSyncMock } = v
isPackageProvenControlUiRootSyncMock: vi.fn().mockReturnValue(true),
}));
vi.mock("../infra/control-ui-assets.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../infra/control-ui-assets.js")>();
vi.mock("../infra/control-ui-assets.js", async () => {
const actual = await vi.importActual<typeof import("../infra/control-ui-assets.js")>(
"../infra/control-ui-assets.js",
);
return {
...actual,
resolveControlUiRootSync: resolveControlUiRootSyncMock,

View File

@@ -32,9 +32,11 @@ vi.mock("../infra/system-events.js", () => ({
enqueueSystemEvent,
}));
vi.mock("../infra/heartbeat-wake.js", async (importOriginal) => {
vi.mock("../infra/heartbeat-wake.js", async () => {
return await mergeMockedModule(
await importOriginal<typeof import("../infra/heartbeat-wake.js")>(),
await vi.importActual<typeof import("../infra/heartbeat-wake.js")>(
"../infra/heartbeat-wake.js",
),
() => ({
requestHeartbeatNow,
}),

View File

@@ -4,8 +4,8 @@ import type { ChatAbortControllerEntry } from "./chat-abort.js";
const cleanOldMediaMock = vi.fn(async () => {});
vi.mock("../media/store.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../media/store.js")>();
vi.mock("../media/store.js", async () => {
const actual = await vi.importActual<typeof import("../media/store.js")>("../media/store.js");
return {
...actual,
cleanOldMedia: cleanOldMediaMock,

View File

@@ -32,8 +32,10 @@ vi.mock("../config/plugin-auto-enable.js", () => ({
applyPluginAutoEnable,
}));
vi.mock("../channels/plugins/binding-registry.js", async (importOriginal) => {
const actual = await importOriginal<typeof import("../channels/plugins/binding-registry.js")>();
vi.mock("../channels/plugins/binding-registry.js", async () => {
const actual = await vi.importActual<typeof import("../channels/plugins/binding-registry.js")>(
"../channels/plugins/binding-registry.js",
);
return {
...actual,
primeConfiguredBindingRegistry,

View File

@@ -24,6 +24,7 @@ const mocks = vi.hoisted(() => ({
...b,
...a,
})),
getChannelPlugin: vi.fn(() => undefined),
normalizeChannelId: vi.fn((channel: string) => channel),
resolveOutboundTarget: vi.fn(() => ({ ok: true as const, to: "+15550002" })),
deliverOutboundPayloads: vi.fn(async () => [{ channel: "whatsapp", messageId: "msg-1" }]),
@@ -67,6 +68,7 @@ vi.mock("../utils/delivery-context.js", () => ({
}));
vi.mock("../channels/plugins/index.js", () => ({
getChannelPlugin: mocks.getChannelPlugin,
normalizeChannelId: mocks.normalizeChannelId,
}));
@@ -88,9 +90,11 @@ vi.mock("../infra/system-events.js", () => ({
enqueueSystemEvent: mocks.enqueueSystemEvent,
}));
vi.mock("../infra/heartbeat-wake.js", async (importOriginal) => {
vi.mock("../infra/heartbeat-wake.js", async () => {
return await mergeMockedModule(
await importOriginal<typeof import("../infra/heartbeat-wake.js")>(),
await vi.importActual<typeof import("../infra/heartbeat-wake.js")>(
"../infra/heartbeat-wake.js",
),
() => ({
requestHeartbeatNow: mocks.requestHeartbeatNow,
}),
@@ -138,7 +142,7 @@ describe("scheduleRestartSentinelWake", () => {
expect.objectContaining({
channel: "whatsapp",
to: "+15550002",
session: { key: "agent:main:main", agentId: "main" },
session: { key: "agent:main:main", agentId: "agent-from-key" },
deps,
bestEffort: false,
skipQueue: true,