feat(sessions): Integrate chat recording into GeminiChat (#6721)

packages/core/src/services/chatRecordingService.test.ts

@@ -19,7 +19,14 @@ import { getProjectHash } from '../utils/paths.js';
 
 vi.mock('node:fs');
 vi.mock('node:path');
-vi.mock('node:crypto');
+vi.mock('node:crypto', () => ({
+  randomUUID: vi.fn(),
+  createHash: vi.fn(() => ({
+    update: vi.fn(() => ({
+      digest: vi.fn(() => 'mocked-hash'),
+    })),
+  })),
+}));
 vi.mock('../utils/paths.js');
 
 describe('ChatRecordingService', () => {
@@ -40,6 +47,13 @@ describe('ChatRecordingService', () => {
       },
       getModel: vi.fn().mockReturnValue('gemini-pro'),
       getDebugMode: vi.fn().mockReturnValue(false),
+      getToolRegistry: vi.fn().mockReturnValue({
+        getTool: vi.fn().mockReturnValue({
+          displayName: 'Test Tool',
+          description: 'A test tool',
+          isOutputMarkdown: false,
+        }),
+      }),
     } as unknown as Config;
 
     vi.mocked(getProjectHash).mockReturnValue('test-project-hash');
@@ -124,7 +138,7 @@ describe('ChatRecordingService', () => {
       expect(conversation.messages[0].type).toBe('user');
     });
 
-    it('should append to the last message if append is true and types match', () => {
+    it('should create separate messages when recording multiple messages', () => {
       const writeFileSyncSpy = vi
         .spyOn(fs, 'writeFileSync')
         .mockImplementation(() => undefined);
@@ -146,8 +160,7 @@ describe('ChatRecordingService', () => {
 
       chatRecordingService.recordMessage({
         type: 'user',
-        content: ' World',
-        append: true,
+        content: 'World',
       });
 
       expect(mkdirSyncSpy).toHaveBeenCalled();
@@ -155,8 +168,9 @@ describe('ChatRecordingService', () => {
       const conversation = JSON.parse(
         writeFileSyncSpy.mock.calls[0][1] as string,
       ) as ConversationRecord;
-      expect(conversation.messages).toHaveLength(1);
-      expect(conversation.messages[0].content).toBe('Hello World');
+      expect(conversation.messages).toHaveLength(2);
+      expect(conversation.messages[0].content).toBe('Hello');
+      expect(conversation.messages[1].content).toBe('World');
     });
   });
 
@@ -204,10 +218,10 @@ describe('ChatRecordingService', () => {
       );
 
       chatRecordingService.recordMessageTokens({
-        input: 1,
-        output: 2,
-        total: 3,
-        cached: 0,
+        promptTokenCount: 1,
+        candidatesTokenCount: 2,
+        totalTokenCount: 3,
+        cachedContentTokenCount: 0,
       });
 
       expect(mkdirSyncSpy).toHaveBeenCalled();
@@ -217,7 +231,14 @@ describe('ChatRecordingService', () => {
       ) as ConversationRecord;
       expect(conversation.messages[0]).toEqual({
         ...initialConversation.messages[0],
-        tokens: { input: 1, output: 2, total: 3, cached: 0 },
+        tokens: {
+          input: 1,
+          output: 2,
+          total: 3,
+          cached: 0,
+          thoughts: 0,
+          tool: 0,
+        },
       });
     });
 
@@ -240,10 +261,10 @@ describe('ChatRecordingService', () => {
       );
 
       chatRecordingService.recordMessageTokens({
-        input: 2,
-        output: 2,
-        total: 4,
-        cached: 0,
+        promptTokenCount: 2,
+        candidatesTokenCount: 2,
+        totalTokenCount: 4,
+        cachedContentTokenCount: 0,
       });
 
       // @ts-expect-error private property
@@ -252,6 +273,8 @@ describe('ChatRecordingService', () => {
         output: 2,
         total: 4,
         cached: 0,
+        thoughts: 0,
+        tool: 0,
       });
     });
   });
@@ -297,7 +320,14 @@ describe('ChatRecordingService', () => {
       ) as ConversationRecord;
       expect(conversation.messages[0]).toEqual({
         ...initialConversation.messages[0],
-        toolCalls: [toolCall],
+        toolCalls: [
+          {
+            ...toolCall,
+            displayName: 'Test Tool',
+            description: 'A test tool',
+            renderOutputAsMarkdown: false,
+          },
+        ],
       });
     });
 
@@ -343,7 +373,14 @@ describe('ChatRecordingService', () => {
         type: 'gemini',
         thoughts: [],
         content: '',
-        toolCalls: [toolCall],
+        toolCalls: [
+          {
+            ...toolCall,
+            displayName: 'Test Tool',
+            description: 'A test tool',
+            renderOutputAsMarkdown: false,
+          },
+        ],
       });
     });
   });