feat(amazonq): UI Tests Baseline Framework Complete Set Up

packages/amazonq/test/e2e_new/amazonq/helpers/pinContextHelper.ts

@@ -0,0 +1,106 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { By, WebviewView } from 'vscode-extension-tester'
+import { sleep, waitForElement } from '../utils/generalUtils'
+import { WebElement } from 'vscode-extension-tester'
+
+/**
+ * Clicks the "Pin Context" button in the chat interface
+ * @param webview The WebviewView instance
+ * @returns Promise<boolean> True if button was found and clicked, false otherwise
+ */
+export async function clickPinContextButton(webview: WebviewView): Promise<boolean> {
+    try {
+        const topBar = await waitForElement(webview, By.css('.mynah-prompt-input-top-bar'))
+        const buttons = await topBar.findElements(
+            By.css('.mynah-button.mynah-button-secondary.fill-state-always.status-clear.mynah-ui-clickable-item')
+        )
+        // double check the label to make sure it says "Pin Context"
+        for (const button of buttons) {
+            const label = await button.findElement(By.css('.mynah-button-label'))
+            const labelText = await label.getText()
+            console.log('THE BUTTON TEXT LABEL IS:', labelText)
+            if (labelText === '@Pin Context') {
+                console.log('Found Pin Context button, clicking...')
+                await button.click()
+                return true
+            }
+        }
+        console.log('Pin Context button not found')
+        return false
+    } catch (e) {
+        console.error('Error clicking Pin Context button:', e)
+        return false
+    }
+}
+
+/**
+ * Lists all the possible Pin Context menu items in the console.
+ * @param webview The WebviewView instance
+ * @returns Promise<boolean> Returns the items as a WebElement List and the labels in a string array
+ */
+export async function getPinContextMenuItems(webview: WebviewView): Promise<{ items: WebElement[]; labels: string[] }> {
+    try {
+        const menuList = await waitForElement(webview, By.css('.mynah-detailed-list-items-block'))
+        await sleep(3000)
+        const menuListItems = await menuList.findElements(By.css('.mynah-detailed-list-item.mynah-ui-clickable-item'))
+        const labels: string[] = []
+
+        for (const item of menuListItems) {
+            try {
+                const textWrapper = await item.findElement(By.css('.mynah-detailed-list-item-text'))
+                const nameElement = await textWrapper.findElement(By.css('.mynah-detailed-list-item-name'))
+                const labelText = await nameElement.getText()
+                labels.push(labelText)
+                console.log('Menu item found:', labelText)
+            } catch (e) {
+                labels.push('')
+                console.log('Could not get text for menu item')
+            }
+        }
+
+        return { items: menuListItems, labels }
+    } catch (e) {
+        console.error('Error getting Pin Context menu items:', e)
+        return { items: [], labels: [] }
+    }
+}
+
+/**
+ * Clicks a specific item in the Pin Context menu by its label text
+ * @param webview The WebviewView instance
+ * @param itemName The text label of the menu item to click
+ * @returns Promise<boolean> True if the item was found and clicked, false otherwise
+ *
+ * NOTE: To find all possible text labels, you can call getPinContextMenuItems
+ */
+export async function clickPinContextMenuItem(webview: WebviewView, itemName: string): Promise<boolean> {
+    try {
+        const menuList = await waitForElement(webview, By.css('.mynah-detailed-list-items-block'))
+        await sleep(3000)
+        const menuListItems = await menuList.findElements(By.css('.mynah-detailed-list-item.mynah-ui-clickable-item'))
+        for (const item of menuListItems) {
+            try {
+                const textWrapper = await item.findElement(By.css('.mynah-detailed-list-item-text'))
+                const nameElement = await textWrapper.findElement(By.css('.mynah-detailed-list-item-name'))
+                const labelText = await nameElement.getText()
+
+                if (labelText === itemName) {
+                    console.log(`Clicking Pin Context menu item: ${itemName}`)
+                    await item.click()
+                    return true
+                }
+            } catch (e) {
+                continue
+            }
+        }
+
+        console.log(`Pin Context menu item not found: ${itemName}`)
+        return false
+    } catch (e) {
+        console.error(`Error clicking Pin Context menu item ${itemName}:`, e)
+        return false
+    }
+}

packages/amazonq/test/e2e_new/amazonq/helpers/quickActionsHelper.ts

@@ -0,0 +1,74 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { By, WebElement, WebviewView } from 'vscode-extension-tester'
+import { writeToChat } from '../utils/generalUtils'
+import { sleep, waitForElements } from '../utils/generalUtils'
+
+/**
+ * Gets all quick action command menu items
+ * @param webview The WebviewView instance
+ * @returns Promise<{items: WebElement[], texts: string[]}> Array of menu items and their text labels
+ */
+export async function getQuickActionsCommands(webview: WebviewView): Promise<{ items: WebElement[]; texts: string[] }> {
+    try {
+        await writeToChat('/', webview, false)
+        await sleep(2000)
+
+        const menuItems = await waitForElements(
+            webview,
+            By.css('.mynah-detailed-list-item.mynah-ui-clickable-item.target-command'),
+            10000
+        )
+
+        const menuTexts = []
+        for (let i = 0; i < menuItems.length; i++) {
+            try {
+                const text = await menuItems[i].getText()
+                menuTexts.push(text)
+                console.log(`Command ${i + 1}: ${text}`)
+            } catch (e) {
+                menuTexts.push('')
+                console.log(`Could not get text for command ${i + 1}`)
+            }
+        }
+
+        console.log(`Found ${menuItems.length} quick action command items`)
+        return { items: menuItems, texts: menuTexts }
+    } catch (e) {
+        console.error('Error getting quick action commands:', e)
+        return { items: [], texts: [] }
+    }
+}
+
+/**
+ * Clicks a specific quick action command by name
+ * @param webview The WebviewView instance
+ * @param commandName The name of the command to click
+ * @returns Promise<boolean> True if command was found and clicked, false otherwise
+ */
+export async function clickQuickActionsCommand(webview: WebviewView, commandName: string): Promise<boolean> {
+    try {
+        const { items, texts } = await getQuickActionsCommands(webview)
+        if (items.length === 0) {
+            console.log('No quick action commands found to click')
+            return false
+        }
+        const indexToClick = texts.findIndex((text) => text === commandName)
+
+        if (indexToClick === -1) {
+            console.log(`Command "${commandName}" not found`)
+            return false
+        }
+        console.log(`Clicking on command: ${commandName}`)
+        await items[indexToClick].click()
+        await sleep(3000)
+        console.log('Command clicked successfully')
+        return true
+    } catch (e) {
+        console.error('Error clicking quick action command:', e)
+        return false
+    }
+}