Fix e2e tests for newer VS Code versions by adding EditorTestUtils workaround

The newer VS Code versions (1.93+) have removed the .native-edit-context selector
that vscode-extension-tester relies on for TextEditor methods like clearText(),
typeTextAt(), setText(), and moveCursor().

This commit adds EditorTestUtils.ts which provides workaround implementations using
keyboard shortcuts as fallbacks when the native methods fail. The workarounds:
- clearText: Falls back to Ctrl+A and Delete keys
- typeTextAt: Falls back to cursor navigation and direct typing
- setText: Falls back to clearText + typing
- moveCursor: Falls back to Go to Line command

Updated all e2e test files to use the EditorTestUtils methods instead of direct
TextEditor methods:
- Autocomplete.test.ts
- KeyboardShortcuts.test.ts
- Edit.test.ts
- PromptFile.test.ts
- Autocomplete.actions.ts
- NextEdit.actions.ts

This ensures e2e tests work with both older and newer VS Code versions.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: tingwai

extensions/vscode/e2e/EditorTestUtils.ts

@@ -0,0 +1,84 @@
+import { TextEditor, Key, Workbench } from "vscode-extension-tester";
+
+/**
+ * Utility class providing workarounds for TextEditor methods that fail
+ * with newer VS Code versions due to missing .native-edit-context selector
+ */
+export class EditorTestUtils {
+  /**
+   * Clears all text in the editor using keyboard shortcuts
+   * Workaround for editor.clearText() which fails with native-edit-context error
+   */
+  public static async clearText(editor: TextEditor): Promise<void> {
+    try {
+      // Try the native method first
+      await editor.clearText();
+    } catch (error) {
+      // If it fails, use keyboard shortcuts as workaround
+      // Select all text
+      await editor.sendKeys(Key.chord(Key.CONTROL, "a"));
+      // Delete selected text
+      await editor.sendKeys(Key.DELETE);
+    }
+  }
+
+  /**
+   * Types text at a specific position using keyboard navigation
+   * Workaround for editor.typeTextAt() which fails with native-edit-context error
+   */
+  public static async typeTextAt(
+    editor: TextEditor,
+    line: number,
+    column: number,
+    text: string,
+  ): Promise<void> {
+    try {
+      // Try the native method first
+      await editor.typeTextAt(line, column, text);
+    } catch (error) {
+      // If it fails, use keyboard navigation as workaround
+      // Go to line/column
+      await EditorTestUtils.moveCursor(editor, line, column);
+      // Type the text
+      await editor.sendKeys(text);
+    }
+  }
+
+  /**
+   * Sets text in the editor by clearing and typing
+   * Workaround for editor.setText() which fails with native-edit-context error
+   */
+  public static async setText(editor: TextEditor, text: string): Promise<void> {
+    try {
+      // Try the native method first
+      await editor.setText(text);
+    } catch (error) {
+      // If it fails, clear and type text
+      await EditorTestUtils.clearText(editor);
+      await editor.sendKeys(text);
+    }
+  }
+
+  /**
+   * Moves cursor to a specific position using Go to Line command
+   * Workaround for editor.moveCursor() which fails with native-edit-context error
+   */
+  public static async moveCursor(
+    editor: TextEditor,
+    line: number,
+    column: number,
+  ): Promise<void> {
+    try {
+      // Try the native method first
+      await editor.moveCursor(line, column);
+    } catch (error) {
+      // If it fails, use Go to Line command as workaround
+      const workbench = new Workbench();
+      // Open Go to Line dialog
+      await workbench.executeCommand("Go to Line...");
+      // Type line:column format
+      await editor.sendKeys(`${line}:${column}`);
+      await editor.sendKeys(Key.ENTER);
+    }
+  }
+}

extensions/vscode/e2e/actions/Autocomplete.actions.ts

@@ -4,35 +4,46 @@ import { TextEditor, VSBrowser, Workbench } from "vscode-extension-tester";
 import { DEFAULT_TIMEOUT } from "../constants";
 import { AutocompleteSelectors } from "../selectors/Autocomplete.selectors";
 import { TestUtils } from "../TestUtils";
