Jest 测试中处理模块内部函数间接调用的 Mock 策略

// app.js (或类似模块)
var sendToEH = function sendToEH() { /* ... */ };
var sendDataHandler = function sendDataHandler() {
  sendToEH(); // 内部调用 sendToEH
};
// 假设这是通过某种方式暴露的，例如作为 app 对象的方法
// app.sendToEH = sendToEH;
// app.sendDataHandler = sendDataHandler;

// test.js
app.sendToEH = jest.fn(); // Mock app 对象上的 sendToEH

await app.sendDataHandler(req, res, next);

expect(app.sendToEH).toHaveBeenCalled(); // 期望失败

app.sendToEH = jest.fn();

await app.sendToEH('asdf'); // 直接调用 Mock 后的 sendToEH

await app.sendDataHandler(req, res, next);

expect(app.sendToEH).toHaveBeenCalled(); // 此时测试通过

// 假设这是你的业务逻辑模块文件 (e.g., myModule.js)

// 定义内部函数
var sendToEH = function sendToEH() {
  console.log("Original sendToEH called.");
  // 实际的数据发送逻辑
};

// 定义处理函数，它将通过 exportFunctions 对象调用 sendToEH
var sendDataHandler = function sendDataHandler() {
  console.log("sendDataHandler called, invoking sendToEH via exportFunctions.");
  exportFunctions.sendToEH(); // 通过对象引用调用 sendToEH
};

// 将所有相关函数封装在一个对象中并导出
const exportFunctions = {
  sendToEH,
  sendDataHandler
};

export default exportFunctions; // 导出这个对象

// 假设你的测试文件 (e.g., myModule.test.js)
import app from './myModule'; // 导入导出的对象，这里假设导入后命名为 app

describe('sendDataHandler', () => {
  beforeEach(() => {
    // 在每次测试前重置 Mock
    // 注意：这里我们直接 Mock app.sendToEH，因为 app 就是 exportFunctions 对象
    app.sendToEH = jest.fn();
  });

  test('should call sendToEH when sendDataHandler is invoked', async () => {
    const req = {}; // 模拟请求对象
    const res = {}; // 模拟响应对象
    const next = jest.fn(); // 模拟 next 函数

    // 调用 sendDataHandler
    await app.sendDataHandler(req, res, next);

    // 验证 app.sendToEH 是否被调用
    expect(app.sendToEH).toHaveBeenCalledTimes(1);
    // 还可以验证调用参数等
    // expect(app.sendToEH).toHaveBeenCalledWith(/* 期望的参数 */);
  });

  test('sendToEH should not be called if sendDataHandler is not invoked', async () => {
    expect(app.sendToEH).not.toHaveBeenCalled();
  });
});