@lobehub/chat/src/store/chat/slices/portal/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 { describe, expect, it, vi } from 'vitest';

import { useChatStore } from '@/store/chat';

vi.mock('zustand/traditional');

describe('chatDockSlice', () => {
  describe('closeToolUI', () => {
    it('should set dockToolMessage to undefined', () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        result.current.openToolUI('test-id', 'test-identifier');
      });

      expect(result.current.portalToolMessage).toEqual({
        id: 'test-id',
        identifier: 'test-identifier',
      });

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

      expect(result.current.portalToolMessage).toBeUndefined();
    });
  });

  describe('openToolUI', () => {
    it('should set dockToolMessage and open dock if it is closed', () => {
      const { result } = renderHook(() => useChatStore());

      expect(result.current.showPortal).toBe(false);

      act(() => {
        result.current.openToolUI('test-id', 'test-identifier');
      });

      expect(result.current.portalToolMessage).toEqual({
        id: 'test-id',
        identifier: 'test-identifier',
      });
      expect(result.current.showPortal).toBe(true);
    });

    it('should not change dock state if it is already open', () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        result.current.togglePortal(true);
      });

      expect(result.current.showPortal).toBe(true);

      act(() => {
        result.current.openToolUI('test-id', 'test-identifier');
      });

      expect(result.current.portalToolMessage).toEqual({
        id: 'test-id',
        identifier: 'test-identifier',
      });
      expect(result.current.showPortal).toBe(true);
    });
  });

  describe('toggleDock', () => {
    it('should toggle dock state when no argument is provided', () => {
      const { result } = renderHook(() => useChatStore());

      expect(result.current.showPortal).toBe(false);

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

      expect(result.current.showPortal).toBe(true);

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

      expect(result.current.showPortal).toBe(false);
    });

    it('should set dock state to the provided value', () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        result.current.togglePortal(true);
      });

      expect(result.current.showPortal).toBe(true);

      act(() => {
        result.current.togglePortal(false);
      });

      expect(result.current.showPortal).toBe(false);

      act(() => {
        result.current.togglePortal(true);
      });

      expect(result.current.showPortal).toBe(true);
    });
  });
});