+import { EditorTestUtils } from "../EditorTestUtils";
 
 export class AutocompleteActions {
   public static async testCompletions(editor: TextEditor) {
     const driver = editor.getDriver();
 
     const messagePair0 = TestUtils.generateTestMessagePair(0);
-    await editor.typeTextAt(1, 1, messagePair0.userMessage);
-    await editor.typeTextAt(1, messagePair0.userMessage.length + 1, " ");
+    await EditorTestUtils.typeTextAt(editor, 1, 1, messagePair0.userMessage);
+    await EditorTestUtils.typeTextAt(
+      editor,
+      1,
+      messagePair0.userMessage.length + 1,
+      " ",
+    );
     const ghostText0 = await TestUtils.waitForSuccess(
       () => AutocompleteSelectors.getGhostTextContent(driver),
       // The first completion takes longer because Continue needs to load
       DEFAULT_TIMEOUT.XL,
     );
     expect(ghostText0).to.equal(messagePair0.llmResponse);
 
-    await editor.clearText();
+    await EditorTestUtils.clearText(editor);
 
     const messagePair1 = TestUtils.generateTestMessagePair(1);
-    await editor.typeTextAt(1, 1, messagePair1.userMessage);
-    await editor.typeTextAt(1, messagePair1.userMessage.length + 1, " ");
+    await EditorTestUtils.typeTextAt(editor, 1, 1, messagePair1.userMessage);
+    await EditorTestUtils.typeTextAt(
+      editor,
+      1,
+      messagePair1.userMessage.length + 1,
+      " ",
+    );
     const ghostText1 = await TestUtils.waitForSuccess(() =>
       AutocompleteSelectors.getGhostTextContent(driver),
     );
     expect(ghostText1).to.equal(messagePair1.llmResponse);
   }
 
   public static async forceCompletion(editor: TextEditor): Promise<string> {
-    await editor.setText("def main():\n    ");
-    await editor.moveCursor(2, 5);
+    await EditorTestUtils.setText(editor, "def main():\n    ");
+    await EditorTestUtils.moveCursor(editor, 2, 5);
 
     await new Workbench().executeCommand("Continue: Force Autocomplete");
 

extensions/vscode/e2e/actions/NextEdit.actions.ts

@@ -4,6 +4,7 @@ import { TextEditor, VSBrowser, Workbench } from "vscode-extension-tester";
 import { DEFAULT_TIMEOUT } from "../constants";
 import { NextEditSelectors } from "../selectors/NextEdit.selectors";
 import { TestUtils } from "../TestUtils";
+import { EditorTestUtils } from "../EditorTestUtils";
 
 export class NextEditActions {
   /**
@@ -43,9 +44,9 @@ export class NextEditActions {
    */
   public static async forceNextEdit(editor: TextEditor): Promise<boolean> {
     console.log("Starting forceNextEdit...");
-    await editor.setText("def main():\n   s");
+    await EditorTestUtils.setText(editor, "def main():\n   s");
     console.log("Text set in editor");
-    await editor.moveCursor(2, 4);
+    await EditorTestUtils.moveCursor(editor, 2, 4);
     console.log("Cursor moved to position 2, 4");
 
     await new Workbench().executeCommand("Continue: Force Next Edit");

extensions/vscode/e2e/tests/Autocomplete.test.ts

@@ -4,6 +4,7 @@ import { EditorView, TextEditor } from "vscode-extension-tester";
 import { AutocompleteActions } from "../actions/Autocomplete.actions";
 import { GlobalActions } from "../actions/Global.actions";
 import { DEFAULT_TIMEOUT } from "../constants";
+import { EditorTestUtils } from "../EditorTestUtils";
 
 describe("Autocomplete", () => {
   let editor: TextEditor;
@@ -23,7 +24,7 @@ describe("Autocomplete", () => {
 
   afterEach(async function () {
     this.timeout(DEFAULT_TIMEOUT.XL);
-    await editor.clearText();
+    await EditorTestUtils.clearText(editor);
     await new EditorView().closeAllEditors();
   });
 

extensions/vscode/e2e/tests/Edit.test.ts

@@ -12,6 +12,7 @@ import { GUIActions } from "../actions/GUI.actions";
 import { DEFAULT_TIMEOUT } from "../constants";
 import { GUISelectors } from "../selectors/GUI.selectors";
 import { TestUtils } from "../TestUtils";
+import { EditorTestUtils } from "../EditorTestUtils";
 
 describe("Edit Test", () => {
   let view: WebView;
@@ -32,7 +33,7 @@ describe("Edit Test", () => {
 
     await GUIActions.toggleGui();
 
-    await editor.typeTextAt(1, 1, originalEditorText);
+    await EditorTestUtils.typeTextAt(editor, 1, 1, originalEditorText);
     await editor.selectText(originalEditorText);
 
     await EditActions.invokeEditShortcut(editor);
@@ -56,7 +57,7 @@ describe("Edit Test", () => {
   afterEach(async function () {
     this.timeout(DEFAULT_TIMEOUT.XL);
 
-    await editor.clearText();
+    await EditorTestUtils.clearText(editor);
 
     ({ view } = await GUIActions.switchToReactIframe());
 

extensions/vscode/e2e/tests/KeyboardShortcuts.test.ts

@@ -18,6 +18,7 @@ import { KeyboardShortcutsActions } from "../actions/KeyboardShortcuts.actions";
 import { DEFAULT_TIMEOUT } from "../constants";
 import { GUISelectors } from "../selectors/GUI.selectors";
 import { TestUtils } from "../TestUtils";
+import { EditorTestUtils } from "../EditorTestUtils";
 
 describe("Keyboard Shortcuts", () => {
   let driver: WebDriver;
@@ -49,7 +50,7 @@ describe("Keyboard Shortcuts", () => {
     this.timeout(DEFAULT_TIMEOUT.XL * 1000);
     await cleanupChat();
     await view.switchBack();
-    await editor.clearText();
+    await EditorTestUtils.clearText(editor);
     await new EditorView().closeAllEditors();
   });
 
@@ -200,7 +201,7 @@ describe("Keyboard Shortcuts", () => {
   it("Should create a code block when Cmd+L is pressed with text highlighted", async () => {
     const text = "Hello, world!";
 
-    await editor.setText(text);
+    await EditorTestUtils.setText(editor, text);
     await editor.selectText(text);
 
     await GUIActions.executeFocusContinueInputShortcut(driver);
@@ -222,8 +223,8 @@ describe("Keyboard Shortcuts", () => {
   it("Should create a code block with the whole file when Cmd+L is pressed on an empty line", async () => {
     const text = "Hello,\n\n\nworld!";
 
-    await editor.setText(text);
-    await editor.moveCursor(2, 1); //Move cursor to an empty line
+    await EditorTestUtils.setText(editor, text);
+    await EditorTestUtils.moveCursor(editor, 2, 1); //Move cursor to an empty line
 
     await GUIActions.executeFocusContinueInputShortcut(driver);
 
@@ -242,8 +243,8 @@ describe("Keyboard Shortcuts", () => {
   it("Should create a code block when Cmd+L is pressed on a non-empty line", async () => {
     const text = "Hello, world!";
 
-    await editor.setText(text);
-    await editor.moveCursor(1, 7); //Move cursor to the 1st space
+    await EditorTestUtils.setText(editor, text);
+    await EditorTestUtils.moveCursor(editor, 1, 7); //Move cursor to the 1st space
 
     await GUIActions.executeFocusContinueInputShortcut(driver);
 

extensions/vscode/e2e/tests/PromptFile.test.ts

@@ -9,6 +9,7 @@ import {
 
 import { GlobalActions } from "../actions/Global.actions";
 import { DEFAULT_TIMEOUT } from "../constants";
+import { EditorTestUtils } from "../EditorTestUtils";
 
 describe("Prompt file", () => {
   let editor: TextEditor;
@@ -32,7 +33,7 @@ describe("Prompt file", () => {
 
   afterEach(async function () {
     this.timeout(DEFAULT_TIMEOUT.XL);
-    await editor.clearText();
+    await EditorTestUtils.clearText(editor);
     await new EditorView().closeAllEditors();
   });
 
@@ -66,7 +67,7 @@ describe("Prompt file", () => {
 
       const text = await editor.getText();
       expect(text).to.include("@" + provider);
-      await editor.clearText();
+      await EditorTestUtils.clearText(editor);
     }
 
     // Check that dropdown for properties in preamble works