commented test for when the toast order is swapped

Author: lerouxb

packages/compass-assistant/src/compass-assistant-drawer.tsx

@@ -33,6 +33,7 @@ export const CompassAssistantDrawer: React.FunctionComponent<{
         'The current chat will be cleared, and chat history will not be retrievable.',
       buttonText: 'Clear chat',
       variant: 'danger',
+      'data-testid': 'confirm-clear-chat-modal',
     });
     if (confirmed) {
       clearChat();

packages/compass-assistant/src/compass-assistant-provider.spec.tsx

@@ -200,19 +200,19 @@ describe('AssistantProvider', function () {
         userEvent.click(clearButton);
 
         await waitFor(() => {
-          expect(screen.getByTestId('confirmation-modal')).to.exist;
+          expect(screen.getByTestId('confirm-clear-chat-modal')).to.exist;
         });
 
         // There should be messages in the chat
         expect(screen.getByTestId('assistant-message-1')).to.exist;
         expect(screen.getByTestId('assistant-message-2')).to.exist;
 
-        const modal = screen.getByTestId('confirmation-modal');
+        const modal = screen.getByTestId('confirm-clear-chat-modal');
         const confirmButton = within(modal).getByText('Clear chat');
         userEvent.click(confirmButton);
 
         await waitForElementToBeRemoved(() =>
-          screen.getByTestId('confirmation-modal')
+          screen.getByTestId('confirm-clear-chat-modal')
         );
 
         expect(mockChat.messages).to.be.empty;
@@ -229,19 +229,19 @@ describe('AssistantProvider', function () {
         userEvent.click(clearButton);
 
         await waitFor(() => {
-          expect(screen.getByTestId('confirmation-modal')).to.exist;
+          expect(screen.getByTestId('confirm-clear-chat-modal')).to.exist;
         });
 
         // There should be messages in the chat
         expect(screen.getByTestId('assistant-message-1')).to.exist;
         expect(screen.getByTestId('assistant-message-2')).to.exist;
 
-        const modal = screen.getByTestId('confirmation-modal');
+        const modal = screen.getByTestId('confirm-clear-chat-modal');
         const cancelButton = within(modal).getByText('Cancel');
         userEvent.click(cancelButton);
 
         await waitForElementToBeRemoved(() =>
-          screen.getByTestId('confirmation-modal')
+          screen.getByTestId('confirm-clear-chat-modal')
         );
 
         expect(mockChat.messages).to.deep.equal(mockMessages);

packages/compass-e2e-tests/helpers/selectors.ts

@@ -242,6 +242,8 @@ export const ConnectionToastErrorReviewButton =
   '[data-testid="connection-error-review"]';
 export const ConenctionToastCancelConnectionButton =
   '[data-testid="cancel-connection-button"]';
+export const ConnectionToastErrorDebugButton =
+  '[data-testid="connection-error-debug"]';
 
 // Connections sidebar
 export const ConnectionsTitle = '[data-testid="connections-header"]';
@@ -1511,3 +1513,9 @@ export const SideDrawer = `[data-testid="${getDrawerIds().root}"]`;
 export const SideDrawerCloseButton = `[data-testid="${
   getDrawerIds().closeButton
 }"]`;
+
+// Assistant
+export const AssistantChatMessages = '[data-testid="assistant-chat-messages"]';
+export const AssistantClearChatButton = '[data-testid="assistant-clear-chat"]';
+export const ConfirmClearChatModal = '[data-testid="confirm-clear-chat-modal"]';
+export const ConfirmClearChatModalConfirmButton = `${ConfirmClearChatModal} [data-testid="lg-confirmation_modal-footer-confirm_button"]`;

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

@@ -273,6 +273,9 @@ describe('Connection string', function () {
   before(async function () {
     compass = await init(this.test?.fullTitle());
     browser = compass.browser;
+    // TODO(COMPASS-9746) the debug toast obscures the connection toast which
+    // breaks the connect custom commands and who knows what else
+    //await browser.setFeature('enableAIAssistant', true)
   });
 
   beforeEach(async function () {
@@ -326,6 +329,29 @@ describe('Connection string', function () {
       .$(Selectors.ConnectionFormErrorMessage)
       .getText();
     expect(errorText).to.equal('Authentication failed.');
+
+    // close the modal
+    await browser.clickVisible(Selectors.ConnectionModalCloseButton);
+    await browser
+      .$(Selectors.ConnectionModal)
+      .waitForDisplayed({ reverse: true });
+
+    // TODO(COMPASS-9746) the toasts should be swapped around which means these actions probably also need to be swapped around
+    /*
+    await browser.clickVisible(Selectors.ConnectionToastErrorDebugButton);
+    const messagesElement = browser.$(Selectors.AssistantChatMessages)
+    await messagesElement.waitForDisplayed();
+    // TODO(COMPASS-9744) check the response from the chatbot too
+    
+    await browser.waitUntil(async () => {
+      return (await messagesElement.getText()).includes('Given the error message below,');
+    });
+
+    // clear the chat so that a broken message doesn't break every future message
+    await browser.clickVisible(Selectors.AssistantClearChatButton);
+    await browser.clickVisible(Selectors.ConfirmClearChatModalConfirmButton);
+    await browser.clickVisible(Selectors.SideDrawerCloseButton);
+    */
   });
 
   it('can connect to an Atlas replicaset without srv', async function () {