如何在 Jest 中模拟构造函数内调用的实例方法-js教程-PHP中文网

如何在 Jest 中模拟构造函数内调用的实例方法

本文介绍在 javascript 类构造函数中调用实例方法时，如何通过 jest 正确模拟该方法并验证其被调用，解决“方法在实例创建后才可 mock，但实际已在 constructor 中执行”的典型测试难题。

本文介绍在 javascript 类构造函数中调用实例方法时，如何通过 jest 正确模拟该方法并验证其被调用，解决“方法在实例创建后才可 mock，但实际已在 constructor 中执行”的典型测试难题。

在单元测试中，若一个类的 constructor 直接调用了自身的实例方法（如 this.helper() 或无 this 的裸调用），常规的「先 new 再 mock」方式会失效——因为方法已在实例化过程中执行完毕，mock 介入太晚。此时需采用原型级预置模拟（prototype-level mocking），确保构造函数执行时调用的就是已替换的 mock 函数。

关键思路是：不 mock 实例属性，而是 mock 类原型上的方法，使所有新实例共享该 mock 行为。Jest 提供了简洁可靠的实现方式：

// Example.js
class Example {
  constructor() {
    // 注意：此处必须显式使用 this.helper() 才能被原型 mock 拦截
    this.helper();
  }

  helper() {
    console.log('real implementation');
    return true;
  }
}

module.exports = Example;

// Example.test.js
const Example = require('./Example');

describe('Example', () => {
  beforeEach(() => {
    // ✅ 正确做法：直接 mock 原型方法（在 new 之前）
    Example.prototype.helper = jest.fn().mockReturnValue('mocked');
  });

  afterEach(() => {
    // ? 清理 mock，避免测试间污染
    jest.restoreAllMocks();
  });

  it('should call helper() during construction', () => {
    // Arrange & Act
    const instance = new Example();

    // Assert：验证原型方法被调用一次
    expect(Example.prototype.helper).toHaveBeenCalledTimes(1);
    expect(instance.helper()).toBe('mocked'); // 可选：验证后续调用也走 mock

    // 进阶断言：检查是否传入预期参数（若 helper 接收参数）
    // expect(Example.prototype.helper).toHaveBeenCalledWith('expected-arg');
  });
});

⚠️ 重要注意事项：

构造函数中必须使用 this.helper() 调用（而非裸 helper()），否则无法绑定到实例/原型，mock 将无效；
jest.mock() 全局模块模拟（如答案中所示）不适用于此类场景：它会替换整个模块导出，导致 new Example() 实际创建的是 mock 类而非原类，失去构造逻辑验证意义；正确做法是直接操作 Class.prototype.method；
使用 beforeEach + jest.restoreAllMocks() 是最佳实践，确保每个测试用例独立、可预测；
若 helper 是私有方法或依赖闭包状态，应考虑重构为可注入依赖（如将 helper 作为构造函数参数传入），以提升可测性。