[test] Stabilize flaky Data Grid tests (#16053)

mui · Jan 1, 2025 · 88438a1 · 88438a1
1 parent a5f4a73
commit 88438a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/packages/x-data-grid-premium/src/tests/statePersistence.DataGridPremium.test.tsx b/packages/x-data-grid-premium/src/tests/statePersistence.DataGridPremium.test.tsx
@@ -11,7 +11,6 @@ import {
 import { createRenderer, act } from '@mui/internal-test-utils';
 import { expect } from 'chai';
 import { getColumnValues } from 'test/utils/helperFn';
-import { GridInitialStatePremium } from '../models/gridStatePremium';
 
 const isJSDOM = /jsdom/.test(window.navigator.userAgent);
 
@@ -51,7 +50,7 @@ const FULL_INITIAL_STATE: GridInitialState = {
 describe('<DataGridPremium /> - State persistence', () => {
   const { render } = createRenderer();
 
-  let apiRef: React.MutableRefObject<GridApi>;
+  let apiRef: React.RefObject<GridApi>;
 
   function TestCase(props: Omit<DataGridPremiumProps, 'rows' | 'columns' | 'apiRef'>) {
     apiRef = useGridApiRef();
@@ -102,10 +101,7 @@ describe('<DataGridPremium /> - State persistence', () => {
         });
       });
 
-      let exportedState: GridInitialStatePremium = {};
-      await act(() => {
-        exportedState = apiRef.current.exportState();
-      });
+      const exportedState = apiRef.current.exportState();
       expect(exportedState.rowGrouping).to.deep.equal(FULL_INITIAL_STATE.rowGrouping);
       expect(exportedState.aggregation).to.deep.equal(FULL_INITIAL_STATE.aggregation);
     });
@@ -121,10 +117,7 @@ describe('<DataGridPremium /> - State persistence', () => {
         });
       });
 
-      let exportedState: GridInitialStatePremium = {};
-      await act(() => {
-        exportedState = apiRef.current.exportState({ exportOnlyDirtyModels: true });
-      });
+      const exportedState = apiRef.current.exportState({ exportOnlyDirtyModels: true });
       expect(exportedState.rowGrouping).to.deep.equal(FULL_INITIAL_STATE.rowGrouping);
       expect(exportedState.aggregation).to.deep.equal(FULL_INITIAL_STATE.aggregation);
     });

diff --git a/packages/x-data-grid/src/tests/rowSelection.DataGrid.test.tsx b/packages/x-data-grid/src/tests/rowSelection.DataGrid.test.tsx
@@ -519,7 +519,7 @@ describe('<DataGrid /> - Row selection', () => {
       await user.keyboard('{Shift>}[ArrowDown]{/Shift}');
       expect(getSelectedRowIds()).to.deep.equal([1, 2, 3]);
 
-      await user.keyboard('{ArrowDown/}');
+      await user.keyboard('{ArrowDown}');
       expect(getSelectedRowIds()).to.deep.equal([1, 2, 3]); // Already on the last row
     });