test(Table/DownloadManager): fix event listener

ytsaurus · Feb 4, 2025 · 0256ca7 · 0256ca7
1 parent 93722c7
commit 0256ca7
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 9 deletions.
diff --git a/packages/ui/tests/.gitignore b/packages/ui/tests/.gitignore
@@ -3,3 +3,4 @@ node_modules/
 /playwright-report/
 /playwright/.cache/
 .env
+/downloads/
diff --git a/packages/ui/tests/e2e/pages/table.download-manager.base.spec.ts b/packages/ui/tests/e2e/pages/table.download-manager.base.spec.ts
@@ -17,18 +17,25 @@ test('Download manager: downloaded file should be equal to ideal', async ({page}
     await downloadButton.waitFor();
     await downloadButton.click();
 
-    const downloadButtonModal = page.locator('[data-qa="download-static-table"]');
+    const downloadButtonModal = page.getByTestId('download-static-table');
     await downloadButtonModal.waitFor();
+
+    const downloadPromise = page.waitForEvent('download');
     await downloadButtonModal.click();
 
-    page.on('download', async (download) => {
-        const destination = './downloads/' + download.suggestedFilename();
-        await download.saveAs(destination);
-
-        const fileHash = getFileHash(destination);
-        const idealFileHash = getFileHash('./data/static-table/static-table');
-        expect(fileHash).toEqual(idealFileHash);
-    });
+    const download = await downloadPromise;
+
+    const toastTitle = page.locator('.g-toast__title');
+    await toastTitle.waitFor();
+    const text = await toastTitle.innerText();
+    expect(text).toBe('Success');
+
+    const destination = './downloads/' + download.suggestedFilename();
+    await download.saveAs(destination);
+
+    const fileHash = getFileHash(destination);
+    const idealFileHash = getFileHash('./data/static-table/static-table');
+    expect(fileHash).toEqual(idealFileHash);
 });
 
 function getFileHash(filePath: string) {