@aws/lsp-codewhisperer 0.0.27 → 0.0.28

package/out/language-server/telemetryService.test.js

@@ -1,771 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const telemetryService_1 = require("./telemetryService");
-const ts_sinon_1 = require("ts-sinon");
-const constants_1 = require("./constants");
-const types_1 = require("./telemetry/types");
-class MockCredentialsProvider {
-    mockIamCredentials;
-    mockBearerCredentials;
-    mockConnectionMetadata;
-    hasCredentials(type) {
-        if (type === 'iam') {
-            return this.mockIamCredentials !== undefined;
-        }
-        else if (type === 'bearer') {
-            return this.mockBearerCredentials !== undefined;
-        }
-        return false;
-    }
-    getCredentials(type) {
-        if (type === 'iam') {
-            return this.mockIamCredentials;
-        }
-        else if (type === 'bearer') {
-            return this.mockBearerCredentials;
-        }
-        return undefined;
-    }
-    getConnectionMetadata() {
-        return this.mockConnectionMetadata;
-    }
-    setConnectionMetadata(metadata) {
-        this.mockConnectionMetadata = metadata;
-    }
-}
-describe('TelemetryService', () => {
-    let telemetry;
-    let clock;
-    let telemetryService;
-    let mockCredentialsProvider;
-    const mockAwsQRegion = 'mock-aws-q-region';
-    const mockAwsQEndpointUrl = 'mock-aws-q-endpoint-url';
-    const logging = {
-        log: (message) => {
-            console.log(message);
-        },
-    };
-    const mockWorkspace = {};
-    const mockSession = {
-        codewhispererSessionId: 'test-session-id',
-        responseContext: {
-            requestId: 'test-request-id',
-            codewhispererSessionId: 'test-session-id',
-        },
-        customizationArn: 'test-arn',
-        language: 'tsx',
-        suggestions: [
-            {
-                itemId: 'item-id-1',
-                content: 'line1\nline2\nline3',
-                references: [],
-            },
-        ],
-        completionSessionResult: {
-            'item-id-1': {
-                accepted: true,
-                seen: true,
-                discarded: false,
-            },
-        },
-        acceptedSuggestionId: 'item-id-1',
-        firstCompletionDisplayLatency: 100,
-        timeToFirstRecommendation: 200,
-        getAggregatedUserTriggerDecision: () => 'Accept',
-        startPosition: {
-            line: 12,
-            character: 23,
-        },
-    };
-    const mockSdkRuntimeConfigurator = Object.assign(
-    // Default callable function for v3 clients
-    (Ctor, current_config) => new Ctor({ ...current_config }), 
-    // Property for v2 clients
-    {
-        v2: (Ctor, current_config) => new Ctor({ ...current_config }),
-    });
-    beforeEach(() => {
-        clock = ts_sinon_1.default.useFakeTimers({
-            now: 1483228800000,
-        });
-        mockCredentialsProvider = new MockCredentialsProvider();
-        telemetry = {
-            emitMetric: ts_sinon_1.default.stub(),
-            onClientTelemetry: ts_sinon_1.default.stub(),
-        };
-    });
-    afterEach(() => {
-        clock.restore();
-        ts_sinon_1.default.restore();
-    });
-    it('updateUserContext updates the userContext property', () => {
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', {}, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        const mockUserContext = {
-            clientId: 'aaaabbbbccccdddd',
-            ideCategory: 'ECLIPSE',
-            ideVersion: '1.1.1',
-            operatingSystem: 'MAC',
-            product: 'INLINE',
-        };
-        telemetryService.updateUserContext(mockUserContext);
-        ts_sinon_1.default.assert.match(telemetryService.userContext, mockUserContext);
-    });
-    it('updateOptOutPreference updates the optOutPreference property', () => {
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', {}, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        const mockOptOutPreference = 'OPTIN';
-        telemetryService.updateOptOutPreference(mockOptOutPreference);
-        ts_sinon_1.default.assert.match(telemetryService.optOutPreference, mockOptOutPreference);
-    });
-    it('updateEnableTelemetryEventsToDestination updates the enableTelemetryEventsToDestination property', () => {
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', {}, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        telemetryService.updateEnableTelemetryEventsToDestination(true);
-        ts_sinon_1.default.assert.match(telemetryService.enableTelemetryEventsToDestination, true);
-    });
-    it('getSuggestionState fetches the suggestion state from CodeWhispererSession', () => {
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', {}, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        const getSuggestionState = telemetryService.getSuggestionState.bind(telemetryService);
-        let session = {
-            getAggregatedUserTriggerDecision: () => {
-                return 'Accept';
-            },
-        };
-        ts_sinon_1.default.assert.match(getSuggestionState(session), 'ACCEPT');
-        session = {
-            getAggregatedUserTriggerDecision: () => {
-                return 'Reject';
-            },
-        };
-        ts_sinon_1.default.assert.match(getSuggestionState(session), 'REJECT');
-        session = {
-            getAggregatedUserTriggerDecision: () => {
-                return 'Discard';
-            },
-        };
-        ts_sinon_1.default.assert.match(getSuggestionState(session), 'DISCARD');
-        session = {
-            getAggregatedUserTriggerDecision: () => {
-                return 'Empty';
-            },
-        };
-        ts_sinon_1.default.assert.match(getSuggestionState(session), 'EMPTY');
-        session = {
-            getAggregatedUserTriggerDecision: () => {
-                return undefined;
-            },
-        };
-        ts_sinon_1.default.assert.match(getSuggestionState(session), 'EMPTY');
-    });
-    it('should not emit user trigger decision if login is invalid (IAM)', () => {
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'iam', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        const invokeSendTelemetryEventStub = ts_sinon_1.default.stub(telemetryService, 'sendTelemetryEvent');
-        telemetryService.emitUserTriggerDecision(mockSession);
-        ts_sinon_1.default.assert.notCalled(invokeSendTelemetryEventStub);
-    });
-    it('should not emit user trigger decision if login is BuilderID, but user chose OPTOUT option', () => {
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: constants_1.BUILDER_ID_START_URL,
-            },
-        });
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        const invokeSendTelemetryEventStub = ts_sinon_1.default.stub(telemetryService, 'sendTelemetryEvent');
-        telemetryService.updateOptOutPreference('OPTOUT');
-        telemetryService.emitUserTriggerDecision(mockSession);
-        ts_sinon_1.default.assert.notCalled(invokeSendTelemetryEventStub);
-    });
-    it('should handle SSO connection type change at runtime', () => {
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        const sendTelemetryEventStub = ts_sinon_1.default
-            .stub(telemetryService, 'sendTelemetryEvent')
-            .returns(Promise.resolve());
-        telemetryService.updateOptOutPreference('OPTOUT'); // Disables telemetry for builderId startUrl
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: 'https://some-random-test-idc-directory.awsapps.com',
-            },
-        });
-        // Emitting event with IdC connection
-        telemetryService.emitUserTriggerDecision(mockSession);
-        ts_sinon_1.default.assert.calledOnce(sendTelemetryEventStub);
-        // Switch to BuilderId connection
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: constants_1.BUILDER_ID_START_URL,
-            },
-        });
-        sendTelemetryEventStub.resetHistory();
-        // Should not emit event anymore with BuilderId
-        telemetryService.emitUserTriggerDecision(mockSession);
-        ts_sinon_1.default.assert.notCalled(sendTelemetryEventStub);
-    });
-    it('should emit userTriggerDecision event to STE and to the destination', () => {
-        const expectedUserTriggerDecisionEvent = {
-            telemetryEvent: {
-                userTriggerDecisionEvent: {
-                    sessionId: 'test-session-id',
-                    requestId: 'test-request-id',
-                    customizationArn: 'test-arn',
-                    programmingLanguage: { languageName: 'typescript' },
-                    completionType: 'BLOCK',
-                    suggestionState: 'ACCEPT',
-                    recommendationLatencyMilliseconds: 100,
-                    timestamp: new Date(Date.now()),
-                    triggerToResponseLatencyMilliseconds: 200,
-                    suggestionReferenceCount: 0,
-                    generatedLine: 3,
-                    numberOfRecommendations: 1,
-                    perceivedLatencyMilliseconds: undefined,
-                    acceptedCharacterCount: 17,
-                },
-            },
-            optOutPreference: 'OPTIN',
-        };
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: 'idc-start-url',
-            },
-        });
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        telemetryService.updateEnableTelemetryEventsToDestination(true);
-        const invokeSendTelemetryEventStub = ts_sinon_1.default
-            .stub(telemetryService, 'sendTelemetryEvent')
-            .returns(Promise.resolve());
-        telemetryService.updateOptOutPreference('OPTIN');
-        telemetryService.emitUserTriggerDecision(mockSession);
-        ts_sinon_1.default.assert.calledOnceWithExactly(invokeSendTelemetryEventStub, expectedUserTriggerDecisionEvent);
-        ts_sinon_1.default.assert.calledOnceWithExactly(telemetry.emitMetric, {
-            name: 'codewhisperer_userTriggerDecision',
-            data: {
-                codewhispererSessionId: 'test-session-id',
-                codewhispererFirstRequestId: 'test-request-id',
-                credentialStartUrl: undefined,
-                codewhispererSuggestionState: 'Accept',
-                codewhispererCompletionType: 'Block',
-                codewhispererLanguage: 'tsx',
-                codewhispererTriggerType: undefined,
-                codewhispererAutomatedTriggerType: undefined,
-                codewhispererTriggerCharacter: undefined,
-                codewhispererLineNumber: 12,
-                codewhispererCursorOffset: 23,
-                codewhispererSuggestionCount: 1,
-                codewhispererClassifierResult: undefined,
-                codewhispererClassifierThreshold: undefined,
-                codewhispererTotalShownTime: 0,
-                codewhispererTypeaheadLength: 0,
-                codewhispererTimeSinceLastDocumentChange: undefined,
-                codewhispererTimeSinceLastUserDecision: undefined,
-                codewhispererTimeToFirstRecommendation: 200,
-                codewhispererPreviousSuggestionState: undefined,
-                codewhispererSupplementalContextTimeout: undefined,
-                codewhispererSupplementalContextIsUtg: undefined,
-                codewhispererSupplementalContextLength: undefined,
-                codewhispererCustomizationArn: 'test-arn',
-            },
-        });
-    });
-    it('should not emit userTriggerDecision event to destination when enableTelemetryEventsToDestination is disabled', () => {
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: constants_1.BUILDER_ID_START_URL,
-            },
-        });
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        telemetryService.updateEnableTelemetryEventsToDestination(false);
-        telemetryService.updateOptOutPreference('OPTOUT');
-        telemetryService.emitUserTriggerDecision(mockSession);
-        ts_sinon_1.default.assert.neverCalledWithMatch(telemetry.emitMetric, {
-            name: 'codewhisperer_userTriggerDecision',
-        });
-    });
-    describe('Chat interact with message', () => {
-        let telemetryService;
-        let mockCredentialsProvider;
-        let invokeSendTelemetryEventStub;
-        beforeEach(() => {
-            mockCredentialsProvider = new MockCredentialsProvider();
-            mockCredentialsProvider.setConnectionMetadata({
-                sso: {
-                    startUrl: 'idc-start-url',
-                },
-            });
-            telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-            invokeSendTelemetryEventStub = ts_sinon_1.default
-                .stub(telemetryService, 'sendTelemetryEvent')
-                .returns(Promise.resolve());
-        });
-        afterEach(() => {
-            ts_sinon_1.default.restore();
-        });
-        it('should send InteractWithMessage event with correct parameters and emit metric to destination', () => {
-            const metric = {
-                cwsprChatMessageId: 'message123',
-                codewhispererCustomizationArn: 'arn:123',
-                cwsprChatInteractionType: types_1.ChatInteractionType.InsertAtCursor,
-                cwsprChatInteractionTarget: 'CODE',
-                cwsprChatAcceptedCharactersLength: 100,
-            };
-            const conversationId = 'conv123';
-            const acceptedLineCount = 5;
-            telemetryService.updateEnableTelemetryEventsToDestination(true);
-            telemetryService.emitChatInteractWithMessage(metric, {
-                conversationId,
-                acceptedLineCount,
-            });
-            const expectedEvent = {
-                telemetryEvent: {
-                    chatInteractWithMessageEvent: {
-                        conversationId: conversationId,
-                        messageId: metric.cwsprChatMessageId,
-                        customizationArn: 'arn:123',
-                        interactionType: 'INSERT_AT_CURSOR',
-                        interactionTarget: metric.cwsprChatInteractionTarget,
-                        acceptedCharacterCount: metric.cwsprChatAcceptedCharactersLength,
-                        acceptedLineCount: acceptedLineCount,
-                        acceptedSnippetHasReference: false,
-                        hasProjectLevelContext: false,
-                    },
-                },
-            };
-            ts_sinon_1.default.assert.calledOnceWithExactly(invokeSendTelemetryEventStub, expectedEvent);
-            ts_sinon_1.default.assert.calledOnceWithExactly(telemetry.emitMetric, {
-                name: 'amazonq_interactWithMessage',
-                data: {
-                    cwsprChatMessageId: 'message123',
-                    codewhispererCustomizationArn: 'arn:123',
-                    cwsprChatInteractionType: 'insertAtCursor',
-                    cwsprChatInteractionTarget: 'CODE',
-                    cwsprChatAcceptedCharactersLength: 100,
-                    cwsprChatConversationId: 'conv123',
-                    credentialStartUrl: 'idc-start-url',
-                },
-            });
-        });
-        it('should not send InteractWithMessage event to destination when enableTelemetryEventsToDestination flag is disabled', () => {
-            const metric = {
-                cwsprChatMessageId: 'message123',
-                codewhispererCustomizationArn: 'arn:123',
-                cwsprChatInteractionType: types_1.ChatInteractionType.InsertAtCursor,
-                cwsprChatInteractionTarget: 'CODE',
-                cwsprChatAcceptedCharactersLength: 100,
-            };
-            const conversationId = 'conv123';
-            const acceptedLineCount = 5;
-            mockCredentialsProvider.setConnectionMetadata({
-                sso: {
-                    startUrl: constants_1.BUILDER_ID_START_URL,
-                },
-            });
-            telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', {}, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-            telemetryService.updateEnableTelemetryEventsToDestination(false);
-            telemetryService.updateOptOutPreference('OPTOUT');
-            telemetryService.emitChatInteractWithMessage(metric, {
-                conversationId,
-                acceptedLineCount,
-            });
-            ts_sinon_1.default.assert.neverCalledWithMatch(telemetry.emitMetric, {
-                name: 'amazonq_interactWithMessage',
-            });
-        });
-        it('should not send InteractWithMessage when conversationId is undefined', () => {
-            const metric = {
-                cwsprChatMessageId: 'message123',
-                codewhispererCustomizationArn: 'arn:123',
-                cwsprChatInteractionType: types_1.ChatInteractionType.InsertAtCursor,
-                cwsprChatInteractionTarget: 'CODE',
-                cwsprChatAcceptedCharactersLength: 100,
-            };
-            telemetryService.emitChatInteractWithMessage(metric, {
-                acceptedLineCount: 5,
-            });
-            ts_sinon_1.default.assert.notCalled(invokeSendTelemetryEventStub);
-        });
-        it('should not send InteractWithMessage when credentialsType is IAM', () => {
-            telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'iam', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-            invokeSendTelemetryEventStub = ts_sinon_1.default.stub(telemetryService, 'sendTelemetryEvent');
-            const metric = {
-                cwsprChatMessageId: 'message123',
-                codewhispererCustomizationArn: 'arn:123',
-                cwsprChatInteractionType: types_1.ChatInteractionType.InsertAtCursor,
-                cwsprChatInteractionTarget: 'CODE',
-                cwsprChatAcceptedCharactersLength: 100,
-            };
-            telemetryService.emitChatInteractWithMessage(metric, {
-                conversationId: 'conv123',
-                acceptedLineCount: 5,
-            });
-            ts_sinon_1.default.assert.notCalled(invokeSendTelemetryEventStub);
-        });
-        it('should not send InteractWithMessage when login is BuilderID, but user chose OPTOUT option', () => {
-            mockCredentialsProvider.setConnectionMetadata({
-                sso: {
-                    startUrl: constants_1.BUILDER_ID_START_URL,
-                },
-            });
-            telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-            invokeSendTelemetryEventStub = ts_sinon_1.default.stub(telemetryService, 'sendTelemetryEvent');
-            telemetryService.updateOptOutPreference('OPTOUT');
-            const metric = {
-                cwsprChatMessageId: 'message123',
-                codewhispererCustomizationArn: 'arn:123',
-                cwsprChatInteractionType: types_1.ChatInteractionType.InsertAtCursor,
-                cwsprChatInteractionTarget: 'CODE',
-                cwsprChatAcceptedCharactersLength: 100,
-            };
-            telemetryService.emitChatInteractWithMessage(metric, {
-                conversationId: 'conv123',
-                acceptedLineCount: 5,
-            });
-            ts_sinon_1.default.assert.notCalled(invokeSendTelemetryEventStub);
-        });
-        it('should send InteractWithMessage but with optional acceptedLineCount parameter', () => {
-            const metric = {
-                cwsprChatMessageId: 'message123',
-                codewhispererCustomizationArn: 'arn:123',
-                cwsprChatInteractionType: types_1.ChatInteractionType.InsertAtCursor,
-                cwsprChatInteractionTarget: 'CODE',
-                cwsprChatAcceptedCharactersLength: 100,
-            };
-            const conversationId = 'conv123';
-            telemetryService.emitChatInteractWithMessage(metric, {
-                conversationId,
-                acceptedLineCount: undefined,
-            });
-            ts_sinon_1.default.assert.calledWithExactly(invokeSendTelemetryEventStub, {
-                telemetryEvent: {
-                    chatInteractWithMessageEvent: {
-                        conversationId: 'conv123',
-                        messageId: 'message123',
-                        customizationArn: 'arn:123',
-                        interactionType: 'INSERT_AT_CURSOR',
-                        interactionTarget: 'CODE',
-                        acceptedCharacterCount: 100,
-                        acceptedLineCount: undefined, // Should be undefined
-                        acceptedSnippetHasReference: false,
-                        hasProjectLevelContext: false,
-                    },
-                },
-            });
-        });
-    });
-    it('should emit CodeCoverageEvent event to STE and to the destination', () => {
-        const expectedEvent = {
-            telemetryEvent: {
-                codeCoverageEvent: {
-                    customizationArn: 'test-arn',
-                    programmingLanguage: { languageName: 'typescript' },
-                    acceptedCharacterCount: 123,
-                    totalCharacterCount: 456,
-                    timestamp: new Date(Date.now()),
-                },
-            },
-            optOutPreference: 'OPTIN',
-        };
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: 'idc-start-url',
-            },
-        });
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        const invokeSendTelemetryEventStub = ts_sinon_1.default
-            .stub(telemetryService, 'sendTelemetryEvent')
-            .returns(Promise.resolve());
-        telemetryService.updateOptOutPreference('OPTIN');
-        telemetryService.updateEnableTelemetryEventsToDestination(true);
-        telemetryService.emitCodeCoverageEvent({
-            languageId: 'typescript',
-            customizationArn: 'test-arn',
-            acceptedCharacterCount: 123,
-            totalCharacterCount: 456,
-        }, {
-            percentage: 50,
-            successCount: 1,
-        });
-        ts_sinon_1.default.assert.calledOnceWithExactly(invokeSendTelemetryEventStub, expectedEvent);
-        ts_sinon_1.default.assert.calledOnceWithExactly(telemetry.emitMetric, {
-            name: 'codewhisperer_codePercentage',
-            data: {
-                codewhispererTotalTokens: 456,
-                codewhispererLanguage: 'typescript',
-                codewhispererSuggestedTokens: 123,
-                codewhispererPercentage: 50,
-                successCount: 1,
-            },
-        });
-    });
-    it('should not emit CodeCoverageEvent event to destination when enableTelemetryEventsToDestination flag is disabled', () => {
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: constants_1.BUILDER_ID_START_URL,
-            },
-        });
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        telemetryService.updateOptOutPreference('OPTOUT');
-        telemetryService.updateEnableTelemetryEventsToDestination(false);
-        telemetryService.emitCodeCoverageEvent({
-            languageId: 'typescript',
-            customizationArn: 'test-arn',
-            acceptedCharacterCount: 123,
-            totalCharacterCount: 456,
-        }, {
-            percentage: 50,
-            successCount: 1,
-        });
-        ts_sinon_1.default.assert.neverCalledWithMatch(telemetry.emitMetric, {
-            name: 'codewhisperer_codePercentage',
-        });
-    });
-    it('should emit userModificationEvent event', () => {
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: 'idc-start-url',
-            },
-        });
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', {}, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        const invokeSendTelemetryEventStub = ts_sinon_1.default
-            .stub(telemetryService, 'sendTelemetryEvent')
-            .returns(Promise.resolve());
-        telemetryService.updateOptOutPreference('OPTIN');
-        telemetryService.emitUserModificationEvent({
-            sessionId: 'test-session-id',
-            requestId: 'test-request-id',
-            languageId: 'typescript',
-            customizationArn: 'test-arn',
-            timestamp: new Date(),
-            modificationPercentage: 0.2,
-            acceptedCharacterCount: 100,
-            unmodifiedAcceptedCharacterCount: 80,
-        });
-        const expectedEvent = {
-            telemetryEvent: {
-                userModificationEvent: {
-                    sessionId: 'test-session-id',
-                    requestId: 'test-request-id',
-                    programmingLanguage: {
-                        languageName: 'typescript',
-                    },
-                    modificationPercentage: 0.2,
-                    customizationArn: 'test-arn',
-                    timestamp: new Date(),
-                    acceptedCharacterCount: 100,
-                    unmodifiedAcceptedCharacterCount: 80,
-                },
-            },
-            optOutPreference: 'OPTIN',
-        };
-        ts_sinon_1.default.assert.calledOnceWithExactly(invokeSendTelemetryEventStub, expectedEvent);
-    });
-    it('should emit chatUserModificationEvent event including emitting event to destination', () => {
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: 'idc-start-url',
-            },
-        });
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        telemetryService.updateEnableTelemetryEventsToDestination(true);
-        const invokeSendTelemetryEventStub = ts_sinon_1.default
-            .stub(telemetryService, 'sendTelemetryEvent')
-            .returns(Promise.resolve());
-        telemetryService.updateOptOutPreference('OPTIN');
-        telemetryService.emitChatUserModificationEvent({
-            conversationId: 'test-conversation-id',
-            messageId: 'test-message-id',
-            customizationArn: 'test-arn',
-            modificationPercentage: 0.2,
-        });
-        const expectedEvent = {
-            telemetryEvent: {
-                chatUserModificationEvent: {
-                    conversationId: 'test-conversation-id',
-                    messageId: 'test-message-id',
-                    customizationArn: 'test-arn',
-                    modificationPercentage: 0.2,
-                },
-            },
-            optOutPreference: 'OPTIN',
-        };
-        ts_sinon_1.default.assert.calledOnceWithExactly(invokeSendTelemetryEventStub, expectedEvent);
-        ts_sinon_1.default.assert.calledOnceWithExactly(telemetry.emitMetric, {
-            name: 'amazonq_modifyCode',
-            data: {
-                cwsprChatConversationId: 'test-conversation-id',
-                cwsprChatMessageId: 'test-message-id',
-                cwsprChatModificationPercentage: 0.2,
-                codewhispererCustomizationArn: 'test-arn',
-                credentialStartUrl: 'idc-start-url',
-            },
-        });
-    });
-    it('should not emit chatUserModificationEvent event to destination when enableTelemetryEventsToDestination flag is disabled', () => {
-        mockCredentialsProvider.setConnectionMetadata({
-            sso: {
-                startUrl: constants_1.BUILDER_ID_START_URL,
-            },
-        });
-        telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-        telemetryService.updateEnableTelemetryEventsToDestination(false);
-        telemetryService.updateOptOutPreference('OPTOUT');
-        telemetryService.emitChatUserModificationEvent({
-            conversationId: 'test-conversation-id',
-            messageId: 'test-message-id',
-            customizationArn: 'test-arn',
-            modificationPercentage: 0.2,
-        });
-        ts_sinon_1.default.assert.neverCalledWithMatch(telemetry.emitMetric, {
-            name: 'amazonq_modifyCode',
-        });
-    });
-    describe('Chat add message', () => {
-        let telemetryService;
-        let mockCredentialsProvider;
-        let invokeSendTelemetryEventStub;
-        beforeEach(() => {
-            mockCredentialsProvider = new MockCredentialsProvider();
-            mockCredentialsProvider.setConnectionMetadata({
-                sso: {
-                    startUrl: 'idc-start-url',
-                },
-            });
-            telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-            invokeSendTelemetryEventStub = ts_sinon_1.default
-                .stub(telemetryService, 'sendTelemetryEvent')
-                .returns(Promise.resolve());
-        });
-        afterEach(() => {
-            ts_sinon_1.default.restore();
-        });
-        it('should send ChatAddMessage event with correct parameters and emit metric to destination', () => {
-            telemetryService.updateEnableTelemetryEventsToDestination(true);
-            telemetryService.emitChatAddMessage({
-                conversationId: 'conv123',
-                messageId: 'message123',
-                customizationArn: 'cust-123',
-                programmingLanguage: 'jsx',
-                userIntent: 'SUGGEST_ALTERNATE_IMPLEMENTATION',
-                hasCodeSnippet: false,
-                timeToFirstChunkMilliseconds: 100,
-                activeEditorTotalCharacters: 250,
-                fullResponselatency: 400,
-                requestLength: 100,
-                responseLength: 3000,
-                numberOfCodeBlocks: 0,
-            }, {});
-            const expectedEvent = {
-                telemetryEvent: {
-                    chatAddMessageEvent: {
-                        conversationId: 'conv123',
-                        messageId: 'message123',
-                        customizationArn: 'cust-123',
-                        userIntent: 'SUGGEST_ALTERNATE_IMPLEMENTATION',
-                        hasCodeSnippet: false,
-                        programmingLanguage: {
-                            languageName: 'javascript',
-                        },
-                        activeEditorTotalCharacters: 250,
-                        timeToFirstChunkMilliseconds: 100,
-                        timeBetweenChunks: undefined,
-                        fullResponselatency: 400,
-                        requestLength: 100,
-                        responseLength: 3000,
-                        numberOfCodeBlocks: 0,
-                        hasProjectLevelContext: false,
-                    },
-                },
-            };
-            ts_sinon_1.default.assert.calledOnceWithExactly(invokeSendTelemetryEventStub, expectedEvent);
-            ts_sinon_1.default.assert.calledOnceWithExactly(telemetry.emitMetric, {
-                name: 'amazonq_addMessage',
-                data: {
-                    credentialStartUrl: 'idc-start-url',
-                    cwsprChatConversationId: 'conv123',
-                    cwsprChatHasCodeSnippet: false,
-                    cwsprChatTriggerInteraction: undefined,
-                    cwsprChatMessageId: 'message123',
-                    cwsprChatUserIntent: 'SUGGEST_ALTERNATE_IMPLEMENTATION',
-                    cwsprChatProgrammingLanguage: 'jsx',
-                    cwsprChatResponseCodeSnippetCount: 0,
-                    cwsprChatResponseCode: undefined,
-                    cwsprChatSourceLinkCount: undefined,
-                    cwsprChatReferencesCount: undefined,
-                    cwsprChatFollowUpCount: undefined,
-                    cwsprTimeToFirstChunk: 100,
-                    cwsprChatFullResponseLatency: 400,
-                    cwsprChatTimeBetweenChunks: undefined,
-                    cwsprChatRequestLength: 100,
-                    cwsprChatResponseLength: 3000,
-                    cwsprChatConversationType: undefined,
-                    cwsprChatActiveEditorTotalCharacters: 250,
-                    cwsprChatActiveEditorImportCount: undefined,
-                    codewhispererCustomizationArn: 'cust-123',
-                },
-            });
-        });
-        it('should not send ChatAddMessage event to destination when enableTelemetryEventsToDestination flag is disabled', () => {
-            mockCredentialsProvider.setConnectionMetadata({
-                sso: {
-                    startUrl: constants_1.BUILDER_ID_START_URL,
-                },
-            });
-            telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', {}, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-            telemetryService.updateOptOutPreference('OPTOUT');
-            telemetryService.updateEnableTelemetryEventsToDestination(false);
-            telemetryService.emitChatAddMessage({
-                conversationId: 'conv123',
-                messageId: 'message123',
-                customizationArn: 'cust-123',
-                programmingLanguage: 'jsx',
-                userIntent: 'SUGGEST_ALTERNATE_IMPLEMENTATION',
-                hasCodeSnippet: false,
-                timeToFirstChunkMilliseconds: 100,
-                activeEditorTotalCharacters: 250,
-                fullResponselatency: 400,
-                requestLength: 100,
-                responseLength: 3000,
-                numberOfCodeBlocks: 0,
-            }, {});
-            ts_sinon_1.default.assert.neverCalledWithMatch(telemetry.emitMetric, {
-                name: 'amazonq_addMessage',
-            });
-        });
-        it('should not send ChatAddMessage when conversationId is undefined', () => {
-            telemetryService.emitChatAddMessage({
-                messageId: 'message123',
-                customizationArn: 'cust-123',
-            }, {});
-            ts_sinon_1.default.assert.notCalled(invokeSendTelemetryEventStub);
-        });
-        it('should not send ChatAddMessage when messageId is undefined', () => {
-            telemetryService.emitChatAddMessage({
-                conversationId: 'conv123',
-                customizationArn: 'cust-123',
-            }, {});
-            ts_sinon_1.default.assert.notCalled(invokeSendTelemetryEventStub);
-        });
-        it('should not send ChatAddMessage when credentialsType is IAM', () => {
-            telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'iam', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-            invokeSendTelemetryEventStub = ts_sinon_1.default.stub(telemetryService, 'sendTelemetryEvent');
-            telemetryService.emitChatAddMessage({
-                conversationId: 'conv123',
-                messageId: 'message123',
-                customizationArn: 'cust-123',
-            }, {});
-            ts_sinon_1.default.assert.notCalled(invokeSendTelemetryEventStub);
-        });
-        it('should not send ChatAddMessage when login is BuilderID, but user chose OPTOUT option', () => {
-            mockCredentialsProvider.setConnectionMetadata({
-                sso: {
-                    startUrl: constants_1.BUILDER_ID_START_URL,
-                },
-            });
-            telemetryService = new telemetryService_1.TelemetryService(mockCredentialsProvider, 'bearer', telemetry, logging, mockWorkspace, mockAwsQRegion, mockAwsQEndpointUrl, mockSdkRuntimeConfigurator);
-            invokeSendTelemetryEventStub = ts_sinon_1.default.stub(telemetryService, 'sendTelemetryEvent');
-            telemetryService.updateOptOutPreference('OPTOUT');
-            telemetryService.emitChatAddMessage({
-                conversationId: 'conv123',
-                messageId: 'message123',
-                customizationArn: 'cust-123',
-            }, {});
-            ts_sinon_1.default.assert.notCalled(invokeSendTelemetryEventStub);
-        });
-    });
-});
-//# sourceMappingURL=telemetryService.test.js.map