update e2e

Author: paula-stacho

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

@@ -60,7 +60,7 @@ describe('Data Modeling tab', function () {
     await screenshotIfFailed(compass, this.currentTest);
   });
 
-  it('creates a new data model using an existing connection', async function () {
+  it.only('creates a new data model using an existing connection', async function () {
     await browser.navigateToDataModeling();
 
     // Click on create new data model button
@@ -96,12 +96,14 @@ describe('Data Modeling tab', function () {
     const dataModelEditor = browser.$(Selectors.DataModelEditor);
     await dataModelEditor.waitForDisplayed();
 
-    let nodes = await getDiagramNodes(browser);
-    expect(nodes).to.have.lengthOf(2);
-    expect(nodes).to.deep.equal([
-      'test.testCollection1',
-      'test.testCollection2',
-    ]);
+    browser.waitUntil(async () => {
+      let nodes = await getDiagramNodes(browser);
+      expect(nodes).to.have.lengthOf(2);
+      expect(nodes).to.deep.equal([
+        'test.testCollection1',
+        'test.testCollection2',
+      ]);
+    });
 
     // Apply change to the model
     const newModel = {