[test] Fix flaky screenshots (#11391)

mui · Dec 12, 2023 · d7d5ef7 · d7d5ef7
1 parent 2f89700
commit d7d5ef7
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 7 deletions.
diff --git a/docs/data/data-grid/row-selection/ControlledSelectionServerPaginationGrid.js b/docs/data/data-grid/row-selection/ControlledSelectionServerPaginationGrid.js
@@ -1,12 +1,12 @@
 import * as React from 'react';
 import { DataGrid } from '@mui/x-data-grid';
-import { useDemoData } from '@mui/x-data-grid-generator';
+import { useDemoData, randomInt } from '@mui/x-data-grid-generator';
 
 function loadServerRows(page, data) {
   return new Promise((resolve) => {
     setTimeout(() => {
       resolve(data.rows.slice(page * 5, (page + 1) * 5));
-    }, Math.random() * 500 + 100); // simulate network latency
+    }, randomInt(100, 600)); // simulate network latency
   });
 }
 

diff --git a/docs/data/data-grid/row-selection/ControlledSelectionServerPaginationGrid.tsx b/docs/data/data-grid/row-selection/ControlledSelectionServerPaginationGrid.tsx
@@ -1,12 +1,12 @@
 import * as React from 'react';
 import { DataGrid, GridRowsProp, GridRowSelectionModel } from '@mui/x-data-grid';
-import { GridDemoData, useDemoData } from '@mui/x-data-grid-generator';
+import { GridDemoData, useDemoData, randomInt } from '@mui/x-data-grid-generator';
 
 function loadServerRows(page: number, data: GridDemoData): Promise<any> {
   return new Promise((resolve) => {
     setTimeout(() => {
       resolve(data.rows.slice(page * 5, (page + 1) * 5));
-    }, Math.random() * 500 + 100); // simulate network latency
+    }, randomInt(100, 600)); // simulate network latency
   });
 }
 

diff --git a/docs/data/data-grid/row-updates/InfiniteLoadingGrid.js b/docs/data/data-grid/row-updates/InfiniteLoadingGrid.js
@@ -4,6 +4,7 @@ import {
   useDemoData,
   getRealGridData,
   getCommodityColumns,
+  randomInt,
 } from '@mui/x-data-grid-generator';
 import LinearProgress from '@mui/material/LinearProgress';
 
@@ -31,7 +32,7 @@ export default function InfiniteLoadingGrid() {
     setLoading(true);
     const newData = await getRealGridData(newRowLength, getCommodityColumns());
     // Simulate network throttle
-    await sleep(Math.random() * 500 + 100);
+    await sleep(randomInt(100, 600));
 
     if (mounted.current) {
       setLoading(false);

diff --git a/docs/data/data-grid/row-updates/InfiniteLoadingGrid.tsx b/docs/data/data-grid/row-updates/InfiniteLoadingGrid.tsx
@@ -4,6 +4,7 @@ import {
   useDemoData,
   getRealGridData,
   getCommodityColumns,
+  randomInt,
 } from '@mui/x-data-grid-generator';
 import LinearProgress from '@mui/material/LinearProgress';
 
@@ -31,7 +32,7 @@ export default function InfiniteLoadingGrid() {
     setLoading(true);
     const newData = await getRealGridData(newRowLength, getCommodityColumns());
     // Simulate network throttle
-    await sleep(Math.random() * 500 + 100);
+    await sleep(randomInt(100, 600));
 
     if (mounted.current) {
       setLoading(false);

diff --git a/packages/grid/x-data-grid-generator/src/hooks/useQuery.ts b/packages/grid/x-data-grid-generator/src/hooks/useQuery.ts
@@ -16,6 +16,7 @@ import {
   getColumnsFromOptions,
   getInitialState,
 } from './useDemoData';
+import { randomInt } from '../services/random-generator';
 
 const simplifiedValueGetter = (field: string, colDef: GridColDef) => (row: GridRowModel) => {
   const params = { id: row.id, row, field, rowNode: {} };
@@ -118,7 +119,7 @@ export const loadServerRows = (
   if (maxDelay < minDelay) {
     throw new Error('serverOptions.minDelay is larger than serverOptions.maxDelay ');
   }
-  const delay = Math.random() * (maxDelay - minDelay) + minDelay;
+  const delay = randomInt(minDelay, maxDelay);
 
   const { cursor, page = 0, pageSize } = queryOptions;