diff --git a/src/commands/channels.add.test.ts b/src/commands/channels.add.test.ts index 8cabd70039e..7eefa8ac13a 100644 --- a/src/commands/channels.add.test.ts +++ b/src/commands/channels.add.test.ts @@ -1,34 +1,8 @@ -import { beforeEach, describe, expect, it, vi } from "vitest"; +import { beforeEach, describe, expect, it } from "vitest"; import { setDefaultChannelPluginRegistryForTests } from "./channel-test-helpers.js"; -import { baseConfigSnapshot, createTestRuntime } from "./test-runtime-config-helpers.js"; - -const configMocks = vi.hoisted(() => ({ - readConfigFileSnapshot: vi.fn(), - writeConfigFile: vi.fn().mockResolvedValue(undefined), -})); - -const offsetMocks = vi.hoisted(() => ({ - deleteTelegramUpdateOffset: vi.fn().mockResolvedValue(undefined), -})); - -vi.mock("../config/config.js", async (importOriginal) => { - const actual = await importOriginal(); - return { - ...actual, - readConfigFileSnapshot: configMocks.readConfigFileSnapshot, - writeConfigFile: configMocks.writeConfigFile, - }; -}); - -vi.mock("../telegram/update-offset-store.js", async (importOriginal) => { - const actual = await importOriginal(); - return { - ...actual, - deleteTelegramUpdateOffset: offsetMocks.deleteTelegramUpdateOffset, - }; -}); - import { channelsAddCommand } from "./channels.js"; +import { configMocks, offsetMocks } from "./channels.mock-harness.js"; +import { baseConfigSnapshot, createTestRuntime } from "./test-runtime-config-helpers.js"; const runtime = createTestRuntime(); diff --git a/src/commands/channels.adds-non-default-telegram-account.e2e.test.ts b/src/commands/channels.adds-non-default-telegram-account.e2e.test.ts index b3d0acbbb28..84f2ff60dbe 100644 --- a/src/commands/channels.adds-non-default-telegram-account.e2e.test.ts +++ b/src/commands/channels.adds-non-default-telegram-account.e2e.test.ts @@ -1,29 +1,12 @@ import { beforeEach, describe, expect, it, vi } from "vitest"; import { setDefaultChannelPluginRegistryForTests } from "./channel-test-helpers.js"; +import { configMocks, offsetMocks } from "./channels.mock-harness.js"; import { baseConfigSnapshot, createTestRuntime } from "./test-runtime-config-helpers.js"; -const configMocks = vi.hoisted(() => ({ - readConfigFileSnapshot: vi.fn(), - writeConfigFile: vi.fn().mockResolvedValue(undefined), -})); - const authMocks = vi.hoisted(() => ({ loadAuthProfileStore: vi.fn(), })); -const offsetMocks = vi.hoisted(() => ({ - deleteTelegramUpdateOffset: vi.fn().mockResolvedValue(undefined), -})); - -vi.mock("../config/config.js", async (importOriginal) => { - const actual = await importOriginal(); - return { - ...actual, - readConfigFileSnapshot: configMocks.readConfigFileSnapshot, - writeConfigFile: configMocks.writeConfigFile, - }; -}); - vi.mock("../agents/auth-profiles.js", async (importOriginal) => { const actual = await importOriginal(); return { @@ -32,14 +15,6 @@ vi.mock("../agents/auth-profiles.js", async (importOriginal) => { }; }); -vi.mock("../telegram/update-offset-store.js", async (importOriginal) => { - const actual = await importOriginal(); - return { - ...actual, - deleteTelegramUpdateOffset: offsetMocks.deleteTelegramUpdateOffset, - }; -}); - import { channelsAddCommand, channelsListCommand, diff --git a/src/commands/channels.mock-harness.ts b/src/commands/channels.mock-harness.ts new file mode 100644 index 00000000000..e26703f60df --- /dev/null +++ b/src/commands/channels.mock-harness.ts @@ -0,0 +1,27 @@ +import { vi } from "vitest"; + +export const configMocks = { + readConfigFileSnapshot: vi.fn(), + writeConfigFile: vi.fn().mockResolvedValue(undefined), +}; + +export const offsetMocks = { + deleteTelegramUpdateOffset: vi.fn().mockResolvedValue(undefined), +}; + +vi.mock("../config/config.js", async (importOriginal) => { + const actual = await importOriginal(); + return { + ...actual, + readConfigFileSnapshot: configMocks.readConfigFileSnapshot, + writeConfigFile: configMocks.writeConfigFile, + }; +}); + +vi.mock("../telegram/update-offset-store.js", async (importOriginal) => { + const actual = await importOriginal(); + return { + ...actual, + deleteTelegramUpdateOffset: offsetMocks.deleteTelegramUpdateOffset, + }; +});