fix deselect logic for e2e

determined-ai · Nov 5, 2024 · c48f016 · c48f016
1 parent 1ddbd76
commit c48f016
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/webui/react/src/e2e/models/components/TableActionBar.ts b/webui/react/src/e2e/models/components/TableActionBar.ts
@@ -26,6 +26,7 @@ export class TableActionBar extends NamedComponent {
   heatmapToggle = new BaseComponent({ parent: this, selector: '[data-test="heatmapToggle"]' });
   compare = new BaseComponent({ parent: this, selector: '[data-test="compare"]' });
   clearSelection = new BaseComponent({ parent: this, selector: '[data-test="clear-selection"]' });
+  selectAll = new BaseComponent({ parent: this, selector: '[data-test="select-all"]' });
   // TODO a bunch of modals
 }
 

diff --git a/webui/react/src/e2e/tests/experimentList.spec.ts b/webui/react/src/e2e/tests/experimentList.spec.ts
@@ -65,10 +65,15 @@ test.describe('Experiment List', () => {
       const count = await getCount();
       if (count !== 0) {
         await grid.headRow.clickSelectHeader();
-        const isClearSelectionVisible =
-          await projectDetailsPage.f_experimentList.tableActionBar.clearSelection.pwLocator.isVisible();
-        if (isClearSelectionVisible) {
-          await projectDetailsPage.f_experimentList.tableActionBar.clearSelection.pwLocator.click();
+        const selectAllButton = projectDetailsPage.f_experimentList.tableActionBar.selectAll;
+        const clearAllButton = projectDetailsPage.f_experimentList.tableActionBar.clearSelection;
+        if (await selectAllButton.pwLocator.isVisible()) {
+          await selectAllButton.pwLocator.click();
+          await clearAllButton.pwLocator.click();
+        } else if (await clearAllButton.pwLocator.isVisible()) {
+          await clearAllButton.pwLocator.click();
+        } else {
+          await grid.headRow.clickSelectHeader();
         }
       }
     });