added setup.ts and testContext so that auth can be shared across tests

Author: laura-codess

packages/amazonq/test/e2e_new/amazonq/chat.test.ts

@@ -2,36 +2,31 @@
  * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
-import { By, WebviewView } from 'vscode-extension-tester'
-import { loginToAmazonQ } from './framework/loginHelper'
+import './utils/setup'
+import { WebviewView } from 'vscode-extension-tester'
 import { closeAllTabs } from './framework/cleanupHelper'
-import { waitForElement } from './framework/generalHelper'
-import { waitForChatResponse } from './framework/chatHelper'
+import { testContext } from './utils/testContext'
+import { waitForChatResponse, writeToChat } from './framework/chatHelper'
 
 describe('Amazon Q Chat Basic Functionality', function () {
     // this timeout is the general timeout for the entire test suite
     this.timeout(150000)
     let webviewView: WebviewView
 
     before(async function () {
-        const result = await loginToAmazonQ()
-        webviewView = result.webviewView
+        webviewView = testContext.webviewView!
     })
 
     afterEach(async () => {
         await closeAllTabs(webviewView)
     })
 
     it('Chat Prompt Test', async () => {
-        const chatInput = await waitForElement(webviewView, By.css('.mynah-chat-prompt-input'))
-        await chatInput.sendKeys('Hello, Amazon Q!')
-        const sendButton = await waitForElement(webviewView, By.css('.mynah-chat-prompt-button'))
-        await sendButton.click()
+        await writeToChat('Hello, Amazon Q!', webviewView)
         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/framework/chatHelper.ts

@@ -3,6 +3,22 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { By, WebviewView } from 'vscode-extension-tester'
+import { waitForElement } from './generalHelper'
+
+/* Writes a prompt to the chat input and waits for a response
+
+Logic:
+Finds the chat input element using the .mynah-chat-prompt-input CSS selector,
+sends the provided prompt test, clicks the send button, and waits for a chat
+response. Returns true if successful, throws an error if the response times out */
+
+export async function writeToChat(prompt: string, webview: WebviewView): Promise<boolean> {
+    const chatInput = await waitForElement(webview, By.css('.mynah-chat-prompt-input'))
+    await chatInput.sendKeys(prompt)
+    const sendButton = await waitForElement(webview, By.css('.mynah-chat-prompt-button'))
+    await sendButton.click()
+    return true
+}
 
 /* Waits for a chat response and outputs whether the response is "correct"
 

packages/amazonq/test/e2e_new/amazonq/framework/loginHelper.ts

@@ -4,7 +4,7 @@
  */
 import { Workbench, By, WebviewView } from 'vscode-extension-tester'
 import { waitForElement, findItemByText } from './generalHelper'
-
+import { testContext } from '../utils/testContext'
 /* Completes the entire Amazon Q login flow
 
 Currently, the function will
@@ -16,7 +16,7 @@ Currently, the function will
 TO-DO: Currently this loginToAmazonQ is not fully autonomous as we ran into a blocker when the browser window pops up
 Documentation: https://quip-amazon.com/PoJOAyt4ja8H/Authentication-for-UI-Tests-Documentation */
 
-export async function loginToAmazonQ(): Promise<{ workbench: Workbench; webviewView: WebviewView }> {
+export async function loginToAmazonQ(): Promise<void> {
     const workbench = new Workbench()
     await workbench.executeCommand('Amazon Q: Open Chat')
 
@@ -48,5 +48,6 @@ export async function loginToAmazonQ(): Promise<{ workbench: Workbench; webviewV
     webviewView = new WebviewView()
     await webviewView.switchToFrame()
 
-    return { workbench, webviewView }
+    testContext.workbench = workbench
+    testContext.webviewView = webviewView
 }

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

@@ -0,0 +1,13 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { loginToAmazonQ } from '../framework/loginHelper'
+
+before(async function () {
+    this.timeout(60000) // Increase timeout to 60 seconds
+    console.log('\n\n*** MANUAL INTERVENTION REQUIRED ***')
+    console.log('When prompted, you must manually click to open the browser and complete authentication')
+    console.log('You have 60 seconds to complete this step\n\n')
+    await loginToAmazonQ()
+})

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

@@ -0,0 +1,13 @@
+/*!
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { Workbench, WebviewView } from 'vscode-extension-tester'
+
+export interface TestContext {
+    workbench?: Workbench
+    webviewView?: WebviewView
+}
+
+// arr to store shared context
+export const testContext: TestContext = {}