feat(amazonq): add shortcut test for UI E2E Tests (#7863)

## Problem
We have not implemented the shortcut tests for our UI E2E Test suite
with VSCode-Extension-Tester. Additionally, we face a small problem in
our pressShortcut abstraction that throws a type error when trying to
reference a Key command like `Key.COMMAND` since it is a IKey String.

## Solution
We have implemented the shortcut tests: 
- Optimize Code Shortcut 
- Generate Test Shortcut
- Keybind Check
- Explain Code Shortcut

We have also fixed the pressShortcut abstraction with a small edit in
the logic of accepted parameters.

NOTE: We have chosen to implement the Inline Shortcut within the Inline
Tests even though it could be classified under Shorcuts.

---

- 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

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

@@ -0,0 +1,80 @@
+/*!
+ * 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 {
+    clearChatInput,
+    pressShortcut,
+    createNewTextFile,
+    writeToTextEditor,
+    waitForChatResponse,
+} from '../utils/generalUtils'
+import { closeAllTabs } from '../utils/cleanupUtils'
+
+describe('Amazon Q Shortcut Functionality Tests', 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 clearChatInput(webviewView)
+    })
+    it('Allows User to Verify Command Palette Works as Expected', async () => {
+        const driver = webviewView.getDriver()
+        await pressShortcut(driver, Key.CONTROL, 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('Allows User to Generate Tests Using Keybind', async () => {
+        await writeToTextEditor(textEditor, 'def fibonacci(n):')
+        await textEditor.selectText('def fibonacci(n):')
+
+        const driver = webviewView.getDriver()
+        await pressShortcut(driver, Key.CONTROL, Key.ALT, 't')
+        await textEditor.clearText()
+        await editorView.closeAllEditors()
+        await webviewView.switchToFrame()
+        await waitForChatResponse(webviewView)
+    })
+    it('Allows User to Select and Explain Code Using Keybind', async () => {
+        await writeToTextEditor(textEditor, 'def fibonacci(n):')
+        await textEditor.selectText('def fibonacci(n):')
+
+        const driver = webviewView.getDriver()
+        await pressShortcut(driver, Key.CONTROL, Key.ALT, 'e')
+        await textEditor.clearText()
+        await editorView.closeAllEditors()
+        await webviewView.switchToFrame()
+        await waitForChatResponse(webviewView)
+    })
+    it('Allows User to Optimize Code Using Keybind', async () => {
+        await writeToTextEditor(textEditor, 'def fibonacci(n):')
+        await textEditor.selectText('def fibonacci(n):')
+
+        const driver = webviewView.getDriver()
+        await pressShortcut(driver, Key.CONTROL, Key.ALT, 'a')
+        await textEditor.clearText()
+        await editorView.closeAllEditors()
+        await webviewView.switchToFrame()
+        await waitForChatResponse(webviewView)
+    })
+})

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

@@ -110,11 +110,18 @@ export async function pressKey(driver: WebDriver, key: keyof typeof Key): Promis
  * Ctrl + Shift + T | await pressShortcut(driver, Key.CONTROL, Key.SHIFT, 't')
  */
 export async function pressShortcut(driver: WebDriver, ...keys: (string | keyof typeof Key)[]): Promise<void> {
+    // Replace CONTROL with COMMAND on macOS
+    const platformKeys = keys.map((key) => {
+        if (key === Key.CONTROL && process.platform === 'darwin') {
+            return Key.COMMAND
+        }
+        return key
+    })
     const actions = driver.actions()
-    for (const key of keys) {
+    for (const key of platformKeys) {
         actions.keyDown(key)
     }
-    for (const key of keys.reverse()) {
+    for (const key of platformKeys.reverse()) {
         actions.keyUp(key)
     }
     await actions.perform()