Revert change

mui · Jan 30, 2025 · ee62e4d · ee62e4d
1 parent 6dc1330
commit ee62e4d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/x-data-grid-premium/src/tests/dataSourceAggregation.DataGridPremium.test.tsx b/packages/x-data-grid-premium/src/tests/dataSourceAggregation.DataGridPremium.test.tsx
@@ -157,7 +157,7 @@ describe('<DataGridPremium /> - Data source aggregation', () => {
     expect(footerRow?.id).to.deep.equal({ position: 'footer', value: 10 });
   });
 
-  it('should derive the aggregation values using `dataSource.getAggregatedValue`', () => {
+  it('should derive the aggregation values using `dataSource.getAggregatedValue`', async () => {
     const getAggregatedValue = () => 'Agg value';
     render(
       <TestDataSourceAggregation
@@ -167,7 +167,9 @@ describe('<DataGridPremium /> - Data source aggregation', () => {
         getAggregatedValue={getAggregatedValue}
       />,
     );
-    expect(Object.keys(apiRef.current!.state.aggregation.lookup).length).to.be.greaterThan(0);
+    await waitFor(() => {
+      expect(Object.keys(apiRef.current!.state.aggregation.lookup).length).to.be.greaterThan(0);
+    });
     expect(apiRef.current?.state.aggregation.lookup[GRID_ROOT_GROUP_ID].id.value).to.equal(
       'Agg value',
     );