TypeScript jest-each.bind函数代码示例

const describe = (() => {
  const describe = (blockName: BlockName, blockFn: BlockFn) =>
    _dispatchDescribe(blockFn, blockName, describe);
  const only = (blockName: BlockName, blockFn: BlockFn) =>
    _dispatchDescribe(blockFn, blockName, only, 'only');
  const skip = (blockName: BlockName, blockFn: BlockFn) =>
    _dispatchDescribe(blockFn, blockName, skip, 'skip');

  describe.each = bindEach(describe, false);

  only.each = bindEach(only, false);
  skip.each = bindEach(skip, false);

  describe.only = only;
  describe.skip = skip;

  return describe;
})();

export default (environment: JestEnvironment): void => {
  environment.global.it.each = bindEach(environment.global.it);
  environment.global.fit.each = bindEach(environment.global.fit);
  environment.global.xit.each = bindEach(environment.global.xit);
  environment.global.describe.each = bindEach(
    environment.global.describe,
    false,
  );
  environment.global.xdescribe.each = bindEach(
    environment.global.xdescribe,
    false,
  );
  environment.global.fdescribe.each = bindEach(
    environment.global.fdescribe,
    false,
  );
};

const test: Global.It = (() => {
  const test = (testName: TestName, fn: TestFn, timeout?: number): void =>
    _addTest(testName, undefined, fn, test, timeout);
  const skip = (testName: TestName, fn?: TestFn, timeout?: number): void =>
    _addTest(testName, 'skip', fn, skip, timeout);
  const only = (testName: TestName, fn: TestFn, timeout?: number): void =>
    _addTest(testName, 'only', fn, test.only, timeout);

  test.todo = (testName: TestName, ...rest: Array<any>): void => {
    if (rest.length > 0 || typeof testName !== 'string') {
      throw new ErrorWithStack(
        'Todo must be called with only a description.',
        test.todo,
      );
    }
    return _addTest(testName, 'todo', () => {}, test.todo);
  };

  const _addTest = (
    testName: TestName,
    mode: TestMode,
    fn: TestFn | undefined,
    testFn: (testName: TestName, fn: TestFn, timeout?: number) => void,
    timeout?: number,
  ) => {
    const asyncError = new ErrorWithStack(undefined, testFn);

    if (typeof testName !== 'string') {
      asyncError.message = `Invalid first argument, ${testName}. It must be a string.`;

      throw asyncError;
    }
    if (fn === undefined) {
      asyncError.message =
        'Missing second argument. It must be a callback function. Perhaps you want to use `test.todo` for a test placeholder.';

      throw asyncError;
    }
    if (typeof fn !== 'function') {
      asyncError.message = `Invalid second argument, ${fn}. It must be a callback function.`;

      throw asyncError;
    }

    return dispatch({
      asyncError,
      fn,
      mode,
      name: 'add_test',
      testName,
      timeout,
    });
  };

  test.each = bindEach(test);
  only.each = bindEach(only);
  skip.each = bindEach(skip);

  test.only = only;
  test.skip = skip;

  return test;
})();