@shopify/jest-dom-mocks/dist/animation-frame.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var AnimationFrame = /** @class */ (function () {
    function AnimationFrame() {
        var _this = this;
        this.isUsingMockAnimationFrame = false;
        this.queued = {};
        this.currentAnimationFrame = 0;
        this.requestAnimationFrame = function (callback) {
            _this.currentAnimationFrame += 1;
            _this.queued[_this.currentAnimationFrame] = callback;
            return _this.currentAnimationFrame;
        };
        this.cancelAnimationFrame = function (frame) {
            delete _this.queued[frame];
        };
    }
    AnimationFrame.prototype.mock = function () {
        if (this.isUsingMockAnimationFrame) {
            throw new Error('The animation frame is already mocked, but you tried to mock it again.');
        }
        this.isUsingMockAnimationFrame = true;
        this.originalRequestAnimationFrame = window.requestAnimationFrame;
        window.requestAnimationFrame = this.requestAnimationFrame;
        this.originalCancelAnimationFrame = window.cancelAnimationFrame;
        window.cancelAnimationFrame = this.cancelAnimationFrame;
    };
    AnimationFrame.prototype.restore = function () {
        if (!this.isUsingMockAnimationFrame) {
            throw new Error('The animation frame is already real, but you tried to restore it again.');
        }
        this.isUsingMockAnimationFrame = false;
        window.requestAnimationFrame = this.originalRequestAnimationFrame;
        window.cancelAnimationFrame = this.originalCancelAnimationFrame;
    };
    AnimationFrame.prototype.isMocked = function () {
        return this.isUsingMockAnimationFrame;
    };
    AnimationFrame.prototype.runFrame = function () {
        var _this = this;
        this.ensureAnimationFrameIsMock();
        // We need to do it this way so that frames that queue other frames
        // don't get removed
        Object.keys(this.queued).forEach(function (frame) {
            var callback = _this.queued[frame];
            delete _this.queued[frame];
            callback(Date.now());
        });
    };
    AnimationFrame.prototype.ensureAnimationFrameIsMock = function () {
        if (!this.isUsingMockAnimationFrame) {
            throw new Error('You must call animationFrame.mock() before interacting with the mock request- or cancel- AnimationFrame methods.');
        }
    };
    return AnimationFrame;
}());
exports.default = AnimationFrame;