使用jest覆盖回调函数内部的函数

// 假设我们有一个回调函数，内部调用了一个函数add，我们要覆盖这个函数
function callback(callbackFn) {
  const result = callbackFn(2, 3);
  console.log(result);
}

function add(a, b) {
  return a + b;
}

// 使用jest.fn()创建一个模拟函数来替代add函数
const mockAdd = jest.fn();

// 使用jest.mock()模拟回调函数内部的函数
jest.mock('./path/to/callback', () => {
  return {
    add: mockAdd
  };
});

test('测试回调函数', () => {
  // 设置模拟函数的返回值
  mockAdd.mockReturnValue(5);

  // 调用回调函数
  callback(add);

  // 断言模拟函数被调用，并且返回值正确
  expect(mockAdd).toHaveBeenCalledWith(2, 3);
  expect(console.log).toHaveBeenCalledWith(5);
});