@lobehub/chat/src/store/chat/slices/thread/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 { UIChatMessage } from '@lobechat/types';
import { act, renderHook, waitFor } from '@testing-library/react';
import { mutate } from 'swr';
import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';

import { THREAD_DRAFT_ID } from '@/const/message';
import { chatService } from '@/services/chat';
import { threadService } from '@/services/thread';
import { messageMapKey } from '@/store/chat/utils/messageMapKey';
import { useSessionStore } from '@/store/session';
import { ThreadItem, ThreadStatus, ThreadType } from '@/types/topic';

import { useChatStore } from '../../store';

vi.mock('zustand/traditional');

// Mock version constants
vi.mock('@/const/version', () => ({
  isDeprecatedEdition: false,
  isDesktop: false,
}));

// Mock threadService
vi.mock('@/services/thread', () => ({
  threadService: {
    createThreadWithMessage: vi.fn(),
    getThreads: vi.fn(),
    removeThread: vi.fn(),
    updateThread: vi.fn(),
  },
}));

// Mock chatService
vi.mock('@/services/chat', () => ({
  chatService: {
    fetchPresetTaskResult: vi.fn(),
  },
}));

// Mock mutate from SWR
vi.mock('swr', async () => {
  const actual = await vi.importActual('swr');
  return {
    ...actual,
    mutate: vi.fn(),
  };
});

// Mock store helpers
vi.mock('@/store/global/helpers', () => ({
  globalHelpers: {
    getCurrentLanguage: vi.fn(() => 'en-US'),
  },
}));

vi.mock('@/store/session', () => ({
  useSessionStore: {
    getState: vi.fn(() => ({
      triggerSessionUpdate: vi.fn(),
    })),
  },
}));

vi.mock('@/store/user', () => ({
  useUserStore: {
    getState: vi.fn(() => ({})),
  },
}));

vi.mock('@/store/user/selectors', () => ({
  systemAgentSelectors: {
    thread: vi.fn(() => ({})),
  },
  userProfileSelectors: {
    userAvatar: vi.fn(() => 'avatar-url'),
  },
}));

beforeEach(() => {
  vi.clearAllMocks();
  useChatStore.setState(
    {
      activeId: 'test-session-id',
      activeTopicId: 'test-topic-id',
      isCreatingThread: false,
      isCreatingThreadMessage: false,
      messagesMap: {},
      newThreadMode: ThreadType.Continuation,
      portalThreadId: undefined,
      startToForkThread: undefined,
      threadInputMessage: '',
      threadLoadingIds: [],
      threadMaps: {},
      threadStartMessageId: undefined,
      threadsInit: false,
    },
    false,
  );
});

afterEach(() => {
  vi.restoreAllMocks();
});

