diff options
Diffstat (limited to 'packages/core/src/telemetry/loggers.test.ts')
| -rw-r--r-- | packages/core/src/telemetry/loggers.test.ts | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/packages/core/src/telemetry/loggers.test.ts b/packages/core/src/telemetry/loggers.test.ts index 13617946..7a24bcca 100644 --- a/packages/core/src/telemetry/loggers.test.ts +++ b/packages/core/src/telemetry/loggers.test.ts @@ -23,6 +23,7 @@ import { EVENT_CLI_CONFIG, EVENT_TOOL_CALL, EVENT_USER_PROMPT, + EVENT_FLASH_FALLBACK, } from './constants.js'; import { logApiRequest, @@ -30,6 +31,7 @@ import { logCliConfiguration, logUserPrompt, logToolCall, + logFlashFallback, } from './loggers.js'; import { ApiRequestEvent, @@ -38,6 +40,7 @@ import { ToolCallDecision, ToolCallEvent, UserPromptEvent, + FlashFallbackEvent, } from './types.js'; import * as metrics from './metrics.js'; import * as sdk from './sdk.js'; @@ -350,6 +353,29 @@ describe('loggers', () => { }); }); + describe('logFlashFallback', () => { + const mockConfig = { + getSessionId: () => 'test-session-id', + getUsageStatisticsEnabled: () => true, + } as unknown as Config; + + it('should log flash fallback event', () => { + const event = new FlashFallbackEvent(AuthType.USE_VERTEX_AI); + + logFlashFallback(mockConfig, event); + + expect(mockLogger.emit).toHaveBeenCalledWith({ + body: 'Switching to flash as Fallback.', + attributes: { + 'session.id': 'test-session-id', + 'event.name': EVENT_FLASH_FALLBACK, + 'event.timestamp': '2025-01-01T00:00:00.000Z', + auth_type: 'vertex-ai', + }, + }); + }); + }); + describe('logToolCall', () => { const cfg1 = { getSessionId: () => 'test-session-id', |
