在使用 `jest-mock-extended` 进行单元测试时，未显式模拟的函数默认返回 `undefined`，这可能导致难以追踪的测试失败。本文将介绍如何利用 `jest-mock-extended` 的 `fallbackmockimplementation` 功能，为所有未模拟的函数设置默认抛出错误的实现，从而在测试中更早、更清晰地发现缺失的模拟，显著提升测试的健壮性和调试效率。
Jest Mock中未实现函数返回undefined的挑战
在进行单元测试时，我们经常需要模拟（mock）外部依赖，以隔离被测试代码并控制其行为。jest-mock-extended 是一个强大的工具，它允许我们轻松地创建类型安全的模拟对象。然而，它的默认行为是，如果一个模拟对象的某个方法没有被显式地提供实现，那么调用该方法时它将返回 undefined。
考虑以下接口和模拟示例：
export interface SomeClient {
someFunction(): number;
someOtherFunction(): number;
}
const mockClient = mock();
mockClient.someFunction.mockImplementation(() => 1);
// 调用未模拟的函数
console.log(mockClient.someFunction()); // 输出: 1
console.log(mockClient.someOtherFunction()); // 输出: undefined 尽管在理想情况下，测试中的断言最终会因为 undefined 值而失败，但这种隐式的 undefined 返回可能导致以下问题：
- 难以定位根源： 被测代码可能在深层逻辑中才因 undefined 值而抛出错误，使得调试变得复杂，难以快速定位到是哪个模拟函数缺失了实现。
- 类型不匹配： 如果 TypeScript 项目中期望的返回类型不是 undefined，那么这种默认行为可能导致类型错误，但由于 Jest 的运行时特性，这些错误可能不会在编译时被捕获。
- 掩盖问题： 有时 undefined 值在某些上下文中可能不会立即导致错误，从而掩盖了测试中一个重要的缺失模拟。
传统但低效的解决方案
为了解决上述问题，一种直接但效率不高的方法是为每个模拟函数手动添加一个抛出错误的实现：
const mockClient = mock({ someFunction: jest.fn().mockImplementation(() => { throw new Error('someFunction not mocked'); }), someOtherFunction: jest.fn().mockImplementation(() => { throw new Error('someOtherFunction not mocked'); }), });
这种方法虽然能达到目的，但在实际项目中存在明显缺点：
- 繁琐重复： 当接口包含大量方法时，手动为每个方法添加这样的实现会非常繁琐。
- 维护困难： 每次接口新增方法时，都需要手动更新所有相关的模拟对象，增加了维护成本和出错的可能性。
利用 fallbackMockImplementation 实现默认错误抛出
jest-mock-extended 从版本 3.0.2 开始引入了一个强大的功能：fallbackMockImplementation。这个选项允许我们为所有未显式模拟的方法提供一个默认的实现。通过将这个默认实现设置为抛出错误，我们可以优雅地解决上述问题。
如何使用 fallbackMockImplementation
fallbackMockImplementation 作为 mock() 函数的第二个参数（选项对象）的一部分提供。它的值是一个函数，当任何未显式模拟的方法被调用时，这个函数就会被执行。
以下是使用 fallbackMockImplementation 的示例：
import { mock } from 'jest-mock-extended';
interface SomeClient {
someFunction(): number;
someOtherFunction(): number;
anotherMethod(arg: string): boolean;
}
describe('Jest Mock Default Error Handling', () => {
let mockClient: SomeClient;
beforeEach(() => {
// 创建一个模拟客户端，并设置fallbackMockImplementation
mockClient = mock(
{}, // 第一个参数是可选的，用于提供初始的模拟实现
{
// fallbackMockImplementation 会在任何未显式模拟的方法被调用时执行
fallbackMockImplementation: (methodName: string) => {
throw new Error(`Method '${methodName}' was called but not mocked!`);
},
},
);
});
test('显式模拟的函数按预期返回', () => {
mockClient.someFunction.mockImplementation(() => 123);
expect(mockClient.someFunction()).toBe(123);
});
test('未显式模拟的函数调用时抛出错误', () => {
// someOtherFunction 没有被显式模拟，调用时会触发 fallbackMockImplementation
expect(() => mockClient.someOtherFunction()).toThrowError("Method 'someOtherFunction' was called but not mocked!");
});
test('带有参数的未模拟函数也能正确抛出错误', () => {
// anotherMethod 也没有被显式模拟
expect(() => mockClient.anotherMethod('test')).toThrowError("Method 'anotherMethod' was called but not mocked!");
});
}); 在上面的示例中，fallbackMockImplementation 接收一个 methodName 参数，这使得我们可以在错误消息中包含具体是哪个方法未被模拟，从而提供更清晰的调试信息。
总结与注意事项
使用 jest-mock-extended 的 fallbackMockImplementation 功能，可以显著提升 Jest 测试的质量和可维护性：
- 更早发现问题： 在测试执行时，任何未被模拟的方法调用都会立即抛出错误，而不是默默返回 undefined，从而更快地暴露缺失的模拟。
- 清晰的错误信息： 自定义的错误消息可以明确指出是哪个方法未被模拟，大大简化了调试过程。
- 降低维护成本： 无需为每个方法手动添加错误抛出逻辑，接口变更时也无需大量修改模拟代码。
- 强制模拟完整性： 鼓励开发者在测试中为所有相关的依赖方法提供显式模拟，确保测试覆盖的完整性。
注意事项：
- 确保你的 jest-mock-extended 版本至少为 3.0.2。
- 在某些极少数情况下，如果你的被测代码确实期望一个未模拟的函数返回 undefined 并且这是其正常行为，那么你可能需要谨慎使用此功能，或者为那些特定函数提供一个返回 undefined 的显式模拟。然而，在大多数单元测试场景中，对外部依赖的每个交互都进行模拟是最佳实践。
通过采纳 fallbackMockImplementation，你的 Jest 测试将变得更加健壮、易于调试和维护。
