Skip to content

Commit 3d42eca

Browse files
authored
fix(tests): unreliable codewhisperer tests (#4322)
2 parents 6bebba6 + d37eafd commit 3d42eca

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/test/codewhisperer/testUtil.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ import { Service } from 'aws-sdk'
2424
import userApiConfig = require('./../../codewhisperer/client/user-service-2.json')
2525
import CodeWhispererUserClient = require('../../codewhisperer/client/codewhispereruserclient')
2626
import { codeWhispererClient } from '../../codewhisperer/client/codewhisperer'
27+
import { RecommendationHandler } from '../../codewhisperer/service/recommendationHandler'
2728

2829
export async function resetCodeWhispererGlobalVariables() {
2930
vsCodeState.isIntelliSenseActive = false
@@ -32,6 +33,7 @@ export async function resetCodeWhispererGlobalVariables() {
3233
globals.telemetry.logger.clear()
3334
session.reset()
3435
await CodeSuggestionsState.instance.setSuggestionsEnabled(false)
36+
await RecommendationHandler.instance.clearInlineCompletionStates()
3537
}
3638

3739
export function createMockDocument(

0 commit comments

Comments
 (0)