diff --git a/src/image-generation/provider-registry.test.ts b/src/image-generation/provider-registry.test.ts index fde4e41c7a8..c2a544ed173 100644 --- a/src/image-generation/provider-registry.test.ts +++ b/src/image-generation/provider-registry.test.ts @@ -1,6 +1,7 @@ import { beforeEach, describe, expect, it, vi } from "vitest"; import type { OpenClawConfig } from "../config/types.js"; import type { ImageGenerationProviderPlugin } from "../plugins/types.js"; +import type * as ProviderRegistry from "./provider-registry.js"; const { resolvePluginCapabilityProvidersMock } = vi.hoisted(() => ({ resolvePluginCapabilityProvidersMock: vi.fn<() => ImageGenerationProviderPlugin[]>(() => []), @@ -10,8 +11,8 @@ vi.mock("../plugins/capability-provider-runtime.js", () => ({ resolvePluginCapabilityProviders: resolvePluginCapabilityProvidersMock, })); -const { getImageGenerationProvider, listImageGenerationProviders } = - await import("./provider-registry.js"); +let getImageGenerationProvider: typeof ProviderRegistry.getImageGenerationProvider; +let listImageGenerationProviders: typeof ProviderRegistry.listImageGenerationProviders; function createProvider( params: Pick & Partial, @@ -30,9 +31,12 @@ function createProvider( } describe("image-generation provider registry", () => { - beforeEach(() => { + beforeEach(async () => { + vi.resetModules(); resolvePluginCapabilityProvidersMock.mockReset(); resolvePluginCapabilityProvidersMock.mockReturnValue([]); + ({ getImageGenerationProvider, listImageGenerationProviders } = + await import("./provider-registry.js")); }); it("delegates provider resolution to the capability provider boundary", () => {