fix e2e tests

Author: mabaasit

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

@@ -1447,13 +1447,13 @@ export const DataModelPreviewCollection = (collectionId: string) =>
   `${DataModelPreview} [data-nodeid="${collectionId}"]`;
 export const DataModelApplyEditor = `${DataModelEditor} [data-testid="apply-editor"]`;
 export const DataModelEditorApplyButton = `${DataModelApplyEditor} [data-testid="apply-button"]`;
-export const DataModelDownloadButton = 'button[aria-label="Download"]';
 export const DataModelUndoButton = 'button[aria-label="Undo"]';
 export const DataModelRedoButton = 'button[aria-label="Redo"]';
 export const DataModelExportButton = 'button[aria-label="Export"]';
 export const DataModelExportModal = '[data-testid="export-diagram-modal"]';
 export const DataModelExportPngOption = `${DataModelExportModal} input[aria-label="PNG"]`;
 export const DataModelExportJsonOption = `${DataModelExportModal} input[aria-label="JSON"]`;
+export const DataModelExportDiagramOption = `${DataModelExportModal} input[aria-label="Diagram File"]`;
 export const DataModelExportModalConfirmButton =
   '[data-testid="export-button"]';
 export const DataModelsListItem = (diagramName?: string) => {

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

@@ -386,7 +386,7 @@ describe('Data Modeling tab', function () {
     expect(text).to.include('String string'.toLowerCase());
   });
 
-  it('downloads the data model and opens it', async function () {
+  it('exports the data model to compass format and imports it back', async function () {
     const dataModelName = 'Test Export Model - Save-Open';
     exportFileName = `${dataModelName}.compass`;
     await setupDiagram(browser, {
@@ -406,8 +406,12 @@ describe('Data Modeling tab', function () {
 
     await browser.waitForAnimations(dataModelEditor);
 
-    await browser.clickVisible(Selectors.DataModelDownloadButton);
-    await browser.waitForAnimations(dataModelEditor);
+    await browser.clickVisible(Selectors.DataModelExportButton);
+    const exportModal = browser.$(Selectors.DataModelExportModal);
+    await exportModal.waitForDisplayed();
+
+    await browser.clickParent(Selectors.DataModelExportDiagramOption);
+    await browser.clickVisible(Selectors.DataModelExportModalConfirmButton);
 
     const { fileExists, filePath } = await waitForFileDownload(
       exportFileName,