chore(e2e-tests): add waitForAnimations for document results (#7210)

Author: Anemy

packages/compass-crud/src/components/table-view/document-table-view.tsx

@@ -1004,6 +1004,7 @@ class DocumentTableView extends React.Component<DocumentTableViewProps> {
           'document-table-view-container',
           this.props.darkMode && 'document-table-view-container-darkmode'
         )}
+        data-testid="document-list"
       >
         <div className={cx('ag-parent', this.props.className)}>
           <BreadcrumbComponent

packages/compass-crud/src/components/virtualized-document-json-view.tsx

@@ -111,6 +111,7 @@ const VirtualizedDocumentJsonView: React.FC<
       renderItem={renderItem}
       estimateItemInitialHeight={estimateDocumentInitialHeight}
       rowGap={spacing[200]}
+      dataTestId="document-list"
       itemDataTestId="document-json-item"
       // Keeping the overscanCount low here helps us avoid scroll dangling
       // issues

packages/compass-e2e-tests/helpers/commands/run-find.ts

@@ -26,5 +26,15 @@ export async function runFind(
       const resultId = await browser.getQueryId(tabName);
       return resultId !== initialResultId;
     });
+
+    if (
+      tabName === 'Documents' &&
+      (await browser.$(Selectors.DocumentList).isDisplayed())
+    ) {
+      // Wait for animations to finish on the document page.
+      // Because we're virtualizing the list on crud, the query results may
+      // change after the resultId has changed.
+      await browser.waitForAnimations(Selectors.DocumentList);
+    }
   }
 }

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

@@ -603,6 +603,8 @@ export const ImportFileOption =
 export const DocumentListEntry = '[data-testid="editable-document"]';
 export const DocumentJSONEntry = '[data-testid="document-json-item"]';
 export const DocumentExpandButton = '[data-testid="expand-document-button"]';
+export const DocumentList =
+  '[data-testid="document-list"] [data-testid="virtual-list-inner-container"]';
 export const SelectJSONView = '[data-testid="toolbar-view-json"]';
 export const SelectTableView = '[data-testid="toolbar-view-table"]';
 export const SelectListView = '[data-testid="toolbar-view-list"]';