From 1bf5802a68549e748a0aea9264bb293ea75317a8 Mon Sep 17 00:00:00 2001
From: lugnicca <lugnicca@gmail.com>
Date: Fri, 22 Aug 2025 03:13:34 +0200
Subject: [PATCH] refactor: update MockModelProvider type to use ModelProvider
 and clean up test setup

---
 .../hooks/__tests__/useProviderModels.test.ts    | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/web-app/src/hooks/__tests__/useProviderModels.test.ts b/web-app/src/hooks/__tests__/useProviderModels.test.ts
index 3d107b9f8..89bc9e26a 100644
--- a/web-app/src/hooks/__tests__/useProviderModels.test.ts
+++ b/web-app/src/hooks/__tests__/useProviderModels.test.ts
@@ -10,15 +10,13 @@ vi.mock('@/services/providers', () => ({
 import { fetchModelsFromProvider } from '@/services/providers'
 const mockFetchModelsFromProvider = vi.mocked(fetchModelsFromProvider)
 
+import type { ModelProvider } from '@/types/modelProviders'
+
 // Mock ModelProvider type
-type MockModelProvider = {
-  active: boolean
-  provider: string
-  base_url?: string
-  api_key?: string
-  settings: any[]
-  models: any[]
-}
+type MockModelProvider = Pick<
+  ModelProvider,
+  'active' | 'provider' | 'base_url' | 'api_key' | 'settings' | 'models'
+>
 
 describe('useProviderModels', () => {
   const mockProvider: MockModelProvider = {
@@ -34,8 +32,6 @@ describe('useProviderModels', () => {
 
   beforeEach(() => {
     vi.clearAllMocks()
-    // Reset the cache by clearing any previous state
-    mockFetchModelsFromProvider.mockClear()
   })
 
   it('should initialize with empty state', () => {