From d2fa38f0812ca898c8967e85debfcfd60fefc8f6 Mon Sep 17 00:00:00 2001
From: Louis <louis@jan.ai>
Date: Tue, 5 Nov 2024 08:33:58 +0700
Subject: [PATCH] test: correct tests

---
 web/hooks/useImportModel.test.ts | 6 +++---
 web/hooks/useModels.test.ts      | 2 ++
 web/hooks/useModels.ts           | 6 +++---
 3 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/web/hooks/useImportModel.test.ts b/web/hooks/useImportModel.test.ts
index 9b623226d..571947903 100644
--- a/web/hooks/useImportModel.test.ts
+++ b/web/hooks/useImportModel.test.ts
@@ -31,11 +31,11 @@ describe('useImportModel', () => {
     ] as any
 
     await act(async () => {
-      await result.current.importModels(models, 'local' as any)
+      await result.current.importModels(models, 'copy')
     })
 
-    expect(mockImportModels).toHaveBeenCalledWith('1', '/path/to/model1', undefined)
-    expect(mockImportModels).toHaveBeenCalledWith('2', '/path/to/model2', undefined)
+    expect(mockImportModels).toHaveBeenCalledWith('1', '/path/to/model1', undefined,'copy')
+    expect(mockImportModels).toHaveBeenCalledWith('2', '/path/to/model2', undefined, 'copy')
   })
 
   it('should update model info successfully', async () => {
diff --git a/web/hooks/useModels.test.ts b/web/hooks/useModels.test.ts
index 0440b5443..9b6b898ad 100644
--- a/web/hooks/useModels.test.ts
+++ b/web/hooks/useModels.test.ts
@@ -35,6 +35,8 @@ describe('useModels', () => {
           }),
         }),
         get: () => undefined,
+        has: () => true,
+        // set: () => {}
       },
     })
 
diff --git a/web/hooks/useModels.ts b/web/hooks/useModels.ts
index f3004f823..c856f6e3c 100644
--- a/web/hooks/useModels.ts
+++ b/web/hooks/useModels.ts
@@ -58,13 +58,13 @@ const useModels = () => {
       }
     }
 
-    const getExtensionModels = async () => {
+    const getExtensionModels = () => {
       const models = ModelManager.instance().models.values().toArray()
       setExtensionModels(models)
     }
-
     // Fetch all data
-    getExtensionModels().then(getDownloadedModels)
+    getExtensionModels()
+    getDownloadedModels()
   }, [setDownloadedModels, setExtensionModels])
 
   const reloadData = useDebouncedCallback(() => getData(), 300)