diff --git a/test/helpers/media-generation/provider-http-mocks.ts b/test/helpers/media-generation/provider-http-mocks.ts index 56ab02b641b..e66cbbfd26d 100644 --- a/test/helpers/media-generation/provider-http-mocks.ts +++ b/test/helpers/media-generation/provider-http-mocks.ts @@ -1,6 +1,7 @@ import type { pollProviderOperationJson, resolveProviderHttpRequestConfig, + sanitizeConfiguredModelProviderRequest, } from "openclaw/plugin-sdk/provider-http"; import { afterEach, vi } from "vitest"; @@ -8,6 +9,9 @@ type ResolveProviderHttpRequestConfigParams = Parameters< typeof resolveProviderHttpRequestConfig >[0]; type PollProviderOperationJsonParams = Parameters[0]; +type SanitizeConfiguredModelProviderRequestParams = Parameters< + typeof sanitizeConfiguredModelProviderRequest +>[0]; const providerHttpMocks = vi.hoisted(() => ({ resolveApiKeyForProviderMock: vi.fn(async () => ({ apiKey: "provider-key" })), @@ -16,9 +20,12 @@ const providerHttpMocks = vi.hoisted(() => ({ pollProviderOperationJsonMock: vi.fn(), assertOkOrThrowHttpErrorMock: vi.fn(async (_response: Response, _label: string) => {}), assertOkOrThrowProviderErrorMock: vi.fn(async (_response: Response, _label: string) => {}), + sanitizeConfiguredModelProviderRequestMock: vi.fn( + (request: SanitizeConfiguredModelProviderRequestParams) => request, + ), resolveProviderHttpRequestConfigMock: vi.fn((params: ResolveProviderHttpRequestConfigParams) => ({ baseUrl: params.baseUrl ?? params.defaultBaseUrl, - allowPrivateNetwork: false, + allowPrivateNetwork: params.allowPrivateNetwork === true, headers: new Headers(params.defaultHeaders), dispatcherPolicy: undefined, })), @@ -73,6 +80,8 @@ vi.mock("openclaw/plugin-sdk/provider-http", () => ({ resolveProviderOperationTimeoutMs: ({ defaultTimeoutMs }: { defaultTimeoutMs: number }) => defaultTimeoutMs, resolveProviderHttpRequestConfig: providerHttpMocks.resolveProviderHttpRequestConfigMock, + sanitizeConfiguredModelProviderRequest: + providerHttpMocks.sanitizeConfiguredModelProviderRequestMock, waitProviderOperationPollInterval: async () => {}, })); @@ -88,6 +97,7 @@ export function installProviderHttpMockCleanup(): void { providerHttpMocks.pollProviderOperationJsonMock.mockClear(); providerHttpMocks.assertOkOrThrowHttpErrorMock.mockClear(); providerHttpMocks.assertOkOrThrowProviderErrorMock.mockClear(); + providerHttpMocks.sanitizeConfiguredModelProviderRequestMock.mockClear(); providerHttpMocks.resolveProviderHttpRequestConfigMock.mockClear(); }); }