Jest 测试中模块内函数调用的 Mock 策略：解决引用传递问题

// 假设 app.js 模块内部结构类似
// var sendToEH = function() { /* 原始实现 */ };
// var sendDataHandler = function() { sendToEH(); }; // 内部调用

// 测试文件中的尝试
app.sendToEH = jest.fn(); // 模拟 app 对象上的 sendToEH
await app.sendDataHandler(req, res, next); // 调用 sendDataHandler
expect(app.sendToEH).toHaveBeenCalled(); // 期望模拟函数被调用

// 模块文件 (例如：app.js)
var sendToEH = function sendToEH() {
  console.log('Original sendToEH called');
  // ... 实际的 sendToEH 逻辑
};

var sendDataHandler = function sendDataHandler() {
  console.log('sendDataHandler called');
  // 内部通过 exportFunctions 对象访问 sendToEH
  exportFunctions.sendToEH();
};

const exportFunctions = {
  sendToEH,
  sendDataHandler
};

export default exportFunctions; // 导出包含所有函数的对象

// 测试文件 (例如：app.test.js)
import app from './app'; // 导入包含 exportFunctions 的模块

describe('sendDataHandler functionality', () => {
  let originalSendToEH;

  beforeEach(() => {
    // 备份原始函数，以防需要恢复或避免影响其他测试
    originalSendToEH = app.sendToEH;
    // 模拟 app.sendToEH，现在 sendDataHandler 内部会调用到这个模拟
    app.sendToEH = jest.fn();
  });

  afterEach(() => {
    // 恢复原始函数，确保测试隔离
    app.sendToEH = originalSendToEH;
    jest.restoreAllMocks(); // 恢复所有模拟
  });

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

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

    // 验证模拟函数是否被调用
    expect(app.sendToEH).toHaveBeenCalledTimes(1);
    // 可以进一步验证调用参数
    // expect(app.sendToEH).toHaveBeenCalledWith(/* 期望的参数 */);
  });

  test('sendDataHandler should call sendToEH with specific arguments', async () => {
    app.sendToEH.mockImplementation(() => {
      console.log('Mocked sendToEH called');
    });

    const req = { data: 'test data' };
    const res = {};
    const next = jest.fn();

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

    expect(app.sendToEH).toHaveBeenCalled();
    // 假设 sendDataHandler 内部会传递 req.data 给 sendToEH
    // expect(app.sendToEH).toHaveBeenCalledWith(req.data);
  });
});