如何模拟外部模块方法？

Question

我有一个带有函数的模块：

const utils = {
  redirectTo(url) {
    if (url) {
      window.location = url;
    }
  },
};

export default utils;

它在 React 组件中的某个地方使用，如下所示：

import utils from '../../lib/utils';

componentWillUpdate() {
  this.redirectTo('foo')
}

现在我想检查redirectTo被调用的值是否等于foo 。

  it('should redirect if no rights', () => {
    const mockRedirectFn = jest.fn();
    utils.redirectTo = mockRedirectFn;

    mount(
      <SomeComponent />,
    );

    expect(mockRedirectFn).toBeCalled();
    expect(mockRedirectFn).toBeCalledWith('foo');
    console.log(mockRedirectFn.mock);
    // { calls: [], instances: [] }
  });

这就是我所拥有的，但它不起作用。 我该怎么做呢？

Answer 1

你必须像这样模拟lib/utils模块：

import utils from '../../lib/utils';
jest.mock('../../lib/utils', () => ({
  redirect: jest.fn()
}))

it('should redirect if no rights', () => {
  mount(
    <SomeComponent />,
  );
  expect(utils.redirect).toHaveBeenCalledWith();
});

这将用只返回{redirect:jest.fn()}的模拟替换模块。 这个模块也被导入到你的测试中，然后你可以访问间谍进行redirect并测试它是否使用正确的参数调用。

Answer 2

这是我最终使用的：

 it('should redirect if no rights', () => {

    // this way we mock only one method: redirectTo
    jest.mock('lib/utils', () => {
      const original = require.requireActual('lib/utils');
      original.default.redirectTo = jest.fn();
      return original;
    });

    const redirectTo = require.requireMock('lib/utils').default.redirectTo;

    mount(
      <SomeComponent />,
    );

    expect(redirectTo).toHaveBeenCalled();
  });