65 lines
2.3 KiB
JavaScript
65 lines
2.3 KiB
JavaScript
|
'use strict';
|
||
|
|
/**
|
||
|
|
* @license Angular v<unknown>
|
||
|
|
* (c) 2010-2022 Google LLC. https://angular.io/
|
||
|
|
* License: MIT
|
||
|
|
*/
|
||
|
|
/**
|
||
|
|
* Promise for async/fakeAsync zoneSpec test
|
||
|
|
* can support async operation which not supported by zone.js
|
||
|
|
* such as
|
||
|
|
* it ('test jsonp in AsyncZone', async() => {
|
||
|
|
* new Promise(res => {
|
||
|
|
* jsonp(url, (data) => {
|
||
|
|
* // success callback
|
||
|
|
* res(data);
|
||
|
|
* });
|
||
|
|
* }).then((jsonpResult) => {
|
||
|
|
* // get jsonp result.
|
||
|
|
*
|
||
|
|
* // user will expect AsyncZoneSpec wait for
|
||
|
|
* // then, but because jsonp is not zone aware
|
||
|
|
* // AsyncZone will finish before then is called.
|
||
|
|
* });
|
||
|
|
* });
|
||
|
|
*/
|
||
|
|
Zone.__load_patch('promisefortest', (global, Zone, api) => {
|
||
|
|
const symbolState = api.symbol('state');
|
||
|
|
const UNRESOLVED = null;
|
||
|
|
const symbolParentUnresolved = api.symbol('parentUnresolved');
|
||
|
|
// patch Promise.prototype.then to keep an internal
|
||
|
|
// number for tracking unresolved chained promise
|
||
|
|
// we will decrease this number when the parent promise
|
||
|
|
// being resolved/rejected and chained promise was
|
||
|
|
// scheduled as a microTask.
|
||
|
|
// so we can know such kind of chained promise still
|
||
|
|
// not resolved in AsyncTestZone
|
||
|
|
Promise[api.symbol('patchPromiseForTest')] = function patchPromiseForTest() {
|
||
|
|
let oriThen = Promise[Zone.__symbol__('ZonePromiseThen')];
|
||
|
|
if (oriThen) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
oriThen = Promise[Zone.__symbol__('ZonePromiseThen')] = Promise.prototype.then;
|
||
|
|
Promise.prototype.then = function () {
|
||
|
|
const chained = oriThen.apply(this, arguments);
|
||
|
|
if (this[symbolState] === UNRESOLVED) {
|
||
|
|
// parent promise is unresolved.
|
||
|
|
const asyncTestZoneSpec = Zone.current.get('AsyncTestZoneSpec');
|
||
|
|
if (asyncTestZoneSpec) {
|
||
|
|
asyncTestZoneSpec.unresolvedChainedPromiseCount++;
|
||
|
|
chained[symbolParentUnresolved] = true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return chained;
|
||
|
|
};
|
||
|
|
};
|
||
|
|
Promise[api.symbol('unPatchPromiseForTest')] = function unpatchPromiseForTest() {
|
||
|
|
// restore origin then
|
||
|
|
const oriThen = Promise[Zone.__symbol__('ZonePromiseThen')];
|
||
|
|
if (oriThen) {
|
||
|
|
Promise.prototype.then = oriThen;
|
||
|
|
Promise[Zone.__symbol__('ZonePromiseThen')] = undefined;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
});
|