Function patchAsyncTest

packages/zone.js/lib/zone-spec/async-test.ts:226–335

(Zone: ZoneType)

}

export function patchAsyncTest(Zone: ZoneType): void {
  // Export the class so that new instances can be created with proper
  // constructor params.
  (Zone as any)['AsyncTestZoneSpec'] = AsyncTestZoneSpec;

  Zone.__load_patch('asynctest', (global: any, Zone: ZoneType, api: _ZonePrivate) => {
    /**
     * Wraps a test function in an asynchronous test zone. The test will automatically
     * complete when all asynchronous calls within this zone are done.
     */
    (Zone as any)[api.symbol('asyncTest')] = function asyncTest(fn: Function): (done: any) => any {
      // If we're running using the Jasmine test framework, adapt to call the 'done'
      // function when asynchronous activity is finished.
      if (global.jasmine) {
        // Not using an arrow function to preserve context passed from call site
        return function (this: unknown, done: any) {
          if (!done) {
            // if we run beforeEach in @angular/core/testing/testing_internal then we get no done
            // fake it here and assume sync.
            done = function () {};
            done.fail = function (e: any) {
              throw e;
            };
          }
          runInTestZone(fn, this, undefined, done, (err: any) => {
            if (typeof err === 'string') {
              return done.fail(new Error(err));
            } else {
              done.fail(err);
            }
          });
        };
      }
      // Otherwise, return a promise which will resolve when asynchronous activity
      // is finished. This will be correctly consumed by the Mocha framework with
      // it('...', async(myFn)); or can be used in a custom framework.
      // Not using an arrow function to preserve context passed from call site
      return function (this: unknown, ...args: unknown[]) {
        return new Promise<void>((finishCallback, failCallback) => {
          runInTestZone(fn, this, args, finishCallback, failCallback);
        });
      };
    };

    function runInTestZone(
      fn: Function,
      context: any,
      applyArgs: unknown[] | undefined,
      finishCallback: Function,
      failCallback: Function,
    ) {
      const currentZone = Zone.current;
      const AsyncTestZoneSpec = (Zone as any)['AsyncTestZoneSpec'];
      if (AsyncTestZoneSpec === undefined) {
        throw new Error(
          'AsyncTestZoneSpec is needed for the async() test helper but could not be found. ' +
            'Please make sure that your environment includes zone.js/plugins/async-test',
        );