在 jest 单元测试中，使用 `jest-mock-extended` 创建的模拟对象，其未显式实现的函数默认返回 `undefined`，这可能导致难以追踪的测试失败和类型不匹配问题。本文将介绍如何利用 `jest-mock-extended` 的 `fallbackmockimplementation` 选项，为所有未实现的模拟函数设置一个默认的错误抛出行为，从而在第一时间发现并解决缺失的模拟实现，显著提升测试的清晰度和调试效率。
单元测试中模拟函数默认行为的挑战
在进行单元测试时，我们经常使用模拟（Mock）对象来隔离被测试代码的依赖项。jest-mock-extended 是一个流行的库，它能帮助我们更方便地创建类型安全的模拟。然而，其默认行为在某些情况下可能会带来挑战。
当我们使用 jest-mock-extended 创建一个模拟对象时，例如：
import { mock } from 'jest-mock-extended';
interface SomeClient {
someFunction(): number;
someOtherFunction(): number;
}
const mockClient = mock<SomeClient>();
mockClient.someFunction.mockImplementation(() => 1);此时，如果 someOtherFunction 未被显式地提供模拟实现，它在被调用时将默认返回 undefined：
mockClient.someFunction(); // 返回 1 mockClient.someOtherFunction(); // 返回 undefined
这种默认返回 undefined 的行为，尽管在某些情况下最终也会导致测试失败，但问题往往出现在被测试代码的深层逻辑中，而非模拟函数被调用的直接位置。这使得问题的根源难以定位，例如：
- 被测试代码可能不期望 undefined，从而在更远的执行路径上抛出错误。
- 在 TypeScript 项目中，undefined 的返回可能与预期的类型不匹配，导致类型安全问题。
- 测试失败的原因变得模糊，调试过程耗时且复杂。
为了提高测试的健壮性和调试效率，我们期望所有未被显式模拟的函数在被调用时能立即抛出错误，明确指出“此函数未被模拟”。虽然可以通过为每个函数手动添加 jest.fn().mockImplementation(() => { throw new Error(...) }) 来实现，但这无疑会增加大量的重复代码，并且在接口新增方法时需要手动维护，效率低下。
解决方案：使用 jest-mock-extended 的 fallbackMockImplementation
jest-mock-extended 库从 3.0.2 版本开始引入了一个强大的功能：fallbackMockImplementation 选项。这个选项允许我们为模拟对象设置一个“回退”的默认实现，当任何未被显式模拟的函数被调用时，都会执行这个回退实现。
通过利用 fallbackMockImplementation，我们可以轻松地实现所有未显式模拟的函数在被调用时抛出错误的行为，从而在第一时间发现并解决缺失的模拟实现。
使用方法
fallbackMockImplementation 选项作为 mock() 函数的第二个参数（选项对象）的一部分提供。其基本语法如下：
import { mock } from 'jest-mock-extended';
// ... 接口定义 ...
const mockObject = mock<YourInterface>(
{}, // 第一个参数：可选的初始模拟实现
{
fallbackMockImplementation: () => {
throw new Error('此函数未被模拟');
},
}
);示例代码
让我们通过一个具体的例子来演示如何应用 fallbackMockImplementation 来确保未模拟的函数抛出错误：
import { mock } from 'jest-mock-extended';
// 定义一个示例接口
interface SomeClient {
someFunction(): number;
someOtherFunction(): number;
anotherFunction(param: string): boolean;
}
describe('jest-mock-extended 默认错误抛出行为', () => {
test('当未指定模拟实现时，函数应抛出错误', () => {
// 创建一个模拟客户端，并设置 fallbackMockImplementation
const mockClient = mock<SomeClient>(
{}, // 第一个参数可以传入初始的模拟实现，这里我们留空
{
// 设置回退实现：任何未显式模拟的函数被调用时都抛出错误
fallbackMockImplementation: (methodName: string) => {
// 可以根据方法名提供更具体的错误信息
throw new Error(`方法 '${methodName}' 未被模拟`);
},
},
);
// 验证 someFunction 在未被模拟时会抛出错误
expect(() => mockClient.someFunction()).toThrowError("方法 'someFunction' 未被模拟");
// 验证 someOtherFunction 在未被模拟时会抛出错误
expect(() => mockClient.someOtherFunction()).toThrowError("方法 'someOtherFunction' 未被模拟");
// 验证 anotherFunction 即使有参数，在未被模拟时也会抛出错误
expect(() => mockClient.anotherFunction('test')).toThrowError("方法 'anotherFunction' 未被模拟");
});
test('当指定模拟实现时，函数应按预期执行', () => {
const mockClient = mock<SomeClient>(
{},
{
fallbackMockImplementation: (methodName: string) => {
throw new Error(`方法 '${methodName}' 未被模拟`);
},
},
);
// 为 someFunction 提供一个具体的模拟实现
mockClient.someFunction.mockImplementation(() => 42);
// 验证 someFunction 按其模拟实现返回结果
expect(mockClient.someFunction()).toBe(42);
// 验证 someOtherFunction 仍然会抛出错误，因为它没有被模拟
expect(() => mockClient.someOtherFunction()).toThrowError("方法 'someOtherFunction' 未被模拟");
});
});在上面的示例中，我们为 mockClient 设置了 fallbackMockImplementation。当 someFunction 或 someOtherFunction 在没有显式模拟的情况下被调用时，它们会立即抛出包含特定错误信息的 Error。这使得测试失败的原因一目了然，极大地简化了调试过程。
注意事项
- 版本要求：确保你的 jest-mock-extended 版本是 3.0.2 或更高。如果版本过低，此功能将不可用。
- 错误信息：fallbackMockImplementation 函数可以接收被调用方法的名称作为参数，这允许你生成更具描述性的错误信息，例如 throw new Error(\方法 '${methodName}' 未被模拟`);`。
- 与其他模拟的结合：fallbackMockImplementation 不会影响你为特定方法提供的显式模拟。如果一个方法有自己的 mockImplementation，它将优先于回退实现被执行。
- 适用场景：此功能特别适用于那些要求所有依赖项交互都必须被明确定义的严格测试环境。它强制开发者思考并模拟所有必要的行为，从而避免隐式 undefined 带来的潜在问题。
总结
通过 jest-mock-extended 提供的 fallbackMockImplementation 选项，我们可以优雅地解决模拟函数默认返回 undefined 带来的调试难题。将未显式模拟的函数设置为默认抛出错误，不仅能帮助我们更早、更清晰地发现测试中的问题，还能提升单元测试的健壮性和可维护性。这使得测试代码更加可靠，也让开发人员能够专注于业务逻辑的实现，而不是花费大量时间去追踪模拟行为导致的意外错误。在构建高质量的测试套件时，充分利用这一特性将极大地提升开发效率和代码质量。
