redux-form/es/__tests__/addExpectations.js

A higher order component decorator for forms using Redux and React

import expect from 'expect';

/**
 * Takes expectations and extends expect with them. Cannot use expect.extends due to the
 * asynchronous nature of the tests.
 * @param expectations Expectations to add
 */
var addExpectations = function addExpectations(expectations) {
  var decorate = function decorate(dest) {
    var wrap = function wrap(value, key) {
      if (typeof value === 'function' && key !== 'actual') {
        dest[key] = function () {
          for (var _len = arguments.length, params = Array(_len), _key = 0; _key < _len; _key++) {
            params[_key] = arguments[_key];
          }

          return decorate(value.apply(dest, params));
        };
      }
    };
    for (var key in dest) {
      wrap(dest[key], key);
    }
    for (var _key2 in expectations) {
      wrap(expectations[_key2], _key2);
    }
    return dest;
  };
  return function () {
    return decorate(expect.apply(undefined, arguments));
  };
};

export default addExpectations;