Jest模块化测试：解决Mock函数引用传递失效的挑战

// 假设 app 是导入的模块
// app.sendToEH 是 sendDataHandler 内部会调用的函数

app.sendToEH = jest.fn(); // 尝试 Mock app 上的 sendToEH

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

expect(app.sendToEH).toHaveBeenCalled(); // 期望 sendToEH 被调用

app.sendToEH = jest.fn();

await app.sendToEH('some data'); // 直接调用 Mock 后的函数，会成功记录调用

await app.sendDataHandler(req, res, next); // sendDataHandler 仍然调用原始函数

expect(app.sendToEH).toHaveBeenCalled(); // 此时测试通过，但记录的是我们手动调用的一次，而非 sendDataHandler 内部的调用

// module.js
// 原始的 sendToEH 函数实现
var sendToEH = function sendToEH() {
  console.log('Original sendToEH called');
  // ... sendToEH 的具体逻辑 ...
};

// sendDataHandler 函数，现在通过 exportFunctions 对象调用 sendToEH
var sendDataHandler = function sendDataHandler(req, res, next) {
  console.log('sendDataHandler calling exportFunctions.sendToEH');
  // 关键：通过共享的 exportFunctions 对象来调用 sendToEH
  exportFunctions.sendToEH();
  // ... sendDataHandler 的其他逻辑 ...
};

// 导出这些函数的对象
const exportFunctions = {
  sendToEH,
  sendDataHandler
};

export default exportFunctions;

// test.js
import app from './module'; // 导入你的模块

describe('sendDataHandler 功能测试', () => {
  let originalSendToEH; // 用于在测试后恢复原始函数

  beforeEach(() => {
    // 在每个测试用例之前，保存原始引用并Mock sendToEH
    originalSendToEH = app.exportFunctions.sendToEH;
    app.exportFunctions.sendToEH = jest.fn();
  });

  afterEach(() => {
    // 在每个测试用例之后，恢复原始函数以避免测试间的副作用
    app.exportFunctions.sendToEH = originalSendToEH;
    jest.clearAllMocks(); // 清除所有 Mock 的调用记录
  });

  test('当 sendDataHandler 被调用时，应该调用 sendToEH', async () => {
    // 模拟请求、响应和 next 函数
    const req = { /* 模拟请求数据 */ };
    const res = { /* 模拟响应数据 */ };
    const next = jest.fn();

    // 调用被测函数
    await app.exportFunctions.sendDataHandler(req, res, next);

    // 断言 Mock 函数是否被调用
    expect(app.exportFunctions.sendToEH).toHaveBeenCalledTimes(1);
    // 如果 sendToEH 接收参数，你也可以断言参数
    // expect(app.exportFunctions.sendToEH).toHaveBeenCalledWith(expectedArgs);
  });
});