changes to rules

Author: laura-codess

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

@@ -48,6 +48,7 @@ describe('Amazon Q Rules Functionality', function () {
 
     after(async function () {
         await closeAllTabs(webviewView)
+        await VSBrowser.instance.quit()
     })
 
     it('Rules Option Test', async () => {

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

@@ -20,7 +20,7 @@ export async function sleep(timeout: number) {
  * @param timeout The timeout in milliseconds (Optional)
  * @returns Promise<WebElement> Returns the element found
  */
-export async function waitForElement(webview: WebviewView, locator: By, timeout?: number): Promise<WebElement> {
+export async function waitForElement(webview: WebviewView, locator: By, timeout: number = 8000): Promise<WebElement> {
     const driver = webview.getDriver()
     await driver.wait(until.elementsLocated(locator), timeout)
     return await webview.findWebElement(locator)
@@ -33,7 +33,11 @@ export async function waitForElement(webview: WebviewView, locator: By, timeout?
  * @param timeout The timeout in milliseconds (Optional)
  * @returns Promise<WebElement[]> Returns an array of elements found
  */
-export async function waitForElements(webview: WebviewView, locator: By, timeout?: number): Promise<WebElement[]> {
+export async function waitForElements(
+    webview: WebviewView,
+    locator: By,
+    timeout: number = 8000
+): Promise<WebElement[]> {
     const driver = webview.getDriver()
     await driver.wait(until.elementsLocated(locator), timeout)
     return await webview.findWebElements(locator)
@@ -91,7 +95,7 @@ export async function writeToChat(prompt: string, webview: WebviewView, send = t
  * @param timeout The timeout in milliseconds
  * @returns Promise<boolean> True if a response was detected, false if timeout occurred
  */
-export async function waitForChatResponse(webview: WebviewView, timeout = 15000): Promise<boolean> {
+export async function waitForChatResponse(webview: WebviewView, timeout = 8000): Promise<boolean> {
     const startTime = Date.now()
 
     while (Date.now() - startTime < timeout) {