vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-06 14:20:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: update provider HTTP mocks

Browse Source
This commit is contained in:
Barron Roth
2026-04-25 18:34:37 -07:00
committed by Ayaan Zaidi
parent 9975de89d1
commit f801fe7d27
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/helpers/media-generation/provider-http-mocks.ts
View File

@@ -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<typeof pollProviderOperationJson>[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();
});
}