1 files changed, 50 insertions, 0 deletions
diff --git a/packages/core/src/telemetry/loggers.test.ts b/packages/core/src/telemetry/loggers.test.ts
index 2d7835bf..5b922333 100644
--- a/packages/core/src/telemetry/loggers.test.ts
+++ b/packages/core/src/telemetry/loggers.test.ts
@@ -43,15 +43,22 @@ import * as metrics from './metrics.js';
 import * as sdk from './sdk.js';
 import { vi, describe, beforeEach, it, expect } from 'vitest';
 import { GenerateContentResponseUsageMetadata } from '@google/genai';
+import * as uiTelemetry from './uiTelemetry.js';
 
 describe('loggers', () => {
   const mockLogger = {
     emit: vi.fn(),
   };
+  const mockUiEvent = {
+    addEvent: vi.fn(),
+  };
 
   beforeEach(() => {
     vi.spyOn(sdk, 'isTelemetrySdkInitialized').mockReturnValue(true);
     vi.spyOn(logs, 'getLogger').mockReturnValue(mockLogger);
+    vi.spyOn(uiTelemetry.uiTelemetryService, 'addEvent').mockImplementation(
+      mockUiEvent.addEvent,
+    );
     vi.useFakeTimers();
     vi.setSystemTime(new Date('2025-01-01T00:00:00.000Z'));
   });
@@ -215,6 +222,7 @@ describe('loggers', () => {
           cached_content_token_count: 10,
           thoughts_token_count: 5,
           tool_token_count: 2,
+          total_token_count: 0,
           response_text: 'test-response',
         },
       });
@@ -233,6 +241,12 @@ describe('loggers', () => {
         50,
         'output',
       );
+
+      expect(mockUiEvent.addEvent).toHaveBeenCalledWith({
+        ...event,
+        'event.name': EVENT_API_RESPONSE,
+        'event.timestamp': '2025-01-01T00:00:00.000Z',
+      });
     });
 
     it('should log an API response with an error', () => {
@@ -263,6 +277,12 @@ describe('loggers', () => {
           'error.message': 'test-error',
         },
       });
+
+      expect(mockUiEvent.addEvent).toHaveBeenCalledWith({
+        ...event,
+        'event.name': EVENT_API_RESPONSE,
+        'event.timestamp': '2025-01-01T00:00:00.000Z',
+      });
     });
   });
 
@@ -417,6 +437,12 @@ describe('loggers', () => {
         true,
         ToolCallDecision.ACCEPT,
       );
+
+      expect(mockUiEvent.addEvent).toHaveBeenCalledWith({
+        ...event,
+        'event.name': EVENT_TOOL_CALL,
+        'event.timestamp': '2025-01-01T00:00:00.000Z',
+      });
     });
     it('should log a tool call with a reject decision', () => {
       const call: ErroredToolCall = {
@@ -471,6 +497,12 @@ describe('loggers', () => {
         false,
         ToolCallDecision.REJECT,
       );
+
+      expect(mockUiEvent.addEvent).toHaveBeenCalledWith({
+        ...event,
+        'event.name': EVENT_TOOL_CALL,
+        'event.timestamp': '2025-01-01T00:00:00.000Z',
+      });
     });
 
     it('should log a tool call with a modify decision', () => {
@@ -527,6 +559,12 @@ describe('loggers', () => {
         true,
         ToolCallDecision.MODIFY,
       );
+
+      expect(mockUiEvent.addEvent).toHaveBeenCalledWith({
+        ...event,
+        'event.name': EVENT_TOOL_CALL,
+        'event.timestamp': '2025-01-01T00:00:00.000Z',
+      });
     });
 
     it('should log a tool call without a decision', () => {
@@ -581,6 +619,12 @@ describe('loggers', () => {
         true,
         undefined,
       );
+
+      expect(mockUiEvent.addEvent).toHaveBeenCalledWith({
+        ...event,
+        'event.name': EVENT_TOOL_CALL,
+        'event.timestamp': '2025-01-01T00:00:00.000Z',
+      });
     });
 
     it('should log a failed tool call with an error', () => {
@@ -641,6 +685,12 @@ describe('loggers', () => {
         false,
         undefined,
       );
+
+      expect(mockUiEvent.addEvent).toHaveBeenCalledWith({
+        ...event,
+        'event.name': EVENT_TOOL_CALL,
+        'event.timestamp': '2025-01-01T00:00:00.000Z',
+      });
     });
   });
 });