@lobehub/chat/src/hooks/useInterceptingRoutes.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 { INBOX_SESSION_ID } from '@/const/session';
import { useIsMobile } from '@/hooks/useIsMobile';
import { useAgentStore } from '@/store/agent';
import { ChatSettingsTabs } from '@/store/global/initialState';
import { useSessionStore } from '@/store/session';

import { useOpenChatSettings } from './useInterceptingRoutes';

// Mocks
vi.mock('next/navigation', () => ({
  useRouter: vi.fn(() => ({
    push: vi.fn((href) => href),
    replace: vi.fn((href) => href),
  })),
}));
vi.mock('@/hooks/useQuery', () => ({
  useQuery: vi.fn(() => ({})),
}));
vi.mock('@/hooks/useIsMobile', () => ({
  useIsMobile: vi.fn(),
}));
vi.mock('@/store/session', () => ({
  useSessionStore: vi.fn(),
}));
vi.mock('@/store/global', () => ({
  useGlobalStore: {
    setState: vi.fn(),
  },
}));

describe('useOpenChatSettings', () => {
  it('should handle inbox session id correctly', () => {
    vi.mocked(useSessionStore).mockReturnValue(INBOX_SESSION_ID);
    const { result } = renderHook(() => useOpenChatSettings());

    expect(result.current()).toBe('/settings/agent'); // Assuming openSettings returns a function
  });

  it('should handle mobile route for chat settings', () => {
    vi.mocked(useSessionStore).mockReturnValue('123');
    vi.mocked(useIsMobile).mockReturnValue(true);
    const { result } = renderHook(() => useOpenChatSettings(ChatSettingsTabs.Meta));
    expect(result.current()).toBe('/chat/settings?session=123');
  });

  it('should handle desktop route for chat settings with session and tab', () => {
    vi.mocked(useSessionStore).mockReturnValue('456');
    vi.mocked(useIsMobile).mockReturnValue(false);

    const { result } = renderHook(() => useOpenChatSettings(ChatSettingsTabs.Meta));

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

    expect(useAgentStore.getState().showAgentSetting).toBeTruthy();
  });
});