Extend tests

elastic · Nov 16, 2023 · cc321d6 · cc321d6
1 parent 70ad0aa
commit cc321d6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/components/datagrid/body/header/data_grid_header_cell.test.tsx b/src/components/datagrid/body/header/data_grid_header_cell.test.tsx
@@ -256,9 +256,19 @@ describe('EuiDataGridHeaderCell', () => {
       fireEvent.click(toggle);
       waitForEuiPopoverOpen();
       expect(mockFocusContext.setFocusedCell).toHaveBeenCalledWith([0, -1]);
+      expect(
+        container.querySelector(
+          '.euiDataGridHeaderCell__button--withOpenPopover'
+        )
+      ).toBeInTheDocument();
 
       fireEvent.click(toggle);
       waitForEuiPopoverClose();
+      expect(
+        container.querySelector(
+          '.euiDataGridHeaderCell__button--withOpenPopover'
+        )
+      ).not.toBeInTheDocument();
     });
 
     describe('keyboard arrow navigation', () => {