describe('thread action', () => {
  describe('updateThreadInputMessage', () => {
    it('should update thread input message', () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        result.current.updateThreadInputMessage('test message');
      });

      expect(result.current.threadInputMessage).toBe('test message');
    });

    it('should not update if message is the same', () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        useChatStore.setState({ threadInputMessage: 'test message' });
      });

      const stateBefore = useChatStore.getState();

      act(() => {
        result.current.updateThreadInputMessage('test message');
      });

      expect(useChatStore.getState()).toBe(stateBefore);
    });
  });

  describe('openThreadCreator', () => {
    it('should set thread creator state and open portal', () => {
      const { result } = renderHook(() => useChatStore());
      const togglePortalSpy = vi.spyOn(result.current, 'togglePortal');

      act(() => {
        result.current.openThreadCreator('message-id');
      });

      expect(result.current.threadStartMessageId).toBe('message-id');
      expect(result.current.portalThreadId).toBeUndefined();
      expect(result.current.startToForkThread).toBe(true);
      expect(togglePortalSpy).toHaveBeenCalledWith(true);
    });
  });

  describe('openThreadInPortal', () => {
    it('should set portal thread state and open portal', () => {
      const { result } = renderHook(() => useChatStore());
      const togglePortalSpy = vi.spyOn(result.current, 'togglePortal');

      act(() => {
        result.current.openThreadInPortal('thread-id', 'source-message-id');
      });

      expect(result.current.portalThreadId).toBe('thread-id');
      expect(result.current.threadStartMessageId).toBe('source-message-id');
      expect(result.current.startToForkThread).toBe(false);
      expect(togglePortalSpy).toHaveBeenCalledWith(true);
    });
  });

  describe('closeThreadPortal', () => {
    it('should clear thread portal state and close portal', () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        useChatStore.setState({
          portalThreadId: 'thread-id',
          startToForkThread: true,
          threadStartMessageId: 'message-id',
        });
      });

      const togglePortalSpy = vi.spyOn(result.current, 'togglePortal');

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

      expect(result.current.portalThreadId).toBeUndefined();
      expect(result.current.threadStartMessageId).toBeUndefined();
      expect(result.current.startToForkThread).toBeUndefined();
      expect(togglePortalSpy).toHaveBeenCalledWith(false);
    });
  });

  describe('switchThread', () => {
    it('should set active thread id', () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        result.current.switchThread('thread-id');
      });

      expect(result.current.activeThreadId).toBe('thread-id');
    });
  });

  describe('createThread', () => {
    it('should create thread with message and return ids', async () => {
      const { result } = renderHook(() => useChatStore());

      const mockResult = { messageId: 'new-message-id', threadId: 'new-thread-id' };
      (threadService.createThreadWithMessage as Mock).mockResolvedValue(mockResult);

      let createResult;
      await act(async () => {
        createResult = await result.current.createThread({
          message: {
            content: 'test message',
            role: 'user',
            sessionId: 'test-session-id',
          },
          sourceMessageId: 'source-msg-id',
          topicId: 'test-topic-id',
          type: ThreadType.Continuation,
        });
      });

      expect(threadService.createThreadWithMessage).toHaveBeenCalledWith({
        message: {
          content: 'test message',
          role: 'user',
          sessionId: 'test-session-id',
        },
        sourceMessageId: 'source-msg-id',
        topicId: 'test-topic-id',
        type: ThreadType.Continuation,
      });
      expect(createResult).toEqual(mockResult);
      expect(result.current.isCreatingThread).toBe(false);
    });

    it('should set isCreatingThread during creation', async () => {
      const { result } = renderHook(() => useChatStore());

      (threadService.createThreadWithMessage as Mock).mockImplementation(async () => {
        expect(useChatStore.getState().isCreatingThread).toBe(true);
        return { messageId: 'message-id', threadId: 'thread-id' };
      });

      await act(async () => {
        await result.current.createThread({
          message: { content: 'test', role: 'user', sessionId: 'test-session-id' },
          sourceMessageId: 'source-msg-id',
          topicId: 'test-topic-id',
          type: ThreadType.Continuation,
        });
      });

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

  describe('useFetchThreads', () => {
    it('should fetch threads for a given topic id', async () => {
      const topicId = 'test-topic-id';
      const threads: ThreadItem[] = [
        {
          createdAt: new Date(),
          id: 'thread-1',
          lastActiveAt: new Date(),
          sourceMessageId: 'msg-1',
          status: ThreadStatus.Active,
          title: 'Thread 1',
          topicId,
          type: ThreadType.Continuation,
          updatedAt: new Date(),
          userId: 'user-1',
        },
      ];

      (threadService.getThreads as Mock).mockResolvedValue(threads);

      const { result } = renderHook(() => useChatStore().useFetchThreads(true, topicId));

      await waitFor(() => {
        expect(result.current.data).toEqual(threads);
      });

      expect(useChatStore.getState().threadsInit).toBeTruthy();
      expect(useChatStore.getState().threadMaps).toEqual({ [topicId]: threads });
    });

    it('should not fetch when enable is false', async () => {
      const topicId = 'test-topic-id';

      const { result } = renderHook(() => useChatStore().useFetchThreads(false, topicId));

      expect(threadService.getThreads).not.toHaveBeenCalled();
      expect(result.current.data).toBeUndefined();
    });

    it('should not fetch when topicId is undefined', async () => {
      const { result } = renderHook(() => useChatStore().useFetchThreads(true, undefined));

      expect(threadService.getThreads).not.toHaveBeenCalled();
      expect(result.current.data).toBeUndefined();
    });
  });

  describe('refreshThreads', () => {
    it('should trigger SWR mutate for active topic', async () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        useChatStore.setState({ activeTopicId: 'test-topic-id' });
      });

      await act(async () => {
        await result.current.refreshThreads();
      });

      expect(mutate).toHaveBeenCalledWith(['SWR_USE_FETCH_THREADS', 'test-topic-id']);
    });

    it('should not mutate when activeTopicId is undefined', async () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        useChatStore.setState({ activeTopicId: undefined });
      });

      await act(async () => {
        await result.current.refreshThreads();
      });

      expect(mutate).not.toHaveBeenCalled();
    });
  });

  describe('removeThread', () => {
    it('should remove thread and refresh threads', async () => {
      const { result } = renderHook(() => useChatStore());

      (threadService.removeThread as Mock).mockResolvedValue(undefined);

      const refreshThreadsSpy = vi.spyOn(result.current, 'refreshThreads').mockResolvedValue();

      await act(async () => {
        await result.current.removeThread('thread-id');
      });

      expect(threadService.removeThread).toHaveBeenCalledWith('thread-id');
      expect(refreshThreadsSpy).toHaveBeenCalled();
    });

    it('should clear activeThreadId if removing active thread', async () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        useChatStore.setState({ activeThreadId: 'thread-id' });
      });

      (threadService.removeThread as Mock).mockResolvedValue(undefined);
      vi.spyOn(result.current, 'refreshThreads').mockResolvedValue();

      await act(async () => {
        await result.current.removeThread('thread-id');
      });

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

    it('should not clear activeThreadId if removing different thread', async () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        useChatStore.setState({ activeThreadId: 'active-thread-id' });
      });

      (threadService.removeThread as Mock).mockResolvedValue(undefined);
      vi.spyOn(result.current, 'refreshThreads').mockResolvedValue();

      await act(async () => {
        await result.current.removeThread('different-thread-id');
      });

      expect(result.current.activeThreadId).toBe('active-thread-id');
    });
  });

  describe('updateThreadTitle', () => {
    it('should update thread title via internal_updateThread', async () => {
      const { result } = renderHook(() => useChatStore());

      const internalUpdateSpy = vi
        .spyOn(result.current, 'internal_updateThread')
        .mockResolvedValue();

      await act(async () => {
        await result.current.updateThreadTitle('thread-id', 'New Title');
      });

      expect(internalUpdateSpy).toHaveBeenCalledWith('thread-id', { title: 'New Title' });
    });
  });

  describe('summaryThreadTitle', () => {
    it('should generate and update thread title via AI', async () => {
      const { result } = renderHook(() => useChatStore());

      const mockThread: ThreadItem = {
        createdAt: new Date(),
        id: 'thread-id',
        lastActiveAt: new Date(),
        sourceMessageId: 'msg-1',
        status: ThreadStatus.Active,
        title: 'Old Title',
        topicId: 'test-topic-id',
        type: ThreadType.Continuation,
        updatedAt: new Date(),
        userId: 'user-1',
      };

      act(() => {
        useChatStore.setState({
          portalThreadId: 'thread-id',
          threadMaps: {
            'test-topic-id': [mockThread],
          },
        });
      });

      const messages: UIChatMessage[] = [
        {
          content: 'Hello',
          createdAt: Date.now(),
          id: 'msg-1',
          meta: {},
          role: 'user',
          sessionId: 'test-session-id',
          updatedAt: Date.now(),
        },
      ];

      (chatService.fetchPresetTaskResult as Mock).mockImplementation(
        async ({ onMessageHandle, onFinish }) => {
          await onMessageHandle?.({ text: 'New', type: 'text' });
          await onMessageHandle?.({ text: ' Generated', type: 'text' });
          await onMessageHandle?.({ text: ' Title', type: 'text' });
          await onFinish?.('New Generated Title');
        },
      );

      const internalUpdateSpy = vi
        .spyOn(result.current, 'internal_updateThread')
        .mockResolvedValue();

      await act(async () => {
        await result.current.summaryThreadTitle('thread-id', messages);
      });

      expect(chatService.fetchPresetTaskResult).toHaveBeenCalled();
      expect(internalUpdateSpy).toHaveBeenCalledWith('thread-id', {
        title: 'New Generated Title',
      });
    });

    it('should show loading indicator during generation', async () => {
      const { result } = renderHook(() => useChatStore());

      const mockThread: ThreadItem = {
        createdAt: new Date(),
        id: 'thread-id',
        lastActiveAt: new Date(),
        sourceMessageId: 'msg-1',
        status: ThreadStatus.Active,
        title: 'Old Title',
        topicId: 'test-topic-id',
        type: ThreadType.Continuation,
        updatedAt: new Date(),
        userId: 'user-1',
      };

      act(() => {
        useChatStore.setState({
          portalThreadId: 'thread-id',
          threadMaps: {
            'test-topic-id': [mockThread],
          },
        });
      });

      (chatService.fetchPresetTaskResult as Mock).mockImplementation(
        async ({ onLoadingChange, onFinish }) => {
          await onLoadingChange?.(true);
          await onFinish?.('Title');
          await onLoadingChange?.(false);
        },
      );

      vi.spyOn(result.current, 'internal_updateThread').mockResolvedValue();

      await act(async () => {
        await result.current.summaryThreadTitle('thread-id', []);
      });

      expect(chatService.fetchPresetTaskResult).toHaveBeenCalled();
    });

    it('should revert title on error', async () => {
      const { result } = renderHook(() => useChatStore());

      const mockThread: ThreadItem = {
        createdAt: new Date(),
        id: 'thread-id',
        lastActiveAt: new Date(),
        sourceMessageId: 'msg-1',
        status: ThreadStatus.Active,
        title: 'Old Title',
        topicId: 'test-topic-id',
        type: ThreadType.Continuation,
        updatedAt: new Date(),
        userId: 'user-1',
      };

      act(() => {
        useChatStore.setState({
          portalThreadId: 'thread-id',
          threadMaps: {
            'test-topic-id': [mockThread],
          },
        });
      });

      (chatService.fetchPresetTaskResult as Mock).mockImplementation(async ({ onError }) => {
        await onError?.();
      });

      vi.spyOn(result.current, 'internal_updateThread').mockResolvedValue();

      await act(async () => {
        await result.current.summaryThreadTitle('thread-id', []);
      });

      // Should have called with LOADING_FLAT first, then reverted to old title on error
      expect(chatService.fetchPresetTaskResult).toHaveBeenCalled();
    });

    it('should not run if no portal thread found', async () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        useChatStore.setState({
          portalThreadId: undefined,
        });
      });

      await act(async () => {
        await result.current.summaryThreadTitle('thread-id', []);
      });

      expect(chatService.fetchPresetTaskResult).not.toHaveBeenCalled();
    });
  });

  describe('sendThreadMessage', () => {
    describe('validation', () => {
      it('should not send when activeId is undefined', async () => {
        const { result } = renderHook(() => useChatStore());

        act(() => {
          useChatStore.setState({ activeId: undefined });
        });

        await act(async () => {
          await result.current.sendThreadMessage({ message: 'test' });
        });

        expect(useChatStore.getState().isCreatingThreadMessage).toBeFalsy();
      });

      it('should not send when activeTopicId is undefined', async () => {
        const { result } = renderHook(() => useChatStore());

        act(() => {
          useChatStore.setState({ activeTopicId: undefined });
        });

        await act(async () => {
          await result.current.sendThreadMessage({ message: 'test' });
        });

        expect(useChatStore.getState().isCreatingThreadMessage).toBeFalsy();
      });

      it('should not send when message is empty', async () => {
        const { result } = renderHook(() => useChatStore());

        await act(async () => {
          await result.current.sendThreadMessage({ message: '' });
        });

        expect(useChatStore.getState().isCreatingThreadMessage).toBeFalsy();
      });
    });

    describe('new thread creation flow', () => {
      it('should create new thread and send first message', async () => {
        const { result } = renderHook(() => useChatStore());

        act(() => {
          useChatStore.setState({
            newThreadMode: ThreadType.Continuation,
            portalThreadId: undefined,
            threadStartMessageId: 'source-msg-id',
          });
        });

        const createThreadSpy = vi
          .spyOn(result.current, 'createThread')
          .mockResolvedValue({ messageId: 'new-msg-id', threadId: 'new-thread-id' });

        const refreshThreadsSpy = vi.spyOn(result.current, 'refreshThreads').mockResolvedValue();
        const refreshMessagesSpy = vi.spyOn(result.current, 'refreshMessages').mockResolvedValue();
        const openThreadSpy = vi.spyOn(result.current, 'openThreadInPortal');
        const coreProcessSpy = vi
          .spyOn(result.current, 'internal_coreProcessMessage')
          .mockResolvedValue();
        vi.spyOn(result.current, 'internal_createTmpMessage');
        vi.spyOn(result.current, 'internal_toggleMessageLoading');

        await act(async () => {
          await result.current.sendThreadMessage({ message: 'test message' });
        });

        expect(createThreadSpy).toHaveBeenCalledWith({
          message: expect.objectContaining({
            content: 'test message',
            role: 'user',
            sessionId: 'test-session-id',
            threadId: undefined,
            topicId: 'test-topic-id',
          }),
          sourceMessageId: 'source-msg-id',
          topicId: 'test-topic-id',
          type: ThreadType.Continuation,
        });

        expect(refreshThreadsSpy).toHaveBeenCalled();
        expect(refreshMessagesSpy).toHaveBeenCalled();
        expect(openThreadSpy).toHaveBeenCalledWith('new-thread-id', 'source-msg-id');
        expect(coreProcessSpy).toHaveBeenCalled();
      });

      it('should use temp message with THREAD_DRAFT_ID for optimistic update', async () => {
        const { result } = renderHook(() => useChatStore());

        act(() => {
          useChatStore.setState({
            portalThreadId: undefined,
            threadStartMessageId: 'source-msg-id',
          });
        });

        const createTmpSpy = vi
          .spyOn(result.current, 'internal_createTmpMessage')
          .mockReturnValue('temp-msg-id');

        vi.spyOn(result.current, 'createThread').mockResolvedValue({
          messageId: 'new-msg-id',
          threadId: 'new-thread-id',
        });
        vi.spyOn(result.current, 'refreshThreads').mockResolvedValue();
        vi.spyOn(result.current, 'refreshMessages').mockResolvedValue();
        vi.spyOn(result.current, 'openThreadInPortal');
        vi.spyOn(result.current, 'internal_coreProcessMessage').mockResolvedValue();
        vi.spyOn(result.current, 'internal_toggleMessageLoading');

        await act(async () => {
          await result.current.sendThreadMessage({ message: 'test message' });
        });

        expect(createTmpSpy).toHaveBeenCalledWith(
          expect.objectContaining({
            threadId: THREAD_DRAFT_ID,
          }),
        );
      });

      it('should auto-summarize thread title after first message', async () => {
        const { result } = renderHook(() => useChatStore());

        const mockThread: ThreadItem = {
          createdAt: new Date(),
          id: 'new-thread-id',
          lastActiveAt: new Date(),
          sourceMessageId: 'msg-1',
          status: ThreadStatus.Active,
          title: 'test message',
          topicId: 'test-topic-id',
          type: ThreadType.Continuation,
          updatedAt: new Date(),
          userId: 'user-1',
        };

        act(() => {
          useChatStore.setState({
            messagesMap: {
              [messageMapKey('test-session-id', 'test-topic-id')]: [
                {
                  content: 'test',
                  createdAt: Date.now(),
                  id: 'msg-1',
                  meta: {},
                  role: 'user',
                  sessionId: 'test-session-id',
                  updatedAt: Date.now(),
                },
              ],
            },
            portalThreadId: undefined,
            threadStartMessageId: 'source-msg-id',
          });
        });

        vi.spyOn(result.current, 'createThread').mockResolvedValue({
          messageId: 'new-msg-id',
          threadId: 'new-thread-id',
        });

        vi.spyOn(result.current, 'refreshThreads').mockImplementation(async () => {
          act(() => {
            useChatStore.setState({
              portalThreadId: 'new-thread-id',
              threadMaps: { 'test-topic-id': [mockThread] },
            });
          });
        });
        vi.spyOn(result.current, 'refreshMessages').mockResolvedValue();
        vi.spyOn(result.current, 'openThreadInPortal').mockImplementation((threadId) => {
          act(() => {
            useChatStore.setState({ portalThreadId: threadId });
          });
        });
        vi.spyOn(result.current, 'internal_coreProcessMessage').mockResolvedValue();
        vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-msg-id');
        vi.spyOn(result.current, 'internal_toggleMessageLoading');

        const summaryTitleSpy = vi.spyOn(result.current, 'summaryThreadTitle').mockResolvedValue();

        await act(async () => {
          await result.current.sendThreadMessage({ message: 'test message' });
        });

        expect(summaryTitleSpy).toHaveBeenCalledWith('new-thread-id', expect.any(Array));
      });
    });

    describe('existing thread flow', () => {
      it('should append message to existing thread', async () => {
        const { result } = renderHook(() => useChatStore());

        act(() => {
          useChatStore.setState({
            portalThreadId: 'existing-thread-id',
          });
        });

        const createMessageSpy = vi
          .spyOn(result.current, 'internal_createMessage')
          .mockResolvedValue('new-msg-id');
        const coreProcessSpy = vi
          .spyOn(result.current, 'internal_coreProcessMessage')
          .mockResolvedValue();
        vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-msg-id');
        vi.spyOn(result.current, 'internal_toggleMessageLoading');

        await act(async () => {
          await result.current.sendThreadMessage({ message: 'follow-up message' });
        });

        expect(createMessageSpy).toHaveBeenCalledWith(
          expect.objectContaining({
            content: 'follow-up message',
            role: 'user',
            threadId: 'existing-thread-id',
          }),
          { tempMessageId: 'temp-msg-id' },
        );

        expect(coreProcessSpy).toHaveBeenCalledWith(
          expect.any(Array),
          'new-msg-id',
          expect.objectContaining({
            inPortalThread: true,
            threadId: 'existing-thread-id',
          }),
        );
      });

      it('should not auto-summarize title for existing threads', async () => {
        const { result } = renderHook(() => useChatStore());

        act(() => {
          useChatStore.setState({
            portalThreadId: 'existing-thread-id',
          });
        });

        vi.spyOn(result.current, 'internal_createMessage').mockResolvedValue('new-msg-id');
        vi.spyOn(result.current, 'internal_coreProcessMessage').mockResolvedValue();
        vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-msg-id');
        vi.spyOn(result.current, 'internal_toggleMessageLoading');

        const summaryTitleSpy = vi.spyOn(result.current, 'summaryThreadTitle').mockResolvedValue();

        await act(async () => {
          await result.current.sendThreadMessage({ message: 'follow-up message' });
        });

        expect(summaryTitleSpy).not.toHaveBeenCalled();
      });
    });

    describe('message processing', () => {
      it('should trigger session update', async () => {
        const { result } = renderHook(() => useChatStore());
        const triggerUpdateMock = vi.fn();

        (useSessionStore.getState as Mock).mockReturnValue({
          triggerSessionUpdate: triggerUpdateMock,
        });

        act(() => {
          useChatStore.setState({
            portalThreadId: 'existing-thread-id',
          });
        });

        vi.spyOn(result.current, 'internal_createMessage').mockResolvedValue('new-msg-id');
        vi.spyOn(result.current, 'internal_coreProcessMessage').mockResolvedValue();
        vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-msg-id');
        vi.spyOn(result.current, 'internal_toggleMessageLoading');

        await act(async () => {
          await result.current.sendThreadMessage({ message: 'test' });
        });

        expect(triggerUpdateMock).toHaveBeenCalledWith('test-session-id');
      });

      it('should pass RAG query if RAG is enabled', async () => {
        const { result } = renderHook(() => useChatStore());

        act(() => {
          useChatStore.setState({
            portalThreadId: 'existing-thread-id',
          });
        });

        vi.spyOn(result.current, 'internal_shouldUseRAG').mockReturnValue(true);
        vi.spyOn(result.current, 'internal_createMessage').mockResolvedValue('new-msg-id');
        vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-msg-id');
        vi.spyOn(result.current, 'internal_toggleMessageLoading');

        const coreProcessSpy = vi
          .spyOn(result.current, 'internal_coreProcessMessage')
          .mockResolvedValue();

        await act(async () => {
          await result.current.sendThreadMessage({ message: 'test with rag' });
        });

        expect(coreProcessSpy).toHaveBeenCalledWith(
          expect.any(Array),
          'new-msg-id',
          expect.objectContaining({
            inPortalThread: true,
            ragQuery: 'test with rag',
            threadId: 'existing-thread-id',
          }),
        );
      });
    });
  });

  describe('resendThreadMessage', () => {
    it('should resend message in thread context', async () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        useChatStore.setState({
          portalThreadId: 'thread-id',
        });
      });

      const resendSpy = vi.spyOn(result.current, 'internal_resendMessage').mockResolvedValue();

      await act(async () => {
        await result.current.resendThreadMessage('message-id');
      });

      expect(resendSpy).toHaveBeenCalledWith(
        'message-id',
        expect.objectContaining({
          inPortalThread: true,
          messages: expect.any(Array),
          threadId: 'thread-id',
        }),
      );
    });
  });

  describe('delAndResendThreadMessage', () => {
    it('should delete and resend message', async () => {
      const { result } = renderHook(() => useChatStore());

      const resendSpy = vi.spyOn(result.current, 'resendThreadMessage').mockResolvedValue();
      const deleteSpy = vi.spyOn(result.current, 'deleteMessage').mockResolvedValue();

      await act(async () => {
        await result.current.delAndResendThreadMessage('message-id');
      });

      expect(resendSpy).toHaveBeenCalledWith('message-id');
      expect(deleteSpy).toHaveBeenCalledWith('message-id');
    });
  });

  describe('internal_updateThreadTitleInSummary', () => {
    it('should dispatch thread update', () => {
      const { result } = renderHook(() => useChatStore());

      const dispatchSpy = vi.spyOn(result.current, 'internal_dispatchThread');

      act(() => {
        useChatStore.setState({
          activeTopicId: 'test-topic-id',
          threadMaps: {
            'test-topic-id': [
              {
                createdAt: new Date(),
                id: 'thread-id',
                lastActiveAt: new Date(),
                sourceMessageId: 'msg-1',
                status: ThreadStatus.Active,
                title: 'Old Title',
                topicId: 'test-topic-id',
                type: ThreadType.Continuation,
                updatedAt: new Date(),
                userId: 'user-1',
              },
            ],
          },
        });
      });

      act(() => {
        result.current.internal_updateThreadTitleInSummary('thread-id', 'New Title');
      });

      expect(dispatchSpy).toHaveBeenCalledWith(
        { id: 'thread-id', type: 'updateThread', value: { title: 'New Title' } },
        'updateThreadTitleInSummary',
      );
    });
  });

  describe('internal_updateThreadLoading', () => {
    it('should add thread id to loading list', () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        result.current.internal_updateThreadLoading('thread-id', true);
      });

      expect(result.current.threadLoadingIds).toContain('thread-id');
    });

    it('should remove thread id from loading list', () => {
      const { result } = renderHook(() => useChatStore());

      act(() => {
        useChatStore.setState({ threadLoadingIds: ['thread-id'] });
      });

      act(() => {
        result.current.internal_updateThreadLoading('thread-id', false);
      });

      expect(result.current.threadLoadingIds).not.toContain('thread-id');
    });
  });

  describe('internal_updateThread', () => {
    it('should update thread locally and on server', async () => {
      const { result } = renderHook(() => useChatStore());

      (threadService.updateThread as Mock).mockResolvedValue(undefined);

      const dispatchSpy = vi.spyOn(result.current, 'internal_dispatchThread');
      const refreshSpy = vi.spyOn(result.current, 'refreshThreads').mockResolvedValue();
      const loadingSpy = vi.spyOn(result.current, 'internal_updateThreadLoading');

      await act(async () => {
        await result.current.internal_updateThread('thread-id', { title: 'Updated Title' });
      });

      expect(dispatchSpy).toHaveBeenCalledWith({
        id: 'thread-id',
        type: 'updateThread',
        value: { title: 'Updated Title' },
      });
      expect(threadService.updateThread).toHaveBeenCalledWith('thread-id', {
        title: 'Updated Title',
      });
      expect(refreshSpy).toHaveBeenCalled();
      expect(loadingSpy).toHaveBeenCalledWith('thread-id', true);
      expect(loadingSpy).toHaveBeenCalledWith('thread-id', false);
    });
  });

  describe('internal_dispatchThread', () => {
    it('should update threadMaps with reducer result', () => {
      const { result } = renderHook(() => useChatStore());

      const mockThread: ThreadItem = {
        createdAt: new Date(),
        id: 'thread-id',
        lastActiveAt: new Date(),
        sourceMessageId: 'msg-1',
        status: ThreadStatus.Active,
        title: 'Old Title',
        topicId: 'test-topic-id',
        type: ThreadType.Continuation,
        updatedAt: new Date(),
        userId: 'user-1',
      };

      act(() => {
        useChatStore.setState({
          activeTopicId: 'test-topic-id',
          threadMaps: {
            'test-topic-id': [mockThread],
          },
        });
      });

      act(() => {
        result.current.internal_dispatchThread({
          id: 'thread-id',
          type: 'updateThread',
          value: { title: 'New Title' },
        });
      });

      const updatedThread = result.current.threadMaps['test-topic-id']?.find(
        (t) => t.id === 'thread-id',
      );
      expect(updatedThread?.title).toBe('New Title');
    });

    it('should not update if result is the same', () => {
      const { result } = renderHook(() => useChatStore());

      const mockThread: ThreadItem = {
        createdAt: new Date(),
        id: 'thread-id',
        lastActiveAt: new Date(),
        sourceMessageId: 'msg-1',
        status: ThreadStatus.Active,
        title: 'Title',
        topicId: 'test-topic-id',
        type: ThreadType.Continuation,
        updatedAt: new Date(),
        userId: 'user-1',
      };

      act(() => {
        useChatStore.setState({
          activeTopicId: 'test-topic-id',
          threadMaps: {
            'test-topic-id': [mockThread],
          },
        });
      });

      const mapsBefore = result.current.threadMaps;

      // Update with non-existent thread id - should not change anything
      act(() => {
        result.current.internal_dispatchThread({
          id: 'non-existent-thread',
          type: 'updateThread',
          value: { title: 'New Title' },
        });
      });

      // Maps should remain the same reference due to isEqual check
      expect(result.current.threadMaps).toEqual(mapsBefore);
    });
  });
});