@testing-library/react-native
Version:
Simple and complete React Native testing utilities that encourage good testing practices.
179 lines (171 loc) • 6.7 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.default = waitFor;
var _config = require("./config");
var _flushMicroTasks = require("./flush-micro-tasks");
var _errors = require("./helpers/errors");
var _timers = require("./helpers/timers");
var _wrapAsync = require("./helpers/wrap-async");
/* globals jest */
const DEFAULT_INTERVAL = 50;
function waitForInternal(expectation, {
timeout = (0, _config.getConfig)().asyncUtilTimeout,
interval = DEFAULT_INTERVAL,
stackTraceError,
onTimeout
}) {
if (typeof expectation !== 'function') {
throw new TypeError('Received `expectation` arg must be a function');
}
// eslint-disable-next-line no-async-promise-executor
return new Promise(async (resolve, reject) => {
let lastError, intervalId;
let finished = false;
let promiseStatus = 'idle';
let overallTimeoutTimer = null;
const usingFakeTimers = (0, _timers.jestFakeTimersAreEnabled)();
if (usingFakeTimers) {
checkExpectation();
// this is a dangerous rule to disable because it could lead to an
// infinite loop. However, eslint isn't smart enough to know that we're
// setting finished inside `onDone` which will be called when we're done
// waiting or when we've timed out.
let fakeTimeRemaining = timeout;
while (!finished) {
if (!(0, _timers.jestFakeTimersAreEnabled)()) {
const error = new Error(`Changed from using fake timers to real timers while using waitFor. This is not allowed and will result in very strange behavior. Please ensure you're awaiting all async things your test is doing before changing to real timers. For more info, please go to https://github.com/testing-library/dom-testing-library/issues/830`);
if (stackTraceError) {
(0, _errors.copyStackTrace)(error, stackTraceError);
}
reject(error);
return;
}
// when fake timers are used we want to simulate the interval time passing
if (fakeTimeRemaining <= 0) {
handleTimeout();
return;
} else {
fakeTimeRemaining -= interval;
}
// we *could* (maybe should?) use `advanceTimersToNextTimer` but it's
// possible that could make this loop go on forever if someone is using
// third party code that's setting up recursive timers so rapidly that
// the user's timer's don't get a chance to resolve. So we'll advance
// by an interval instead. (We have a test for this case).
jest.advanceTimersByTime(interval);
// It's really important that checkExpectation is run *before* we flush
// in-flight promises. To be honest, I'm not sure why, and I can't quite
// think of a way to reproduce the problem in a test, but I spent
// an entire day banging my head against a wall on this.
checkExpectation();
// In this rare case, we *need* to wait for in-flight promises
// to resolve before continuing. We don't need to take advantage
// of parallelization so we're fine.
// https://stackoverflow.com/a/59243586/971592
await (0, _flushMicroTasks.flushMicroTasks)();
}
} else {
overallTimeoutTimer = (0, _timers.setTimeout)(handleTimeout, timeout);
intervalId = setInterval(checkRealTimersCallback, interval);
checkExpectation();
}
function onDone(done) {
finished = true;
if (overallTimeoutTimer) {
(0, _timers.clearTimeout)(overallTimeoutTimer);
}
if (!usingFakeTimers) {
clearInterval(intervalId);
}
if (done.type === 'error') {
reject(done.error);
} else {
resolve(done.result);
}
}
function checkRealTimersCallback() {
if ((0, _timers.jestFakeTimersAreEnabled)()) {
const error = new Error(`Changed from using real timers to fake timers while using waitFor. This is not allowed and will result in very strange behavior. Please ensure you're awaiting all async things your test is doing before changing to fake timers. For more info, please go to https://github.com/testing-library/dom-testing-library/issues/830`);
if (stackTraceError) {
(0, _errors.copyStackTrace)(error, stackTraceError);
}
return reject(error);
} else {
return checkExpectation();
}
}
function checkExpectation() {
if (promiseStatus === 'pending') return;
try {
const result = expectation();
// @ts-expect-error result can be a promise
if (typeof result?.then === 'function') {
const promiseResult = result;
promiseStatus = 'pending';
// eslint-disable-next-line promise/catch-or-return, promise/prefer-await-to-then
promiseResult.then(resolvedValue => {
promiseStatus = 'resolved';
onDone({
type: 'result',
result: resolvedValue
});
return;
}, rejectedValue => {
promiseStatus = 'rejected';
lastError = rejectedValue;
return;
});
} else {
onDone({
type: 'result',
result: result
});
}
// If `callback` throws, wait for the next mutation, interval, or timeout.
} catch (error) {
// Save the most recent callback error to reject the promise with it in the event of a timeout
lastError = error;
}
}
function handleTimeout() {
let error;
if (lastError) {
if (lastError instanceof Error) {
error = lastError;
} else {
error = new Error(String(lastError));
}
if (stackTraceError) {
(0, _errors.copyStackTrace)(error, stackTraceError);
}
} else {
error = new Error('Timed out in waitFor.');
if (stackTraceError) {
(0, _errors.copyStackTrace)(error, stackTraceError);
}
}
if (typeof onTimeout === 'function') {
const result = onTimeout(error);
if (result) {
error = result;
}
}
onDone({
type: 'error',
error
});
}
});
}
function waitFor(expectation, options) {
// Being able to display a useful stack trace requires generating it before doing anything async
const stackTraceError = new _errors.ErrorWithStack('STACK_TRACE_ERROR', waitFor);
const optionsWithStackTrace = {
stackTraceError,
...options
};
return (0, _wrapAsync.wrapAsync)(() => waitForInternal(expectation, optionsWithStackTrace));
}
//# sourceMappingURL=wait-for.js.map