move to reset

Author: leigaol

src/test/codewhisperer/commands/onAcceptance.test.ts

@@ -78,7 +78,6 @@ describe('onAcceptance', function () {
         })
 
         it('Should report telemetry that records this user decision event', async function () {
-            RecommendationHandler.instance.clearRecommendations()
             await globals.context.globalState.update(CodeWhispererConstants.userGroupKey, {
                 group: CodeWhispererConstants.UserGroup.Control,
                 version: extensionVersion,

src/test/codewhisperer/commands/onInlineAcceptance.test.ts

@@ -55,7 +55,6 @@ describe('onInlineAcceptance', function () {
         })
 
         it('Should report telemetry that records this user decision event', async function () {
-            await RecommendationHandler.instance.clearInlineCompletionStates()
             await globals.context.globalState.update(CodeWhispererConstants.userGroupKey, {
                 group: CodeWhispererConstants.UserGroup.Classifier,
                 version: extensionVersion,

src/test/codewhisperer/testUtil.ts

@@ -24,6 +24,7 @@ import { Service } from 'aws-sdk'
 import userApiConfig = require('./../../codewhisperer/client/user-service-2.json')
 import CodeWhispererUserClient = require('../../codewhisperer/client/codewhispereruserclient')
 import { codeWhispererClient } from '../../codewhisperer/client/codewhisperer'
+import { RecommendationHandler } from '../../codewhisperer/service/recommendationHandler'
 
 export async function resetCodeWhispererGlobalVariables() {
     vsCodeState.isIntelliSenseActive = false
@@ -32,6 +33,7 @@ export async function resetCodeWhispererGlobalVariables() {
     globals.telemetry.logger.clear()
     session.reset()
     await CodeSuggestionsState.instance.setSuggestionsEnabled(false)
+    await RecommendationHandler.instance.clearInlineCompletionStates()
 }
 
 export function createMockDocument(