如何使用 Jest 将 prop 中的 spyOn 方法传递给组件?

How to spyOn method inside prop passed down to a component using Jest?

背景:

我的测试框架是 Jest 和 Enzyme。我有一个名为 Lazyload 的组件,它使用 React.ContextAPI 耦合到 LazyloadProvider。我想编写一个测试来保证 Lazyload 组件内部 prop 方法 this.props.lazyload.add()componentDidMount 已被调用。使用 Jest spy 我希望 hasBeenCalledWith(this.lazyRef) 有效

我很高兴能够监视 Lazyload 的 register 方法;但是,我无法弄清楚如何监视内部道具方法 this.props.lazyload.add

问题:

如何在 this.props.lazyload.add 上编写一个有趣的间谍程序并确保它是通过 this.lazyRef 调用的?

class Lazyload extends Component<LazyloadProps, LazyloadState> {
  lazyRef: ReactRef;

  constructor(props) {
    super(props);
    this.lazyRef = createRef();
  }

  componentDidMount() {
   this.register()
  }

  register() { // not spy on this.
    this.props.lazyload.add(this.lazyRef); // spyOn this
  }
}

测试:

describe('lazyload', () => {
  let provider;
  beforeEach(() => {
    provider = shallow(
      <LazyloadProvider>
        <p>Wow</p>
      </LazyloadProvider>
    ).instance();
  });

  it('should register a lazyloader with add', () => {
    const spy = jest.spyOn(Lazyload.prototype, 'register');

    const wrapper = shallow(
      <Lazyload lazyload={provider.engine}>
        <p>doge</p>
      </Lazyload>
    ).instance();

    expect(spy).toHaveBeenCalled(); // this works however it's a better test to spy on the this.prop.lazyload.add method.. but how?
  });
})

你可以通过 stubbed add in lazyload prop, and check with toHaveBeenCalledWith matcher if it accepts instance()lazyref :

describe('lazyload', () => {

  it('should add ref', () => {
    const lazyloadStub = {
        add: jest.fn();
    };

    const wrapper = shallow(
      <Lazyload lazyload={lazyloadStub}>
        <p>doge</p>
      </Lazyload>
    );

    expect(lazyloadStub.add).toHaveBeenCalledWith(wrapper.instance().lazyRef); 
  });
})