jan/web/hooks/useImportModel.test.ts

// useImportModel.test.ts

import { renderHook, act } from '@testing-library/react'
import { extensionManager } from '@/extension'
import useImportModel from './useImportModel'

// Mock dependencies
jest.mock('@janhq/core')
jest.mock('@/extension')
jest.mock('@/containers/Toast')
jest.mock('uuid', () => ({ v4: () => 'mocked-uuid' }))

describe('useImportModel', () => {
  beforeEach(() => {
    jest.clearAllMocks()
  })

  it('should import models successfully', async () => {
    const mockImportModels = jest.fn().mockResolvedValue(undefined)
    const mockExtension = {
      importModels: mockImportModels,
    } as any

    jest.spyOn(extensionManager, 'get').mockReturnValue(mockExtension)

    const { result } = renderHook(() => useImportModel())

    const models = [
      { importId: '1', name: 'Model 1', path: '/path/to/model1' },
      { importId: '2', name: 'Model 2', path: '/path/to/model2' },
    ] as any

    await act(async () => {
      await result.current.importModels(models, 'local' as any)
    })

    expect(mockImportModels).toHaveBeenCalledWith(models, 'local')
  })

  it('should update model info successfully', async () => {
    const mockUpdateModelInfo = jest
      .fn()
      .mockResolvedValue({ id: 'model-1', name: 'Updated Model' })
    const mockExtension = {
      updateModelInfo: mockUpdateModelInfo,
    } as any

    jest.spyOn(extensionManager, 'get').mockReturnValue(mockExtension)

    const { result } = renderHook(() => useImportModel())

    const modelInfo = { id: 'model-1', name: 'Updated Model' }

    await act(async () => {
      await result.current.updateModelInfo(modelInfo)
    })

    expect(mockUpdateModelInfo).toHaveBeenCalledWith(modelInfo)
  })

  it('should handle empty file paths', async () => {
    const { result } = renderHook(() => useImportModel())

    await act(async () => {
      await result.current.sanitizeFilePaths([])
    })

    // Expect no state changes or side effects
  })
})