angular-cn/packages/zone.js/test/node/process.spec.ts

/**
 * @license
 * Copyright Google LLC All Rights Reserved.
 *
 * Use of this source code is governed by an MIT-style license that can be
 * found in the LICENSE file at https://angular.io/license
 */

import {zoneSymbol} from '../../lib/common/utils';

describe('process related test', () => {
  let zoneA: Zone, result: any[];
  beforeEach(() => {
    zoneA = Zone.current.fork({name: 'zoneA'});
    result = [];
  });
  it('process.nextTick callback should in zone', (done) => {
    zoneA.run(function() {
      process.nextTick(() => {
        expect(Zone.current.name).toEqual('zoneA');
        done();
      });
    });
  });
  it('process.nextTick should be executed before macroTask and promise', (done) => {
    zoneA.run(function() {
      setTimeout(() => {
        result.push('timeout');
      }, 0);
      process.nextTick(() => {
        result.push('tick');
      });
      setTimeout(() => {
        expect(result).toEqual(['tick', 'timeout']);
        done();
      });
    });
  });
  it('process.nextTick should be treated as microTask', (done) => {
    let zoneTick = Zone.current.fork({
      name: 'zoneTick',
      onScheduleTask: (
          parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task):
          Task => {
            result.push({callback: 'scheduleTask', targetZone: targetZone.name, task: task.source});
            return parentZoneDelegate.scheduleTask(targetZone, task);
          },
      onInvokeTask:
          (parentZoneDelegate: ZoneDelegate, currentZone: Zone, targetZone: Zone, task: Task,
           applyThis?: any, applyArgs?: any): any => {
            result.push({callback: 'invokeTask', targetZone: targetZone.name, task: task.source});
            return parentZoneDelegate.invokeTask(targetZone, task, applyThis, applyArgs);
          }
    });
    zoneTick.run(() => {
      process.nextTick(() => {
        result.push('tick');
      });
    });
    setTimeout(() => {
      expect(result.length).toBe(3);
      expect(result[0]).toEqual(
          {callback: 'scheduleTask', targetZone: 'zoneTick', task: 'process.nextTick'});
      expect(result[1]).toEqual(
          {callback: 'invokeTask', targetZone: 'zoneTick', task: 'process.nextTick'});
      done();
    });
  });

  it('should support process.on(unhandledRejection)', function(done) {
    const hookSpy = jasmine.createSpy('hook');
    (Zone as any)[zoneSymbol('ignoreConsoleErrorUncaughtError')] = true;
    Zone.current.fork({name: 'promise'}).run(function() {
      const listener = function(reason: any, promise: any) {
        hookSpy(promise, reason.message);
        process.removeListener('unhandledRejection', listener);
      };
      process.on('unhandledRejection', listener);
      const p = new Promise((resolve, reject) => {
        throw new Error('promise error');
      });

      setTimeout(function() {
        expect(hookSpy).toHaveBeenCalledWith(p, 'promise error');
        done();
      }, 10);
    });
  });

  it('should support process.on(rejectionHandled)', function(done) {
    (Zone as any)[zoneSymbol('ignoreConsoleErrorUncaughtError')] = true;
    Zone.current.fork({name: 'promise'}).run(function() {
      const listener = function(promise: any) {
        expect(promise).toEqual(p);
        process.removeListener('rejectionHandled', listener);
        done();
      };
      process.on('rejectionHandled', listener);
      const p = new Promise((resolve, reject) => {
        throw new Error('promise error');
      });

      setTimeout(function() {
        p.catch(reason => {});
      }, 10);
    });
  });

  it('should support multiple process.on(unhandledRejection)', function(done) {
    const hookSpy = jasmine.createSpy('hook');
    (Zone as any)[zoneSymbol('ignoreConsoleErrorUncaughtError')] = true;
    Zone.current.fork({name: 'promise'}).run(function() {
      const listener1 = function(reason: any, promise: any) {
        hookSpy(promise, reason.message);
        process.removeListener('unhandledRejection', listener1);
      };
      const listener2 = function(reason: any, promise: any) {
        hookSpy(promise, reason.message);
        process.removeListener('unhandledRejection', listener2);
      };
      process.on('unhandledRejection', listener1);
      process.on('unhandledRejection', listener2);
      const p = new Promise((resolve, reject) => {
        throw new Error('promise error');
      });

      setTimeout(function() {
        expect(hookSpy.calls.count()).toBe(2);
        expect(hookSpy.calls.allArgs()).toEqual([[p, 'promise error'], [p, 'promise error']]);
        done();
      }, 10);
    });
  });
});