@lobehub/chat/src/store/file/slices/chat/action.test.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import { act, renderHook } from '@testing-library/react';
import useSWR from 'swr';
import { Mock, afterEach, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';

import { notification } from '@/components/AntdStaticMethods';
import { DB_File } from '@/database/_deprecated/schemas/files';
import { fileService } from '@/services/file';
import { uploadService } from '@/services/upload';

import { useFileStore as useStore } from '../../store';

vi.mock('zustand/traditional');

// Mock necessary modules and functions
vi.mock('@/components/AntdStaticMethods', () => ({
  notification: {
    error: vi.fn(),
  },
}));

// Mock for useSWR
vi.mock('swr', () => ({
  default: vi.fn(),
}));

//  mock the arrayBuffer
beforeAll(() => {
  Object.defineProperty(File.prototype, 'arrayBuffer', {
    writable: true,
    value: function () {
      return new Promise((resolve) => {
        const reader = new FileReader();
        reader.onloadend = () => {
          resolve(reader.result);
        };
        reader.readAsArrayBuffer(this);
      });
    },
  });
});

beforeEach(() => {
  // Reset all mocks before each test
  vi.resetAllMocks();
});

describe('useFileStore:chat', () => {
  it('clearChatUploadFileList should clear the inputFilesList', () => {
    const { result } = renderHook(() => useStore());

    // Populate the list to clear it later
    act(() => {
      useStore.setState({ chatUploadFileList: [{ id: 'abc' }] as any });
    });

    expect(result.current.chatUploadFileList).toEqual([{ id: 'abc' }]);

    act(() => {
      result.current.clearChatUploadFileList();
    });

    expect(result.current.chatUploadFileList).toEqual([]);
  });

  // it('removeFile should call fileService.removeFile and update the store', async () => {
  //   const { result } = renderHook(() => useStore());
  //
  //   const fileId = 'test-id';
  //
  //   // Mock the fileService.removeFile to resolve
  //   vi.spyOn(fileService, 'removeFile').mockResolvedValue(undefined);
  //
  //   // Populate the list to remove an item later
  //   act(() => {
  //     useStore.setState(({ inputFilesList }) => ({ inputFilesList: [...inputFilesList, fileId] }));
  //     //   // result.current.inputFilesList.push(fileId);
  //   });
  //
  //   await act(async () => {
  //     await result.current.removeFile(fileId);
  //   });
  //
  //   expect(fileService.removeFile).toHaveBeenCalledWith(fileId);
  //   expect(result.current.inputFilesList).toEqual([]);
  // });

  // describe('uploadFile', () => {
  //   it('uploadFile should handle errors', async () => {
  //     const { result } = renderHook(() => useStore());
  //     const testFile = new File(['content'], 'test.png', { type: 'image/png' });
  //
  //     // 模拟 fileService.uploadFile 抛出错误
  //     const errorMessage = 'Upload failed';
  //     vi.spyOn(uploadService, 'uploadFile').mockRejectedValue(new Error(errorMessage));
  //
  //     // Mock console.error for testing
  //
  //     await act(async () => {
  //       await result.current.uploadFile(testFile);
  //     });
  //
  //     expect(uploadService.uploadFile).toHaveBeenCalledWith({
  //       createdAt: testFile.lastModified,
  //       data: await testFile.arrayBuffer(),
  //       fileType: testFile.type,
  //       name: testFile.name,
  //       saveMode: 'local',
  //       size: testFile.size,
  //     });
  //     // 由于上传失败，inputFilesList 应该没有变化
  //     expect(result.current.inputFilesList).toEqual([]);
  //
  //     // 确保错误提示被调用
  //     expect(notification.error).toHaveBeenCalled();
  //   });
  //
  //   it('uploadFile should upload the file and update inputFilesList', async () => {
  //     const { result } = renderHook(() => useStore());
  //     const testFile = new File(['content'], 'test.png', { type: 'image/png' });
  //
  //     // 模拟 fileService.uploadFile 返回的数据
  //     const uploadedFileData = {
  //       createdAt: testFile.lastModified,
  //       data: await testFile.arrayBuffer(),
  //       fileType: testFile.type,
  //       name: testFile.name,
  //       saveMode: 'local',
  //       size: testFile.size,
  //     };
  //
  //     // Mock the fileService.uploadFile to resolve with uploadedFileData
  //     vi.spyOn(uploadService, 'uploadFile').mockResolvedValue(uploadedFileData as DB_File);
  //     vi.spyOn(fileService, 'createFile').mockResolvedValue({ id: 'new-file-id', url: '' });
  //
  //     await act(async () => {
  //       await result.current.uploadFile(testFile);
  //     });
  //
  //     expect(fileService.createFile).toHaveBeenCalledWith({
  //       createdAt: testFile.lastModified,
  //       data: await testFile.arrayBuffer(),
  //       fileType: testFile.type,
  //       name: testFile.name,
  //       saveMode: 'local',
  //       size: testFile.size,
  //     });
  //     expect(result.current.inputFilesList).toContain('new-file-id');
  //   });
  // });
});