feature: add shortcut test

Author: surajrdy-aws

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

@@ -0,0 +1,96 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import '../utils/setup'
+import { Workbench, EditorView, TextEditor, InputBox, WebviewView, Key } from 'vscode-extension-tester'
+import { testContext } from '../utils/testContext'
+import {
+    clearChat,
+    pressShortcut,
+    createNewTextFile,
+    writeToTextEditor,
+    waitForChatResponse,
+} from '../utils/generalUtils'
+import { closeAllTabs } from '../utils/cleanupUtils'
+
+describe('Amazon Q Shortcut 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
+
+    beforeEach(async function () {
+        webviewView = testContext.webviewView
+        await webviewView.switchBack()
+        workbench = testContext.workbench
+        editorView = new EditorView()
+        testContext.editorView = editorView
+        textEditor = await createNewTextFile(workbench, editorView)
+    })
+
+    afterEach(async function () {
+        await closeAllTabs(webviewView)
+        await clearChat(webviewView)
+    })
+    it('Keybind Check', async () => {
+        const driver = webviewView.getDriver()
+        // Open Command Palette
+        await pressShortcut(driver, Key.COMMAND, Key.SHIFT, 'p')
+        const input = new InputBox()
+        await input.sendKeys('Preferences: Open Keyboard Shortcuts')
+        await input.sendKeys(Key.ENTER)
+        await editorView.closeAllEditors()
+        await webviewView.switchToFrame()
+    })
+    it('Generate Test Keybind', async () => {
+        await writeToTextEditor(textEditor, 'def fibonacci(n):')
+        await textEditor.selectText('def fibonacci(n):')
+
+        const driver = webviewView.getDriver()
+        await pressShortcut(driver, Key.COMMAND, Key.ALT, 't')
+        //Clean Up Text
+        await textEditor.clearText()
+        await editorView.closeAllEditors()
+        await webviewView.switchToFrame()
+        const responseReceived = await waitForChatResponse(webviewView)
+        if (!responseReceived) {
+            throw new Error('Chat response not received  within timeout')
+        }
+        console.log('Chat response detected successfully')
+    })
+    it('Explain Code Shortcut', async () => {
+        await writeToTextEditor(textEditor, 'def fibonacci(n):')
+        await textEditor.selectText('def fibonacci(n):')
+
+        const driver = webviewView.getDriver()
+        await pressShortcut(driver, Key.COMMAND, Key.ALT, 'e')
+        //Clean Up Text
+        await textEditor.clearText()
+        await editorView.closeAllEditors()
+        await webviewView.switchToFrame()
+        const responseReceived = await waitForChatResponse(webviewView)
+        if (!responseReceived) {
+            throw new Error('Chat response not received  within timeout')
+        }
+        console.log('Chat response detected successfully')
+    })
+    it('Optimize Code Shortcut', async () => {
+        await writeToTextEditor(textEditor, 'def fibonacci(n):')
+        await textEditor.selectText('def fibonacci(n):')
+
+        const driver = webviewView.getDriver()
+        await pressShortcut(driver, Key.COMMAND, Key.ALT, 'a')
+        //Clean Up Text
+        await textEditor.clearText()
+        await editorView.closeAllEditors()
+        await webviewView.switchToFrame()
+        const responseReceived = await waitForChatResponse(webviewView)
+        if (!responseReceived) {
+            throw new Error('Chat response not received  within timeout')
+        }
+        console.log('Chat response detected successfully')
+    })
+})

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

@@ -57,7 +57,7 @@ export async function pressKey(driver: WebDriver, key: keyof typeof Key): Promis
  * 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> {
+export async function pressShortcut(driver: WebDriver, ...keys: (string | keyof typeof Key)[]): Promise<void> {
     const actions = driver.actions()
     for (const key of keys) {
         actions.keyDown(key)