[DataGrid] Initialize apiRef early (#11792)

mui · Jan 23, 2024 · 49587c6 · 49587c6
1 parent 4e9553d
commit 49587c6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/packages/grid/x-data-grid/src/hooks/core/useGridApiInitialization.ts b/packages/grid/x-data-grid/src/hooks/core/useGridApiInitialization.ts
@@ -141,6 +141,10 @@ export function useGridApiInitialization<
 
   useGridApiMethod(privateApiRef, { subscribeEvent, publishEvent } as any, 'public');
 
+  if (inputApiRef && !inputApiRef.current?.state) {
+    inputApiRef.current = publicApiRef.current;
+  }
+
   React.useImperativeHandle(inputApiRef, () => publicApiRef.current, [publicApiRef]);
 
   React.useEffect(() => {

diff --git a/packages/grid/x-data-grid/src/tests/rows.DataGrid.test.tsx b/packages/grid/x-data-grid/src/tests/rows.DataGrid.test.tsx
@@ -31,6 +31,7 @@ import {
   getRow,
   getActiveCell,
   getCell,
+  microtasks,
 } from 'test/utils/helperFn';
 import Dialog from '@mui/material/Dialog';
 
@@ -311,6 +312,7 @@ describe('<DataGrid /> - Rows', () => {
       await waitFor(() => {
         expect(getRow(0)).not.to.have.class('Mui-selected');
       });
+      await microtasks();
     });
 
     it('should not select the row when opening the menu', async () => {