refactor: update MockModelProvider type to use ModelProvider and clean up test setup
This commit is contained in:
parent
4e8dd9281f
commit
1bf5802a68
@ -10,15 +10,13 @@ vi.mock('@/services/providers', () => ({
|
|||||||
import { fetchModelsFromProvider } from '@/services/providers'
|
import { fetchModelsFromProvider } from '@/services/providers'
|
||||||
const mockFetchModelsFromProvider = vi.mocked(fetchModelsFromProvider)
|
const mockFetchModelsFromProvider = vi.mocked(fetchModelsFromProvider)
|
||||||
|
|
||||||
|
import type { ModelProvider } from '@/types/modelProviders'
|
||||||
|
|
||||||
// Mock ModelProvider type
|
// Mock ModelProvider type
|
||||||
type MockModelProvider = {
|
type MockModelProvider = Pick<
|
||||||
active: boolean
|
ModelProvider,
|
||||||
provider: string
|
'active' | 'provider' | 'base_url' | 'api_key' | 'settings' | 'models'
|
||||||
base_url?: string
|
>
|
||||||
api_key?: string
|
|
||||||
settings: any[]
|
|
||||||
models: any[]
|
|
||||||
}
|
|
||||||
|
|
||||||
describe('useProviderModels', () => {
|
describe('useProviderModels', () => {
|
||||||
const mockProvider: MockModelProvider = {
|
const mockProvider: MockModelProvider = {
|
||||||
@ -34,8 +32,6 @@ describe('useProviderModels', () => {
|
|||||||
|
|
||||||
beforeEach(() => {
|
beforeEach(() => {
|
||||||
vi.clearAllMocks()
|
vi.clearAllMocks()
|
||||||
// Reset the cache by clearing any previous state
|
|
||||||
mockFetchModelsFromProvider.mockClear()
|
|
||||||
})
|
})
|
||||||
|
|
||||||
it('should initialize with empty state', () => {
|
it('should initialize with empty state', () => {
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user