|
1 | 1 | import { beforeEach, describe, expect, it, vi } from "vitest"; |
2 | 2 | import type { OpenClawConfig } from "../config/types.js"; |
3 | 3 | import type { ImageGenerationProviderPlugin } from "../plugins/types.js"; |
| 4 | +import type * as ProviderRegistry from "./provider-registry.js"; |
4 | 5 | |
5 | 6 | const { resolvePluginCapabilityProvidersMock } = vi.hoisted(() => ({ |
6 | 7 | resolvePluginCapabilityProvidersMock: vi.fn<() => ImageGenerationProviderPlugin[]>(() => []), |
@@ -10,8 +11,8 @@ vi.mock("../plugins/capability-provider-runtime.js", () => ({
|
10 | 11 | resolvePluginCapabilityProviders: resolvePluginCapabilityProvidersMock, |
11 | 12 | })); |
12 | 13 | |
13 | | -const { getImageGenerationProvider, listImageGenerationProviders } = |
14 | | - await import("./provider-registry.js"); |
| 14 | +let getImageGenerationProvider: typeof ProviderRegistry.getImageGenerationProvider; |
| 15 | +let listImageGenerationProviders: typeof ProviderRegistry.listImageGenerationProviders; |
15 | 16 | |
16 | 17 | function createProvider( |
17 | 18 | params: Pick<ImageGenerationProviderPlugin, "id"> & Partial<ImageGenerationProviderPlugin>, |
@@ -30,9 +31,12 @@ function createProvider(
|
30 | 31 | } |
31 | 32 | |
32 | 33 | describe("image-generation provider registry", () => { |
33 | | -beforeEach(() => { |
| 34 | +beforeEach(async () => { |
| 35 | +vi.resetModules(); |
34 | 36 | resolvePluginCapabilityProvidersMock.mockReset(); |
35 | 37 | resolvePluginCapabilityProvidersMock.mockReturnValue([]); |
| 38 | +({ getImageGenerationProvider, listImageGenerationProviders } = |
| 39 | +await import("./provider-registry.js")); |
36 | 40 | }); |
37 | 41 | |
38 | 42 | it("delegates provider resolution to the capability provider boundary", () => { |
|
