@shopify/jest-dom-mocks/dist/promise.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var tslib_1 = require("tslib");
var promise_1 = tslib_1.__importDefault(require("promise"));
var Promise = /** @class */ (function () {
    function Promise() {
        this.originalPromise = global.Promise;
        this.isUsingFakePromise = false;
    }
    Promise.prototype.mock = function () {
        if (this.isUsingFakePromise) {
            throw new Error('Promise is already mocked, but you tried to mock it again.');
        }
        jest.useFakeTimers();
        global.Promise = promise_1.default;
        this.isUsingFakePromise = true;
    };
    Promise.prototype.restore = function () {
        if (!this.isUsingFakePromise) {
            throw new Error('Promise is already real, but you tried to restore it again.');
        }
        jest.useRealTimers();
        global.Promise = this.originalPromise;
        this.isUsingFakePromise = false;
    };
    Promise.prototype.isMocked = function () {
        return this.isUsingFakePromise;
    };
    Promise.prototype.runPending = function () {
        this.ensureUsingFakePromise();
        jest.runAllTimers();
    };
    Promise.prototype.ensureUsingFakePromise = function () {
        if (!this.isUsingFakePromise) {
            throw new Error('You must call Promise.mock() before interacting with the mock Promise.');
        }
    };
    return Promise;
}());
exports.default = Promise;