dify/web/__mocks__/provider-context.ts

24 lines
702 B
TypeScript

import type { ProviderContextState } from '@/context/provider-context'
import { merge } from 'es-toolkit/compat'
import { noop } from 'es-toolkit/function'
// Avoid being mocked in tests
export const baseProviderContextValue: ProviderContextState = {
modelProviders: [],
modelProviderPlugins: {},
refreshModelProviders: async () => {},
isLoadingModelProviders: false,
isSuccessModelProviders: false,
}
export const createMockProviderContextValue = (
overrides: Partial<ProviderContextState> = {},
): ProviderContextState => {
const merged = merge({}, baseProviderContextValue, overrides)
return {
...merged,
refreshModelProviders: merged.refreshModelProviders ?? noop,
}
}