feat(amazonq): Implement inline completion test (#7752)

## Change

Implemented the inline completion test within inline.test.ts. This test
suite does not need any special helper functions but I added a bunch of
new functions in generalUtils.ts

Note: I added the types@selenium-webdriver because we have started to
interact with elements outside of the Amazon Q webview (ie. the text
editor)


---

- 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: laura-codess

package-lock.json

@@ -83,6 +83,7 @@
     },
     "dependencies": {
         "@types/node": "^22.7.5",
+        "@types/selenium-webdriver": "^4.1.28",
         "buffer": "^6.0.3",
         "jaro-winkler": "^0.2.8",
         "mocha": "^11.7.1",

package.json

@@ -0,0 +1,44 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+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 assert from 'assert'
+
+describe('Amazon Q Inline Completion / Chat Functionality', function () {
+    // this timeout is the general timeout for the entire test suite
+    this.timeout(150000)
+    let workbench: Workbench
+    let editorView: EditorView
+    let textEditor: TextEditor
+    let webviewView: WebviewView
+
+    before(async 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')
+        const text = await textEditor.getText()
+        assert.equal(text, 'Select Me')
+        await textEditor.clearText()
+
+        await workbench.executeCommand('Amazon Q: Inline Chat')
+        const input = new InputBox()
+        await input.sendKeys('Write a simple sentece')
+        await input.sendKeys(Key.ENTER)
+        const driver = textEditor.getDriver()
+        await pressKey(driver, 'ENTER')
+        await pressKey(driver, 'TAB')
+    })
+})

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

@@ -2,8 +2,8 @@
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
-import { By, WebviewView, WebElement } from 'vscode-extension-tester'
-import { until } from 'selenium-webdriver'
+import { By, WebviewView, WebElement, EditorView, InputBox, Workbench, TextEditor, Key } from 'vscode-extension-tester'
+import { until, WebDriver } from 'selenium-webdriver'
 
 /**
  * General sleep function to wait for a specified amount of time
@@ -39,6 +39,35 @@ export async function waitForElements(webview: WebviewView, locator: By, timeout
     return await webview.findWebElements(locator)
 }
 
+/**
+ * Presses a single key globally
+ * @param driver The WebDriver instance
+ * @param key The key to press
+ */
+export async function pressKey(driver: WebDriver, key: keyof typeof Key): Promise<void> {
+    await driver.actions().sendKeys(key).perform()
+}
+
+/**
+ * Presses a keyboard shortcut with modifier keys
+ * @param driver The WebDriver instance
+ * @param key The keys to press
+ *
+ * Examples:
+ * Ctrl + C | await pressShortcut(driver, Key.CONTROL, 'c')
+ * Ctrl + Shift + T | await pressShortcut(driver, Key.CONTROL, Key.SHIFT, 't')
+ */
+export async function pressShortcut(driver: WebDriver, ...keys: (keyof typeof Key)[]): Promise<void> {
+    const actions = driver.actions()
+    for (const key of keys) {
+        actions.keyDown(key)
+    }
+    for (const key of keys.reverse()) {
+        actions.keyUp(key)
+    }
+    await actions.perform()
+}
+
 /**
  * Writes text to the chat input and optionally sends it
  * @param prompt The text to write in the chat input
@@ -104,6 +133,35 @@ export async function clearChat(webview: WebviewView): Promise<boolean> {
     }
 }
 
+/**
+ * Creates a new text file and returns the editor
+ * @param workbench The Workbench instance
+ * @returns Promise<TextEditor> The text editor for the new file
+ */
+export async function createNewTextFile(workbench: Workbench, editorView: EditorView): Promise<TextEditor> {
+    await workbench.executeCommand('Create: New File...')
+    await (await InputBox.create()).selectQuickPick('Text File')
+    await sleep(1000)
+    const editor = await editorView.openEditor('Untitled-1')
+    if (!editor || !(editor instanceof TextEditor)) {
+        throw new Error('Failed to open text editor')
+    }
+    const textEditor = editor as TextEditor
+    return textEditor
+}
+
+/**
+ * Writes the given string in the textEditor in the next empty line
+ * @param textEditor The TextEditor instance
+ * @param text The text the user wants to type
+ * @returns Promise<void>
+ */
+export async function writeToTextEditor(textEditor: TextEditor, text: string): Promise<void> {
+    const currentLines = await textEditor.getNumberOfLines()
+    const nextLine = currentLines + 1
+    await textEditor.typeTextAt(nextLine, 1, text)
+}
+
 /**
  * Finds an item based on the text
  * @param items WebElement array to search

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

@@ -2,11 +2,12 @@
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
-import { Workbench, WebviewView } from 'vscode-extension-tester'
+import { Workbench, WebviewView, EditorView } from 'vscode-extension-tester'
 
 export interface TestContext {
     workbench: Workbench
     webviewView: WebviewView
+    editorView: EditorView
 }
 
 export const testContext = new Proxy<TestContext>({} as TestContext, {
@@ -24,7 +25,8 @@ export const testContext = new Proxy<TestContext>({} as TestContext, {
     },
 })
 
-export function initializeTestContext(workbench: Workbench, webviewView: WebviewView): void {
+export function initializeTestContext(workbench: Workbench, webviewView: WebviewView, editorView: EditorView): void {
     testContext.workbench = workbench
     testContext.webviewView = webviewView
+    testContext.editorView = editorView
 }