fix(amazonq): Inline test after function addition, writeToTextEditor fix, and package.json pathing fix (#7800)

## Problem
The inline test cannot be run consecutively with other tests in our
suite since it switches its focus to the editorView and textEditorView.
Each time the inline test is run, a new driver was launched with caused
a new VSCode instance to instantiate.

The `package.json` should only target the `.test.js` files. Right now it
targets all .js files in the directory.

The `writeToTextEditor` function is faulty since it is wrongly indexed
due to its element not being accessed before counting and referencing
its indices within code.

## Solution
Implemented an `after` function which correctly switches back to our
webviewView (AmazonQ). Added the true inline test into the suite and
fixed driver problem.

Changed the path referenced in the `package.json` to `.test.js` files. 

Fixed the `writeToTextEditor` function such that it includes a "dummy"
space input into the textEditor and then counts the number of indices of
lines. This then writes the desired text within the correct line without
problems.

---

- Treat all work as PUBLIC. Private `feature/x` branches will not be
squash-merged at release time.
- Your code changes must meet the guidelines in
[CONTRIBUTING.md](https://github.com/aws/aws-toolkit-vscode/blob/master/CONTRIBUTING.md#guidelines).
- License: I confirm that my contribution is made under the terms of the
Apache 2.0 license.

Author: surajrdy-aws

package.json

@@ -30,7 +30,7 @@
         "testE2E": "npm run testE2E -w packages/ --if-present",
         "test:ui:prepare": "./node_modules/.bin/extest get-vscode -s ~/.vscode-test-resources -n && extest get-chromedriver -s ~/.vscode-test-resources -n",
         "test:ui:install": "cd packages/amazonq && npm run package 2>&1 | grep -o 'VSIX Version: [^ ]*' | cut -d' ' -f3 | xargs -I{} bash -c 'cd ../../ && ./node_modules/.bin/extest install-vsix -f amazon-q-vscode-{}.vsix -e packages/amazonq/test/e2e_new/amazonq/resources -s ~/.vscode-test-resources'",
-        "test:ui:run": "npm run testCompile && ./node_modules/.bin/extest run-tests -s ~/.vscode-test-resources -e packages/amazonq/test/e2e_new/amazonq/resources packages/amazonq/dist/test/e2e_new/amazonq/tests/*.js",
+        "test:ui:run": "npm run testCompile && ./node_modules/.bin/extest run-tests -s ~/.vscode-test-resources -e packages/amazonq/test/e2e_new/amazonq/resources packages/amazonq/dist/test/e2e_new/amazonq/tests/*.test.js",
         "test:ui": "npm run test:ui:prepare && npm run test:ui:install && npm run test:ui:run",
         "testInteg": "npm run testInteg -w packages/ --if-present",
         "package": "npm run package -w packages/toolkit -w packages/amazonq",

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 { pressKey, createNewTextFile, writeToTextEditor } from '../utils/generalUtils'
+import { createNewTextFile, writeToTextEditor, sleep } from '../utils/generalUtils'
 import assert from 'assert'
 
 describe('Amazon Q Inline Completion / Chat Functionality', function () {
@@ -20,25 +20,31 @@ describe('Amazon Q Inline Completion / Chat Functionality', function () {
         webviewView = testContext.webviewView
         await webviewView.switchBack()
         workbench = testContext.workbench
-
         editorView = new EditorView()
         testContext.editorView = editorView
-
         textEditor = await createNewTextFile(workbench, editorView)
     })
-
-    it('Inline Test', async () => {
-        await writeToTextEditor(textEditor, 'Select Me')
+    after(async function () {
+        // Switch back to Webview Iframe when dealing with external webviews from Amazon Q.
+        await editorView.closeAllEditors()
+        await webviewView.switchToFrame()
+    })
+    it('Inline Test Shortcut', async () => {
+        await writeToTextEditor(textEditor, 'def factorial(n):')
         const text = await textEditor.getText()
-        assert.equal(text, 'Select Me')
+        assert.equal(text, 'def factorial(n): ')
         await textEditor.clearText()
 
+        const textBefore = await textEditor.getText()
         await workbench.executeCommand('Amazon Q: Inline Chat')
         const input = new InputBox()
-        await input.sendKeys('Write a simple sentece')
+        await input.sendKeys('Generate the fibonacci sequence through iteration')
         await input.sendKeys(Key.ENTER)
-        const driver = textEditor.getDriver()
-        await pressKey(driver, 'ENTER')
-        await pressKey(driver, 'TAB')
+        // Must wait for response to be generated.
+        await sleep(8000)
+
+        const textAfter = await textEditor.getText()
+        assert(textAfter.length > textBefore.length, 'Amazon Q should have generated code')
+        await textEditor.clearText()
     })
 })

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

@@ -157,9 +157,11 @@ export async function createNewTextFile(workbench: Workbench, editorView: Editor
  * @returns Promise<void>
  */
 export async function writeToTextEditor(textEditor: TextEditor, text: string): Promise<void> {
+    // We require a "dummy" space to be written such that we can properly index the
+    // number of lines to register the textEditor.
+    await textEditor.typeTextAt(1, 1, ' ')
     const currentLines = await textEditor.getNumberOfLines()
-    const nextLine = currentLines + 1
-    await textEditor.typeTextAt(nextLine, 1, text)
+    await textEditor.typeTextAt(currentLines, 1, text)
 }
 
 /**