From cc321d6827c2f5d5b4b084bd81f8c5ac147127e5 Mon Sep 17 00:00:00 2001
From: Julia Rechkunova <julia.rechkunova@elastic.co>
Date: Thu, 16 Nov 2023 14:05:45 +0100
Subject: [PATCH] Extend tests

---
 .../body/header/data_grid_header_cell.test.tsx         | 10 ++++++++++
 1 file changed, 10 insertions(+)

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
index 109489fedc3..5daee5968ef 100644
--- 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', () => {