fix tests that now only work in Agent Mode

Author: chezsmithy

extensions/vscode/e2e/actions/GUI.actions.ts

@@ -91,6 +91,20 @@ export class GUIActions {
     await dropdownOption.click();
   };
 
+  public static selectModeFromDropdown = async (
+    view: WebView,
+    option: string,
+  ) => {
+    const dropdownButton = await GUISelectors.getModeDropdownButton(view);
+    await dropdownButton.click();
+
+    const dropdownOption = await TestUtils.waitForSuccess(() => {
+      return GUISelectors.getModeDropdownOption(view, option);
+    });
+
+    await dropdownOption.click();
+  };
+
   public static async sendMessage({
     view,
     message,

extensions/vscode/e2e/selectors/GUI.selectors.ts

@@ -67,6 +67,10 @@ export class GUISelectors {
     return SelectorUtils.getElementByDataTestId(view, "model-select-button");
   }
 
+  public static getModeDropdownButton(view: WebView) {
+    return SelectorUtils.getElementByDataTestId(view, "mode-select-button");
+  }
+
   public static getFirstContextProviderDropdownItem(view: WebView) {
     return SelectorUtils.getElementByDataTestId(
       view,
@@ -104,6 +108,12 @@ export class GUISelectors {
     );
   }
 
+  public static getModeDropdownOption(view: WebView, option: string) {
+    return view.findWebElement(
+      By.xpath(`//*[@role="listbox"]//*[contains(text(), "${option}")]`),
+    );
+  }
+
   public static getOnboardingTabButton(view: WebView, title: string) {
     return SelectorUtils.getElementByDataTestId(
       view,

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

@@ -239,9 +239,10 @@ describe("GUI Test", () => {
     }).timeout(DEFAULT_TIMEOUT.XL);
   });
 
-  describe("Chat with tools", () => {
+  describe("Agent with tools", () => {
     it("should render tool call", async () => {
       await GUIActions.selectModelFromDropdown(view, "TOOL MOCK LLM");
+      await GUIActions.selectModeFromDropdown(view, "Agent");
 
       const [messageInput] = await GUISelectors.getMessageInputFields(view);
       await messageInput.sendKeys("Hello");