Skip to content

Commit 2901c1e

Browse files
authored
Accept min and max delay in createSchemaFetch options (#11774)
1 parent 01b45a4 commit 2901c1e

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

src/testing/internal/profile/profile.tsx

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -151,6 +151,9 @@ export function createProfiler<Snapshot extends ValidSnapshot = void>({
151151
let nextRender: Promise<Render<Snapshot>> | undefined;
152152
let resolveNextRender: ((render: Render<Snapshot>) => void) | undefined;
153153
let rejectNextRender: ((error: unknown) => void) | undefined;
154+
function resetNextRender() {
155+
nextRender = resolveNextRender = rejectNextRender = undefined;
156+
}
154157
const snapshotRef = { current: initialSnapshot };
155158
const replaceSnapshot: ReplaceSnapshot<Snapshot> = (snap) => {
156159
if (typeof snap === "function") {
@@ -241,7 +244,7 @@ export function createProfiler<Snapshot extends ValidSnapshot = void>({
241244
});
242245
rejectNextRender?.(error);
243246
} finally {
244-
nextRender = resolveNextRender = rejectNextRender = undefined;
247+
resetNextRender();
245248
}
246249
};
247250

@@ -340,13 +343,12 @@ export function createProfiler<Snapshot extends ValidSnapshot = void>({
340343
rejectNextRender = reject;
341344
}),
342345
new Promise<Render<Snapshot>>((_, reject) =>
343-
setTimeout(
344-
() =>
345-
reject(
346-
applyStackTrace(new WaitForRenderTimeoutError(), stackTrace)
347-
),
348-
timeout
349-
)
346+
setTimeout(() => {
347+
reject(
348+
applyStackTrace(new WaitForRenderTimeoutError(), stackTrace)
349+
);
350+
resetNextRender();
351+
}, timeout)
350352
),
351353
]);
352354
}

0 commit comments

Comments
 (0)