addressed more comments

Author: laura-codess

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

@@ -5,7 +5,8 @@
 import '../utils/setup'
 import { Workbench, EditorView, InputBox, TextEditor, WebviewView, Key } from 'vscode-extension-tester'
 import { testContext } from '../utils/testContext'
-import { sleep, expect, pressKey, createNewTextFile, writeToTextEditor } from '../utils/generalUtils'
+import { pressKey, createNewTextFile, writeToTextEditor } from '../utils/generalUtils'
+import assert from 'assert'
 
 describe('Amazon Q Inline Completion / Chat Functionality', function () {
     // this timeout is the general timeout for the entire test suite
@@ -29,7 +30,7 @@ describe('Amazon Q Inline Completion / Chat Functionality', function () {
     it('Inline Test', async () => {
         await writeToTextEditor(textEditor, 'Select Me')
         const text = await textEditor.getText()
-        expect(text).equals('Select Me')
+        assert.equal(text, 'Select Me')
         await textEditor.clearText()
 
         await workbench.executeCommand('Amazon Q: Inline Chat')

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

@@ -167,7 +167,11 @@ export async function createNewTextFile(workbench: Workbench, editorView: Editor
     await workbench.executeCommand('Create: New File...')
     await (await InputBox.create()).selectQuickPick('Text File')
     await sleep(1000)
-    const textEditor = (await editorView.openEditor('Untitled-1')) as TextEditor
+    const editor = await editorView.openEditor('Untitled-1')
+    if (!editor || !(editor instanceof TextEditor)) {
+        throw new Error('Failed to open text editor')
+    }
+    const textEditor = editor as TextEditor
     return textEditor
 }