Function patchFnWithModifiers

packages/zone.js/lib/vitest/vitest.ts:110–149

Patch functions with modifiers (i.e. `describe`/`it`).

(methodName: string, wrapFn: (fn: Function) => Function)

    /** Patch functions with modifiers (i.e. `describe`/`it`). */
    function patchFnWithModifiers(methodName: string, wrapFn: (fn: Function) => Function) {
      const originalVitestFn: Function & Record<TEST_MODIFIER_NAME, Function> = context[methodName];
      // Skip if already patched
      if (context[Zone.__symbol__(methodName)]) {
        return;
      }
      context[Zone.__symbol__(methodName)] = originalVitestFn;

      // Patching the main function
      context[methodName] = function (this: unknown, ...args: [unknown, Function, ...unknown[]]) {
        args[1] = wrapFn(args[1]);
        return originalVitestFn.apply(this, args);
      };

      // Patching direct modifier calls
      for (const modifierName of DIRECT_MODIFIER_NAMES) {
        context[methodName][modifierName] = function (
          this: unknown,
          ...args: [unknown, Function, ...unknown[]]
        ) {
          args[1] = wrapFn(args[1]);
          return originalVitestFn[modifierName].apply(this, args);
        };
      }

      // Patching curried modifier calls
      for (const modifierName of CURRIED_MODIFIER_NAMES) {
        context[methodName][modifierName] = function (this: unknown, ...modifierArgs: unknown[]) {
          // Since we are patching a curried function, we need
          // to pass the original context first (`originalVitestFn`).
          // Else, the chaining won't be possible (will get an error).
          const originalFn = originalVitestFn[modifierName].apply(originalVitestFn, modifierArgs);

          return function (this: unknown, ...args: [unknown, Function, ...unknown[]]) {
            args[1] = wrapFn(args[1]);
            return originalFn.apply(this, args);
          };
        };
      }
    }

    ['suite', 'describe'].forEach((methodName) =>
      patchFnWithModifiers(methodName, wrapDescribeInZone),