fix test failure

Author: Will-ShaoHua

packages/amazonq/test/unit/codewhisperer/commands/onAcceptance.test.ts

@@ -13,20 +13,20 @@ import {
     CodeWhispererTracker,
     RecommendationHandler,
     AuthUtil,
+    CodeWhispererSession,
 } from 'aws-core-vscode/codewhisperer'
 import { resetCodeWhispererGlobalVariables, createMockTextEditor } from 'aws-core-vscode/test'
 import { assertTelemetryCurried } from 'aws-core-vscode/test'
 
 describe('onAcceptance', function () {
+    let session: CodeWhispererSession
     describe('onAcceptance', function () {
         beforeEach(async function () {
-            const session = CodeWhispererSessionState.instance.getSession()
+            session = CodeWhispererSessionState.instance.getSession()
             await resetCodeWhispererGlobalVariables()
-            session.reset()
         })
 
         afterEach(function () {
-            const session = CodeWhispererSessionState.instance.getSession()
             sinon.restore()
             session.reset()
         })
@@ -72,7 +72,6 @@ describe('onAcceptance', function () {
         })
 
         it('Should report telemetry that records this user decision event', async function () {
-            const session = CodeWhispererSessionState.instance.getSession()
             const testStartUrl = 'testStartUrl'
             sinon.stub(AuthUtil.instance, 'startUrl').value(testStartUrl)
             const mockEditor = createMockTextEditor()

packages/amazonq/test/unit/codewhisperer/commands/onInlineAcceptance.test.ts

@@ -13,20 +13,20 @@ import {
     RecommendationHandler,
     AuthUtil,
     CodeWhispererSessionState,
+    CodeWhispererSession,
 } from 'aws-core-vscode/codewhisperer'
 import { globals } from 'aws-core-vscode/shared'
 import { extensionVersion } from 'aws-core-vscode/shared'
 
 describe('onInlineAcceptance', function () {
+    let session: CodeWhispererSession
     describe('onInlineAcceptance', function () {
         beforeEach(async function () {
-            const session = CodeWhispererSessionState.instance.getSession()
+            session = CodeWhispererSessionState.instance.getSession()
             await resetCodeWhispererGlobalVariables()
-            session.reset()
         })
 
         afterEach(function () {
-            const session = CodeWhispererSessionState.instance.getSession()
             sinon.restore()
             session.reset()
         })
@@ -51,7 +51,6 @@ describe('onInlineAcceptance', function () {
         })
 
         it('Should report telemetry that records this user decision event', async function () {
-            const session = CodeWhispererSessionState.instance.getSession()
             await globals.globalState.update('CODEWHISPERER_USER_GROUP', {
                 version: extensionVersion,
             })

packages/core/src/codewhisperer/index.ts

@@ -59,7 +59,7 @@ export { onAcceptance } from './commands/onAcceptance'
 export { CodeWhispererTracker } from './tracker/codewhispererTracker'
 export { RecommendationHandler } from './service/recommendationHandler'
 export { CodeWhispererUserGroupSettings } from './util/userGroupUtil'
-export { CodeWhispererSessionState } from './util/codeWhispererSession'
+export { CodeWhispererSessionState, CodeWhispererSession } from './util/codeWhispererSession'
 export { onInlineAcceptance } from './commands/onInlineAcceptance'
 export { stopTransformByQ } from './commands/startTransformByQ'
 export { getCompletionItems, getCompletionItem, getLabel } from './service/completionProvider'