本文探讨了在使用Jest进行单元测试时，模拟函数（mock function）在跨模块调用中失效的问题。当一个模块内部函数调用另一个内部函数时，直接对外部对象进行模拟可能无法生效。文章提供了一种解决方案，通过将相关函数封装在一个导出的对象中，确保内部调用和外部模拟都指向同一个可变引用，从而实现有效的测试。
问题剖析：模拟函数传递失效的原因
在使用jest进行单元测试时，我们经常需要模拟（mock）某些函数的行为，以便隔离被测试代码，专注于其逻辑。然而，当被测试的函数（例如senddatahandler）位于一个模块中，并且它内部调用了同一个模块或另一个模块的另一个函数（例如sendtoeh）时，直接在测试文件中对外部对象属性进行模拟可能会遇到问题。
考虑以下场景：一个app对象包含sendDataHandler和sendToEH两个方法。我们希望测试sendDataHandler是否正确调用了sendToEH。直观的测试代码可能如下：
// app.JS (或模块内部) // 假设 sendDataHandler 内部会调用 sendToEH // test.js app.sendToEH = jest.fn(); // 尝试模拟 app 上的 sendToEH 方法 await app.sendDataHandler(req, res, next); // 调用 sendDataHandler expect(app.sendToEH).toHaveBeenCalled(); // 期望模拟函数被调用
在这种情况下，测试往往会失败。调试发现，当sendDataHandler被调用时，它内部执行的sendToEH并不是我们在测试文件中定义的jest.fn()模拟函数，而是app对象上原始的sendToEH方法。
其根本原因在于JavaScript的模块系统和引用机制。当sendDataHandler在其模块内部被定义时，它可能已经捕获了sendToEH的一个特定引用。如果在测试文件中，我们直接修改了app对象上的sendToEH属性，这只会改变app对象外部可见的sendToEH引用。而sendDataHandler内部所持有的sendToEH引用并未随之更新，它依然指向原始的函数实例。因此，sendDataHandler调用的仍是原始函数，而非我们期望的模拟函数。
解决方案：导出对象封装函数
为了解决这个问题，我们需要确保sendDataHandler内部调用的sendToEH与我们在测试文件中模拟的sendToEH是同一个可变引用。一种有效的策略是将所有相关方法封装在一个对象中，并将该对象作为模块的默认导出。这样，sendDataHandler将通过该对象的属性来引用sendToEH，而测试文件也可以通过修改该对象的属性来注入模拟函数。
模块实现示例
假设我们的模块名为myModule.js，它包含sendToEH和sendDataHandler。
// myModule.js // 定义 sendToEH 的原始实现 var sendToEH = function() { console.log("Original sendToEH called"); // 实际的数据发送逻辑 }; // 定义 sendDataHandler 的原始实现 // 注意：这里通过 exportFunctions 对象来引用 sendToEH var sendDataHandler = function(req, res, next) { console.log("sendDataHandler called"); // 模拟一些处理逻辑 exportFunctions.sendToEH(); // 关键：通过 exportFunctions 引用 }; // 将所有需要导出和内部引用的函数封装在一个对象中 const exportFunctions = { sendToEH, sendDataHandler }; // 默认导出这个封装对象 export default exportFunctions;
在这个实现中，sendDataHandler不再直接引用一个独立的sendToEH变量，而是通过exportFunctions.sendToEH来访问。
测试文件实现示例
现在，在测试文件中，我们可以导入这个封装对象，并对其属性进行模拟。
// myModule.test.js import app from './myModule'; // 导入封装对象，这里命名为 app describe('sendDataHandler functionality', () => { let req, res, next; beforeEach(() => { // 模拟请求、响应和next函数 req = {}; res = {}; next = jest.fn(); // 每次测试前重置模拟函数 // 关键：对导入的 app 对象上的 sendToEH 属性进行模拟 app.sendToEH = jest.fn(); }); test('sendDataHandler should call sendToEH', async () => { await app.sendDataHandler(req, res, next); // 调用 sendDataHandler // 期望模拟函数被调用 expect(app.sendToEH).toHaveBeenCalledTimes(1); }); test('sendDataHandler should call sendToEH with specific arguments', async () => { // 可以在这里进一步模拟 sendToEH 的返回值或实现 app.sendToEH.mockReturnValue('data sent successfully'); await app.sendDataHandler(req, res, next); expect(app.sendToEH).toHaveBeenCalledWith(/* 期望的参数 */); // 可以在这里验证 sendDataHandler 的其他行为 }); });
通过这种方式，当app.sendDataHandler被调用时，它内部执行的exportFunctions.sendToEH()实际上就是我们在测试文件中通过app.sendToEH = jest.fn()所注入的模拟函数。因为app（即exportFunctions）是一个共享的引用，修改其属性会影响到所有使用该引用的地方，包括模块内部的调用。
原理分析与最佳实践
这种方法成功的关键在于引用的一致性。当我们将函数封装在一个对象中并导出时，我们实际上是导出了一份对该对象的引用。模块内部的函数（如sendDataHandler）通过这个共享对象的属性来访问其他函数（如sendToEH）。在测试文件中，我们导入的也是这个共享对象的引用。当我们修改app.sendToEH时，我们是在修改这个共享对象上的一个属性，因此，模块内部的sendDataHandler在查找exportFunctions.sendToEH时，会找到我们刚刚注入的模拟函数。
最佳实践总结：
- 统一导出接口： 将模块内部需要互相协作或需要被外部测试/调用的函数，统一封装在一个对象中并作为模块的默认（或命名）导出。
- 内部引用一致： 模块内部的函数在调用其他内部函数时，应通过这个封装对象的属性来引用，而不是直接引用独立的变量。
- 测试时修改属性： 在测试文件中，导入这个封装对象，然后直接修改其属性来注入模拟函数。
这种模式不仅解决了Jest模拟函数在跨模块调用中的失效问题，也提升了模块的内聚性和可测试性。它使得模块的内部协作机制更加清晰，并为单元测试提供了明确的模拟切入点。
