Responding to PR comments from Laura

Author: surajrdy-aws

packages/amazonq/test/e2e_new/amazonq/tests/inline.test.ts

@@ -5,7 +5,7 @@
 import '../utils/setup'
 import { Workbench, EditorView, InputBox, TextEditor, WebviewView, Key } from 'vscode-extension-tester'
 import { testContext } from '../utils/testContext'
-import { createNewTextFile, writeToTextEditor, waitForEditorStabilization } from '../utils/generalUtils'
+import { createNewTextFile, writeToTextEditor, waitForInlineGeneration } from '../utils/generalUtils'
 import assert from 'assert'
 
 describe('Amazon Q Inline Completion / Chat Functionality', function () {
@@ -41,7 +41,7 @@ describe('Amazon Q Inline Completion / Chat Functionality', function () {
         await input.sendKeys('Generate the fibonacci sequence through iteration')
         await input.sendKeys(Key.ENTER)
         // Wait for Amazon Q to finish generating code
-        await waitForEditorStabilization(textEditor)
+        await waitForInlineGeneration(textEditor)
 
         const textAfter = await textEditor.getText()
         assert(textAfter.length > textBefore.length, 'Amazon Q generated code')

packages/amazonq/test/e2e_new/amazonq/utils/generalUtils.ts

@@ -165,13 +165,13 @@ export async function writeToTextEditor(textEditor: TextEditor, text: string): P
 }
 
 /**
- * Waits for editor content to stabilize by checking if line count stops changing
+ * Waits for Inline Generation by Amazon Q by checking if line count stops changing
  * @param editor The TextEditor instance
  * @param timeout Maximum time to wait in milliseconds (default: 15000)
  * @returns Promise<void>
  * @throws Error if timeout is exceeded
  */
-export async function waitForEditorStabilization(editor: TextEditor, timeout = 15000): Promise<void> {
+export async function waitForInlineGeneration(editor: TextEditor, timeout = 15000): Promise<void> {
     const startTime = Date.now()
     let previousLines = await editor.getNumberOfLines()
     let stableCount = 0
@@ -182,7 +182,7 @@ export async function waitForEditorStabilization(editor: TextEditor, timeout = 1
 
         if (currentLines === previousLines) {
             stableCount++
-            if (stableCount >= 2) {
+            if (stableCount >= 3) {
                 return
             }
         } else {