wait for diagram animations

Author: mabaasit

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

@@ -1451,4 +1451,5 @@ export const DataModelsListItem = (diagramName: string) =>
 export const DataModelsListItemActions = (diagramName: string) =>
   `${DataModelsListItem(diagramName)} [aria-label="Show actions"]`;
 export const DataModelsListItemDeleteButton = `[data-action="delete"]`;
+export const DataModelingDiagram = '.react-flow';
 export const DataModelingDiagramNode = '.react-flow__node > div';

packages/compass-e2e-tests/tests/data-modeling-tab.test.ts

@@ -15,6 +15,7 @@ import {
 } from '../helpers/insert-data';
 
 async function getDiagramNodes(browser: CompassBrowser): Promise<string[]> {
+  await browser.waitForAnimations(Selectors.DataModelingDiagram);
   return await browser
     .$$(Selectors.DataModelingDiagramNode)
     .map((element) => element.getAttribute('title'));