如何使用回调测试React ref？

import React, { Component } from 'react';

class MyComponent extends Component {
  constructor(props) {
    super(props);
    this.myRef = React.createRef();
  }

  componentDidMount() {
    this.props.callbackRef(this.myRef);
  }

  render() {
    return <div ref={this.myRef}>Hello World</div>;
  }
}

export default MyComponent;

import React from 'react';
import { mount } from 'enzyme';
import MyComponent from './MyComponent';

describe('MyComponent', () => {
  it('should call the callback function with the ref object', () => {
    const callbackRef = jest.fn();
    const wrapper = mount(<MyComponent callbackRef={callbackRef} />);
    const refObject = wrapper.find('div').instance().myRef;

    expect(callbackRef).toHaveBeenCalledWith(refObject);
  });
});