[test] Fix DataGrid data source error test on React 18 (#16565)

mui · Feb 13, 2025 · 3dca3e1 · 3dca3e1
1 parent 6b4fdc8
commit 3dca3e1
Showing 1 changed file with 13 additions and 17 deletions.
diff --git a/packages/x-data-grid-pro/src/tests/dataSource.DataGridPro.test.tsx b/packages/x-data-grid-pro/src/tests/dataSource.DataGridPro.test.tsx
@@ -277,26 +277,22 @@ describeSkipIf(isJSDOM)('<DataGridPro /> - Data source', () => {
 
   describe('Error handling', () => {
     it('should call `unstable_onDataSourceError` when the data source returns an error', async () => {
-      const onDataSourceError = spy();
-
-      const dataSource: GridDataSource = {
-        getRows: () => {
-          throw new Error('Could not fetch the data');
-        },
-      };
-
-      render(
-        <div style={{ width: 300, height: 300 }}>
-          <DataGridPro
-            columns={[]}
-            unstable_dataSource={dataSource}
-            unstable_onDataSourceError={onDataSourceError}
-          />
-        </div>,
+      const onDataSourceErrorSpy = spy();
+      const { setProps } = render(
+        <TestDataSource
+          unstable_dataSourceCache={null}
+          unstable_onDataSourceError={onDataSourceErrorSpy}
+        />,
       );
+      expect(onDataSourceErrorSpy.callCount).to.equal(0);
+
+      setProps({ shouldRequestsFail: true });
+      act(() => {
+        apiRef.current.unstable_dataSource.fetchRows();
+      });
 
       await waitFor(() => {
-        expect(onDataSourceError.callCount).to.equal(1);
+        expect(onDataSourceErrorSpy.callCount).to.equal(1);
       });
     });
   });