chore(e2e-tests): wait for data modeling components in tests

Author: Anemy

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

@@ -585,11 +585,14 @@ describe('Data Modeling tab', function () {
       const relationshipItem = drawer.$(
         Selectors.DataModelCollectionRelationshipItem(relationshipId)
       );
-      expect(await relationshipItem.isDisplayed()).to.be.true;
+      await relationshipItem.waitForDisplayed();
       expect(await relationshipItem.getText()).to.include('testCollection-one');
 
       // Edit the relationship
       await relationshipItem.waitForDisplayed();
+      await relationshipItem
+        .$(Selectors.DataModelCollectionRelationshipItemEdit)
+        .waitForDisplayed();
       await relationshipItem
         .$(Selectors.DataModelCollectionRelationshipItemEdit)
         .click();
@@ -619,7 +622,7 @@ describe('Data Modeling tab', function () {
         .click();
 
       // Verify that the relationship is removed from the list and the diagram
-      expect(await relationshipItem.isExisting()).to.be.false;
+      await relationshipItem.waitForDisplayed({ reverse: true });
       await getDiagramEdges(browser, 0);
     });