Assistant tests: Add option to useExplainPlanEntryPoint to avoid waiting for messages

Author: kraenhansen

packages/compass-e2e-tests/tests/assistant.test.ts

@@ -275,7 +275,7 @@ describe('MongoDB Assistant', function () {
       });
 
       it('should display opt-in modal for explain plan entry point', async function () {
-        await useExplainPlanEntryPoint(browser);
+        await useExplainPlanEntryPoint(browser, { waitForMessages: false });
 
         await browser.waitForOpenModal(Selectors.AIOptInModal);
 
@@ -284,7 +284,9 @@ describe('MongoDB Assistant', function () {
           reverse: true,
         });
 
-        expect(await getDisplayedMessages(browser)).to.deep.equal([]);
+        expect(
+          await browser.$(Selectors.AssistantChatMessages).isDisplayed()
+        ).equals(false);
       });
     });
   });
@@ -602,7 +604,10 @@ async function waitForMessages(
   }
 }
 
-async function useExplainPlanEntryPoint(browser: CompassBrowser) {
+async function useExplainPlanEntryPoint(
+  browser: CompassBrowser,
+  { waitForMessages = true } = {}
+) {
   await browser.clickVisible(Selectors.AggregationExplainButton);
 
   await browser.clickVisible(Selectors.ExplainPlanInterpretButton);
@@ -611,5 +616,7 @@ async function useExplainPlanEntryPoint(browser: CompassBrowser) {
     reverse: true,
   });
 
-  await browser.$(Selectors.AssistantChatMessages).waitForDisplayed();
+  if (waitForMessages) {
+    await browser.$(Selectors.AssistantChatMessages).waitForDisplayed();
+  }
 }