@lobehub/chat 1.90.3 → 1.90.4

package/src/server/routers/tools/search.test.ts

@@ -104,8 +104,7 @@ describe('searchRouter', () => {
         query: 'test query',
       });
 
-      expect(result).toEqual({
-        costTime: 0,
+      expect(result).toMatchObject({
         query: 'test query',
         results: [
           {
@@ -139,8 +138,7 @@ describe('searchRouter', () => {
         query: 'test query',
       });
 
-      expect(result).toEqual({
-        costTime: 0,
+      expect(result).toMatchObject({
         query: 'test query',
         results: [
           {

package/src/store/chat/slices/aiChat/actions/__tests__/generateAIChat.test.ts

@@ -671,14 +671,16 @@ describe('chatMessage actions', () => {
       (chatService.createAssistantMessage as Mock).mockResolvedValue(aiResponse);
       const spy = vi.spyOn(chatService, 'createAssistantMessageStream');
       // 模拟消息创建
-      (messageService.createMessage as Mock).mockResolvedValue('assistant-message-id');
+      const createMessageSpyOn = vi
+        .spyOn(messageService, 'createMessage')
+        .mockResolvedValue('assistant-message-id');
 
       await act(async () => {
         await result.current.internal_coreProcessMessage(messages, userMessage.id);
       });
 
       // 验证是否创建了代表 AI 响应的消息
-      expect(messageService.createMessage).toHaveBeenCalledWith(
+      expect(createMessageSpyOn).toHaveBeenCalledWith(
         expect.objectContaining({
           role: 'assistant',
           content: LOADING_FLAT,

package/src/store/chat/slices/message/action.test.ts

@@ -369,11 +369,12 @@ describe('chatMessage actions', () => {
       const messageId = 'message-id';
       const newContent = 'Updated content';
 
+      const spy = vi.spyOn(messageService, 'updateMessage');
       await act(async () => {
         await result.current.internal_updateMessageContent(messageId, newContent);
       });
 
-      expect(messageService.updateMessage).toHaveBeenCalledWith(messageId, { content: newContent });
+      expect(spy).toHaveBeenCalledWith(messageId, { content: newContent });
     });
 
     it('should dispatch message update action', async () => {

package/src/store/chat/slices/topic/action.test.ts

@@ -264,7 +264,8 @@ describe('topic action', () => {
       const topicId = 'topic-id';
       const newTitle = 'Updated Topic Title';
       // Mock the topicService.updateTitle to resolve immediately
-      (topicService.updateTopic as Mock).mockResolvedValue(undefined);
+
+      const spyOn = vi.spyOn(topicService, 'updateTopic');
 
       const { result } = renderHook(() => useChatStore());
 
@@ -276,7 +277,7 @@ describe('topic action', () => {
       });
 
       // Verify that the topicService.updateTitle was called with correct parameters
-      expect(topicService.updateTopic).toHaveBeenCalledWith(topicId, {
+      expect(spyOn).toHaveBeenCalledWith(topicId, {
         title: 'Updated Topic Title',
       });