import { afterEach, vi } from "vitest";

import { afterEach, vi, type Mock } from "vitest";

import type {

pollProviderOperationJson,

resolveProviderHttpRequestConfig,

typeof sanitizeConfiguredModelProviderRequest

>[0];

type ResolveProviderHttpRequestConfigResult = {

baseUrl: string;

allowPrivateNetwork: boolean;

headers: Headers;

dispatcherPolicy: undefined;

};

type AnyMock = Mock<(...args: any[]) => any>;

interface ProviderHttpMocks {

resolveApiKeyForProviderMock: Mock<() => Promise<{ apiKey: string }>>;

postJsonRequestMock: AnyMock;

fetchWithTimeoutMock: AnyMock;

pollProviderOperationJsonMock: AnyMock;

assertOkOrThrowHttpErrorMock: Mock<(response: Response, label: string) => Promise<void>>;

assertOkOrThrowProviderErrorMock: Mock<(response: Response, label: string) => Promise<void>>;

sanitizeConfiguredModelProviderRequestMock: Mock<

(

request: SanitizeConfiguredModelProviderRequestParams,

) => SanitizeConfiguredModelProviderRequestParams

>;

resolveProviderHttpRequestConfigMock: Mock<

(params: ResolveProviderHttpRequestConfigParams) => ResolveProviderHttpRequestConfigResult

>;

}

const providerHttpMocks = vi.hoisted(() => ({

resolveApiKeyForProviderMock: vi.fn(async () => ({ apiKey: "provider-key" })),

postJsonRequestMock: vi.fn(),

waitProviderOperationPollInterval: async () => {},

}));

export function getProviderHttpMocks() {

export function getProviderHttpMocks(): ProviderHttpMocks {

return providerHttpMocks;

}