@iblai/iblai-js 1.4.2 → 1.4.3

package/dist/playwright/index.esm.js

@@ -824,6 +824,7 @@ async function expectNoAccessibilityViolationsOnDialogs(page, rules, exclude = [
     expect(results.violations).toEqual([]);
 }
 
+const deferredNotificationText = 'You will be notified once the report is available.';
 async function inviteUserTest(page, inviteModal) {
     const emailInput = inviteModal.locator('#email-invite');
     await expect(emailInput).toBeVisible({ timeout: 10000 });
@@ -847,6 +848,28 @@ async function navigateToAccountComponent(page, profileBtn) {
     await expect(tenantDialog).toBeVisible();
     return tenantDialog;
 }
+/**
+ * Click a report card's primary action and confirm the date-range picker, so callers
+ * always end up in the post-generation state.
+ *
+ * Prefers the "Regenerate report" button when present (completed reports) because the plain
+ * "Download report" button on a completed card re-downloads the existing URL and skips the
+ * picker. When no Regenerate button exists, clicks "Download report" — which on a
+ * non-completed card also opens the picker.
+ *
+ * Either way we wait for the picker and click Generate/Regenerate to start the report.
+ */
+async function triggerReportFromCard(page, card) {
+    const regenerateButton = card.getByRole('button', { name: 'Regenerate report' });
+    const downloadButton = card.getByRole('button', { name: 'Download report' });
+    const hasRegenerate = await regenerateButton.isVisible().catch(() => false);
+    const trigger = hasRegenerate ? regenerateButton : downloadButton;
+    await trigger.click();
+    const dateRangePopover = page.getByTestId('report-date-range-popover');
+    await expect(dateRangePopover).toBeVisible({ timeout: 10000 });
+    await dateRangePopover.getByRole('button', { name: /^(Generate|Regenerate)$/ }).click();
+    await expect(dateRangePopover).not.toBeVisible({ timeout: 10000 });
+}
 async function navigateToDataReports(page) {
     const dataReportsTab = page.getByRole('tab', { name: 'Data Reports' });
     await expect(dataReportsTab).toBeVisible({ timeout: 120000 });
@@ -898,14 +921,40 @@ async function shouldOpenCSVEditorDialog(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = userReportCard.getByRole('button', {
-        name: 'Download report',
-    });
-    await downloadButton.click();
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    await expect(csvEditorDialog).toBeVisible({ timeout: 60000 });
+    await triggerReportFromCard(page, userReportCard);
+    //expect Report generation processing...
+    await expect(page.getByText('Report generation processing...')).toBeVisible({ timeout: 5000 });
+    // After the processing toast appears, the backend either finishes in time (CSV editor opens)
+    // or polling times out (deferred-notification toast appears). Treat both as valid outcomes.
+    await waitForReportOutcomeOrCSVDialogOpen(page);
+}
+async function waitForReportOutcomeOrCSVDialogOpen(page) {
+    const csvEditorDialog = page.getByRole('dialog', { name: 'Edit CSV Data' });
+    const deferredNotification = page.getByText(deferredNotificationText);
+    const downloadPromise = page.waitForEvent('download', { timeout: 60000 });
+    const outcome = await Promise.race([
+        csvEditorDialog
+            .waitFor({ state: 'visible', timeout: 60000 })
+            .then(() => 'csv-editor'),
+        deferredNotification
+            .waitFor({ state: 'visible', timeout: 60000 })
+            .then(() => 'deferred'),
+        downloadPromise.then(() => 'download'),
+    ]).catch(() => null);
+    if (outcome === 'deferred') {
+        logger.info('Report generation deferred — user will be notified when available');
+        return false;
+    }
+    if (outcome === 'download') {
+        const download = await downloadPromise;
+        const filename = download.suggestedFilename();
+        expect(filename).toBe('report.csv');
+        logger.info(`CSV file "${filename}" downloaded successfully`);
+        return false;
+    }
+    if (outcome !== 'csv-editor') {
+        throw new Error('Report generation did not complete: neither CSV editor nor deferred-notification toast appeared within 60s');
+    }
     const dialogTitle = csvEditorDialog.getByRole('heading', {
         name: 'Edit CSV Data',
         level: 2,
@@ -919,6 +968,7 @@ async function shouldOpenCSVEditorDialog(page) {
     await expect(csvEditorDialog.getByRole('button', { name: 'Save' })).toBeVisible();
     await expect(csvEditorDialog.getByRole('button', { name: 'Close' }).first()).toBeVisible();
     logger.info('CSV Editor dialog opened successfully');
+    return csvEditorDialog;
 }
 async function shouldDisplayCSVInEditableTableFormat(page) {
     const dataReportsTab = page.getByRole('tab', { name: 'Data Reports' });
@@ -931,14 +981,12 @@ async function shouldDisplayCSVInEditableTableFormat(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = userReportCard.getByRole('button', {
-        name: 'Download report',
-    });
-    await downloadButton.click();
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    await expect(csvEditorDialog).toBeVisible({ timeout: 60000 });
+    await triggerReportFromCard(page, userReportCard);
+    const csvEditorDialog = await waitForReportOutcomeOrCSVDialogOpen(page);
+    if (!csvEditorDialog) {
+        test$2.skip();
+        return;
+    }
     const table = csvEditorDialog.getByRole('table', {
         name: 'CSV data table',
     });
@@ -962,14 +1010,12 @@ async function shouldAllowEditingCellValuesInCSVEditor(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = userReportCard.getByRole('button', {
-        name: 'Download report',
-    });
-    await downloadButton.click();
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    await expect(csvEditorDialog).toBeVisible({ timeout: 60000 });
+    await triggerReportFromCard(page, userReportCard);
+    const csvEditorDialog = await waitForReportOutcomeOrCSVDialogOpen(page);
+    if (!csvEditorDialog) {
+        test$2.skip();
+        return;
+    }
     const table = csvEditorDialog.getByRole('table', {
         name: 'CSV data table',
     });
@@ -1002,14 +1048,12 @@ async function shouldAddNewRowWhenClickingAddRowButton(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = userReportCard.getByRole('button', {
-        name: 'Download report',
-    });
-    await downloadButton.click();
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    await expect(csvEditorDialog).toBeVisible({ timeout: 60000 });
+    await triggerReportFromCard(page, userReportCard);
+    const csvEditorDialog = await waitForReportOutcomeOrCSVDialogOpen(page);
+    if (!csvEditorDialog) {
+        test$2.skip();
+        return;
+    }
     const table = csvEditorDialog.locator('table');
     const initialRowCount = await table.locator('tbody tr').count();
     const addRowButton = csvEditorDialog.getByRole('button', {
@@ -1032,16 +1076,14 @@ async function shouldSaveEditedCSVAndTriggerDownload(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = userReportCard.getByRole('button', {
-        name: 'Download report',
-    });
-    await downloadButton.click();
+    await triggerReportFromCard(page, userReportCard);
     await page.waitForLoadState('networkidle');
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    await expect(csvEditorDialog).toBeVisible({ timeout: 60000 });
-    const downloadPromise = page.waitForEvent('download', { timeout: 30000 });
+    const csvEditorDialog = await waitForReportOutcomeOrCSVDialogOpen(page);
+    if (!csvEditorDialog) {
+        test$2.skip();
+        return;
+    }
+    const downloadPromise = page.waitForEvent('download', { timeout: 40000 });
     const saveButton = csvEditorDialog.getByRole('button', { name: 'Save' });
     await saveButton.click();
     const download = await downloadPromise;
@@ -1061,15 +1103,13 @@ async function shouldCloseCSVEditorWithoutSavingWhenClickingCancel(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = userReportCard.getByRole('button', {
-        name: 'Download report',
-    });
-    await downloadButton.click();
+    await triggerReportFromCard(page, userReportCard);
     await page.waitForLoadState('networkidle');
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    await expect(csvEditorDialog).toBeVisible({ timeout: 60000 });
+    const csvEditorDialog = await waitForReportOutcomeOrCSVDialogOpen(page);
+    if (!csvEditorDialog) {
+        test$2.skip();
+        return;
+    }
     const cancelButton = csvEditorDialog.getByRole('button', {
         name: 'Cancel',
     });
@@ -1089,15 +1129,13 @@ async function shouldCloseCSVEditorWhenClickingCloseButton(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = userReportCard.getByRole('button', {
-        name: 'Download report',
-    });
-    await downloadButton.click();
+    await triggerReportFromCard(page, userReportCard);
     await page.waitForLoadState('networkidle');
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    await expect(csvEditorDialog).toBeVisible({ timeout: 60000 });
+    const csvEditorDialog = await waitForReportOutcomeOrCSVDialogOpen(page);
+    if (!csvEditorDialog) {
+        test$2.skip();
+        return;
+    }
     const closeButton = csvEditorDialog
         .getByRole('button', {
         name: 'Close',
@@ -1118,15 +1156,13 @@ async function shouldVerifyCSVEditorDialogAccessibility(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = userReportCard.getByRole('button', {
-        name: 'Download report',
-    });
-    await downloadButton.click();
+    await triggerReportFromCard(page, userReportCard);
     await page.waitForLoadState('networkidle');
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    await expect(csvEditorDialog).toBeVisible({ timeout: 60000 });
+    const csvEditorDialog = await waitForReportOutcomeOrCSVDialogOpen(page);
+    if (!csvEditorDialog) {
+        test$2.skip();
+        return;
+    }
     await expect(csvEditorDialog).toHaveAttribute('role', 'dialog');
     const csvTable = csvEditorDialog.getByRole('table', {
         name: 'CSV data table',
@@ -1161,15 +1197,12 @@ async function shouldOpenCSVEditorForUserMetadataReport(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = userMetadataReportCard.getByRole('button', {
-        name: 'Download report',
-    });
-    await downloadButton.click();
-    await page.waitForLoadState('networkidle');
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    await expect(csvEditorDialog).toBeVisible({ timeout: 60000 });
+    await triggerReportFromCard(page, userMetadataReportCard);
+    const csvEditorDialog = await waitForReportOutcomeOrCSVDialogOpen(page);
+    if (!csvEditorDialog) {
+        test$2.skip();
+        return;
+    }
     const table = csvEditorDialog.locator('table');
     const companyHeader = table.locator('thead th input[value="company"]');
     const hasCompanyColumn = await companyHeader.isVisible().catch(() => false);
@@ -1189,21 +1222,32 @@ async function shouldDirectlyDownloadChatHistoryReportWithoutCSVEditor(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = chatHistoryCard.getByRole('button', {
-        name: 'Download report',
-    });
-    const downloadPromise = page.waitForEvent('download', { timeout: 120000 });
-    await expect(downloadButton).toBeVisible({ timeout: 30000 });
-    await downloadButton.click();
-    try {
-        const download = await downloadPromise;
-        const filename = download.suggestedFilename();
+    // Prefer the Regenerate button (guaranteed to open the picker); fall back to Download.
+    // Start listening for the download event before clicking since the auto-download fires
+    // as soon as the picker is confirmed.
+    const downloadPromise = page.waitForEvent('download', { timeout: 40000 });
+    await triggerReportFromCard(page, chatHistoryCard);
+    // Either the browser fires a download within 40s, or the backend defers and shows the
+    // "You will be notified once the report is available." toast. Both are valid outcomes.
+    const deferredNotification = page.getByText(deferredNotificationText);
+    const outcome = await Promise.race([
+        downloadPromise.then((download) => ({ kind: 'download', download })),
+        deferredNotification
+            .waitFor({ state: 'visible', timeout: 40000 })
+            .then(() => ({ kind: 'deferred' })),
+    ]).catch(() => null);
+    if (!outcome) {
+        throw new Error('Chat History generation did not complete: neither a download nor the deferred-notification toast appeared within 40s');
+    }
+    if (outcome.kind === 'download') {
+        const filename = outcome.download.suggestedFilename();
         expect(filename.endsWith('.csv')).toBeTruthy();
         logger.info(`Chat History report downloaded: ${filename}`);
     }
-    catch (_a) {
-        logger.info('Chat History download may have completed without triggering download event');
+    else {
+        logger.info('Chat History generation deferred — user will be notified when available');
     }
+    // Regardless of outcome, the CSV editor dialog should never have opened for chat history.
     const csvEditorDialog = page.getByRole('dialog', {
         name: 'Edit CSV Data',
     });
@@ -1224,23 +1268,23 @@ async function shouldDisableOtherDownloadButtonsWhileGeneratingReport(page) {
         test$2.skip();
         return;
     }
-    await downloadButtons.first().click();
+    // Trigger generation on the first report card (preferring Regenerate when present so the
+    // date-range picker always shows, then confirming it).
+    const firstCard = page.getByLabel(/ report card$/).first();
+    await triggerReportFromCard(page, firstCard);
+    //slight timeout to wait for other download buttons to be disabled
     await page.waitForTimeout(500);
     const secondButton = downloadButtons.nth(1);
     const isSecondDisabled = await secondButton.isDisabled();
     if (isSecondDisabled) {
         logger.info('Other download buttons correctly disabled during report generation');
     }
-    const csvEditorDialog = page.getByRole('dialog', {
-        name: 'Edit CSV Data',
-    });
-    const dialogVisible = await csvEditorDialog
-        .waitFor({ state: 'visible', timeout: 30000 })
-        .then(() => true)
-        .catch(() => false);
-    if (dialogVisible) {
-        await csvEditorDialog.getByRole('button', { name: 'Close' }).first().click();
+    const csvEditorDialog = await waitForReportOutcomeOrCSVDialogOpen(page);
+    if (!csvEditorDialog) {
+        test$2.skip();
+        return;
     }
+    await csvEditorDialog.getByRole('button', { name: 'Close' }).first().click();
 }
 // ============================================
 // Combined Recommendation Reports Test Helpers
@@ -1260,12 +1304,9 @@ async function shouldShowCombiningReportsDialog(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = combinedReportCard
-        .first()
-        .getByRole('button', { name: 'Download report' });
-    await downloadButton.click();
+    await triggerReportFromCard(page, combinedReportCard.first());
     const combiningDialog = page.locator('[data-testid="combining-reports-dialog"]');
-    await expect(combiningDialog).toBeVisible({ timeout: 30000 });
+    await expect(combiningDialog).toBeVisible({ timeout: 40000 });
     await expect(combiningDialog.getByText('Combining Reports')).toBeVisible();
     await expect(combiningDialog.getByText('Loading & combining recommendations data')).toBeVisible();
     logger.info('Combining reports dialog displayed successfully');
@@ -1285,12 +1326,9 @@ async function shouldCancelCombiningReports(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = combinedReportCard
-        .first()
-        .getByRole('button', { name: 'Download report' });
-    await downloadButton.click();
+    await triggerReportFromCard(page, combinedReportCard.first());
     const combiningDialog = page.locator('[data-testid="combining-reports-dialog"]');
-    await expect(combiningDialog).toBeVisible({ timeout: 30000 });
+    await expect(combiningDialog).toBeVisible({ timeout: 40000 });
     const cancelButton = page.locator('[data-testid="cancel-combining-button"]');
     await expect(cancelButton).toBeVisible();
     await cancelButton.click();
@@ -1331,10 +1369,7 @@ async function shouldCombineRecommendationReports(page) {
         test$2.skip();
         return;
     }
-    const downloadButton = combinedReportCard
-        .first()
-        .getByRole('button', { name: 'Download report' });
-    await downloadButton.click();
+    await triggerReportFromCard(page, combinedReportCard.first());
     const combiningDialog = page.locator('[data-testid="combining-reports-dialog"]');
     const dialogOrEditor = await Promise.race([
         combiningDialog.waitFor({ state: 'visible', timeout: 10000 }).then(() => 'dialog'),