refactor: update MockModelProvider type to use ModelProvider and clean up test setup

This commit is contained in:
lugnicca 2025-08-22 03:13:34 +02:00
parent 4e8dd9281f
commit 1bf5802a68

View File

@ -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', () => {