| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778 |
- 'use strict';
- // Try getting the real promise object from the context, if available. Someone
- // could have overridden it in a test.
- const Promise = global[Symbol.for('jest-native-promise')] || global.Promise;
- class CancelError extends Error {
- constructor() {
- super('Promise was canceled');
- this.name = 'CancelError';
- }
- }
- class PCancelable {
- static fn(fn) {
- return function () {
- const args = [].slice.apply(arguments);
- return new PCancelable((onCancel, resolve, reject) => {
- args.unshift(onCancel);
- fn.apply(null, args).then(resolve, reject);
- });
- };
- }
- constructor(executor) {
- this._pending = true;
- this._canceled = false;
- this._promise = new Promise((resolve, reject) => {
- this._reject = reject;
- return executor(fn => {
- this._cancel = fn;
- }, val => {
- this._pending = false;
- resolve(val);
- }, err => {
- this._pending = false;
- reject(err);
- });
- });
- }
- then() {
- return this._promise.then.apply(this._promise, arguments);
- }
- catch() {
- return this._promise.catch.apply(this._promise, arguments);
- }
- cancel() {
- if (!this._pending || this._canceled) {
- return;
- }
- if (typeof this._cancel === 'function') {
- try {
- this._cancel();
- } catch (err) {
- this._reject(err);
- }
- }
- this._canceled = true;
- this._reject(new CancelError());
- }
- get canceled() {
- return this._canceled;
- }
- }
- Object.setPrototypeOf(PCancelable.prototype, Promise.prototype);
- module.exports = PCancelable;
- module.exports.CancelError = CancelError;
|