80 lines
3.2 KiB
JavaScript
80 lines
3.2 KiB
JavaScript
|
"use strict";
|
||
|
|
|
||
|
|
Object.defineProperty(exports, "__esModule", {
|
||
|
|
value: true
|
||
|
|
});
|
||
|
|
exports.default = void 0;
|
||
|
|
exports.getIsReactActEnvironment = getIsReactActEnvironment;
|
||
|
|
exports.setReactActEnvironment = setIsReactActEnvironment;
|
||
|
|
var React = _interopRequireWildcard(require("react"));
|
||
|
|
var DeprecatedReactTestUtils = _interopRequireWildcard(require("react-dom/test-utils"));
|
||
|
|
function _interopRequireWildcard(e, t) { if ("function" == typeof WeakMap) var r = new WeakMap(), n = new WeakMap(); return (_interopRequireWildcard = function (e, t) { if (!t && e && e.__esModule) return e; var o, i, f = { __proto__: null, default: e }; if (null === e || "object" != typeof e && "function" != typeof e) return f; if (o = t ? n : r) { if (o.has(e)) return o.get(e); o.set(e, f); } for (const t in e) "default" !== t && {}.hasOwnProperty.call(e, t) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, t)) && (i.get || i.set) ? o(f, t, i) : f[t] = e[t]); return f; })(e, t); }
|
||
|
|
const reactAct = typeof React.act === 'function' ? React.act : DeprecatedReactTestUtils.act;
|
||
|
|
function getGlobalThis() {
|
||
|
|
/* istanbul ignore else */
|
||
|
|
if (typeof globalThis !== 'undefined') {
|
||
|
|
return globalThis;
|
||
|
|
}
|
||
|
|
/* istanbul ignore next */
|
||
|
|
if (typeof self !== 'undefined') {
|
||
|
|
return self;
|
||
|
|
}
|
||
|
|
/* istanbul ignore next */
|
||
|
|
if (typeof window !== 'undefined') {
|
||
|
|
return window;
|
||
|
|
}
|
||
|
|
/* istanbul ignore next */
|
||
|
|
if (typeof global !== 'undefined') {
|
||
|
|
return global;
|
||
|
|
}
|
||
|
|
/* istanbul ignore next */
|
||
|
|
throw new Error('unable to locate global object');
|
||
|
|
}
|
||
|
|
function setIsReactActEnvironment(isReactActEnvironment) {
|
||
|
|
getGlobalThis().IS_REACT_ACT_ENVIRONMENT = isReactActEnvironment;
|
||
|
|
}
|
||
|
|
function getIsReactActEnvironment() {
|
||
|
|
return getGlobalThis().IS_REACT_ACT_ENVIRONMENT;
|
||
|
|
}
|
||
|
|
function withGlobalActEnvironment(actImplementation) {
|
||
|
|
return callback => {
|
||
|
|
const previousActEnvironment = getIsReactActEnvironment();
|
||
|
|
setIsReactActEnvironment(true);
|
||
|
|
try {
|
||
|
|
// The return value of `act` is always a thenable.
|
||
|
|
let callbackNeedsToBeAwaited = false;
|
||
|
|
const actResult = actImplementation(() => {
|
||
|
|
const result = callback();
|
||
|
|
if (result !== null && typeof result === 'object' && typeof result.then === 'function') {
|
||
|
|
callbackNeedsToBeAwaited = true;
|
||
|
|
}
|
||
|
|
return result;
|
||
|
|
});
|
||
|
|
if (callbackNeedsToBeAwaited) {
|
||
|
|
const thenable = actResult;
|
||
|
|
return {
|
||
|
|
then: (resolve, reject) => {
|
||
|
|
thenable.then(returnValue => {
|
||
|
|
setIsReactActEnvironment(previousActEnvironment);
|
||
|
|
resolve(returnValue);
|
||
|
|
}, error => {
|
||
|
|
setIsReactActEnvironment(previousActEnvironment);
|
||
|
|
reject(error);
|
||
|
|
});
|
||
|
|
}
|
||
|
|
};
|
||
|
|
} else {
|
||
|
|
setIsReactActEnvironment(previousActEnvironment);
|
||
|
|
return actResult;
|
||
|
|
}
|
||
|
|
} catch (error) {
|
||
|
|
// Can't be a `finally {}` block since we don't know if we have to immediately restore IS_REACT_ACT_ENVIRONMENT
|
||
|
|
// or if we have to await the callback first.
|
||
|
|
setIsReactActEnvironment(previousActEnvironment);
|
||
|
|
throw error;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
const act = withGlobalActEnvironment(reactAct);
|
||
|
|
var _default = exports.default = act;
|
||
|
|
/* eslint no-console:0 */
|