test

Author: jpinkney-aws

packages/amazonq/test/unit/codewhisperer/testUtil.ts

@@ -37,15 +37,11 @@ export async function acceptByTab() {
     // we have to wait until the inline suggestion is shown in the editor however we don't have an useable API for that so hacky wait to know if the accept is taking effect
     await waitUntil(
         async () => {
-            await sleep(50)
             await vscode.commands.executeCommand('editor.action.inlineSuggest.commit')
-            console.log(editor.document.getText())
-            if (editor.document.getText() !== originalContent) {
-                return true
-            }
+            return vscode.window.activeTextEditor?.document.getText() !== originalContent
         },
         {
-            interval: 500,
+            interval: 50,
             timeout: 5000,
         }
     )

packages/core/src/codewhisperer/service/recommendationHandler.ts

@@ -70,6 +70,8 @@ const rejectCommand = Commands.declare('aws.amazonq.rejectCodeSuggestion', () =>
         traceId: TelemetryHelper.instance.traceId,
     })
 
+    // eslint-disable-next-line aws-toolkits/no-console-log
+    console.log('hiding inlineSuggest')
     await vscode.commands.executeCommand('editor.action.inlineSuggest.hide')
     RecommendationHandler.instance.reportUserDecisions(-1)
     await Commands.tryExecute('aws.amazonq.refreshAnnotation')
@@ -504,6 +506,8 @@ export class RecommendationHandler {
             // fix a regression that requires user to hit Esc twice to clear inline ghost text
             // because disposing a provider does not clear the UX
             if (isVscHavingRegressionInlineCompletionApi()) {
+                // eslint-disable-next-line aws-toolkits/no-console-log
+                console.log('hiding inlineSuggest')
                 await vscode.commands.executeCommand('editor.action.inlineSuggest.hide')
             }
         } finally {
@@ -694,7 +698,7 @@ export class RecommendationHandler {
                 // a new provider does not auto refresh the inline suggestion widget
                 // by manually refresh it
                 // eslint-disable-next-line aws-toolkits/no-console-log
-                console.log('using the regressioned api')
+                console.log('hiding inlineSuggest')
                 await vscode.commands.executeCommand('editor.action.inlineSuggest.hide')
                 await vscode.commands.executeCommand('editor.action.inlineSuggest.trigger')
             }