如何用Jest仅Mock ES6类中的指定方法

当然有啦！Jest提供了几种灵活的方式来只替换类里的特定方法，不用手动写继承类那么麻烦。下面给你两种最常用的方案：

方案1：使用jest.spyOn（推荐） #

这个方法精准又简单，专门用来监听并替换类的单个方法，其他方法完全保留原实现。

假设你的类A定义如下：

// ../myclass.js
export class A {
  methodToMock() {
    return 'original result';
  }
  keepThisMethod() {
    return 'original untouched value';
  }
}

测试代码可以这么写：

import { A } from '../myclass';

test('仅替换methodToMock，保留其他方法', () => {
  // 监听A原型上的目标方法
  const mockTargetMethod = jest.spyOn(A.prototype, 'methodToMock');
  
  // 替换方法实现：如果只需要固定返回值，用mockReturnValue
  mockTargetMethod.mockReturnValue('mocked result');
  
  // 要是需要自定义逻辑，用mockImplementation
  // mockTargetMethod.mockImplementation(() => {
  //   return 'custom mocked result';
  // });

  const instance = new A();
  expect(instance.methodToMock()).toBe('mocked result');
  expect(instance.keepThisMethod()).toBe('original untouched value');

  // 测试结束后恢复原方法，避免影响其他测试用例
  mockTargetMethod.mockRestore();
});

如果要替换静态方法，只需要把监听对象换成类本身就行：

jest.spyOn(A, 'staticMethod').mockReturnValue('mocked static value');

方案2：手动Mock模块时保留原方法 #

如果你偏好手动编写mock模块的方式，可以通过jest.requireActual导入原模块的完整实现，再针对性修改特定方法：

// 在测试文件顶部定义mock
jest.mock('../myclass', () => {
  // 获取未被mock的原模块内容
  const originalModule = jest.requireActual('../myclass');
  
  return {
    ...originalModule,
    // 重写A类的构造函数
    A: jest.fn().mockImplementation(() => {
      // 创建原类的实例，保留所有原方法
      const originalInstance = new originalModule.A();
      // 替换目标方法
      originalInstance.methodToMock = jest.fn().mockReturnValue('mocked result');
      return originalInstance;
    })
  };
});

import { A } from '../myclass';

test('手动Mock时保留原方法', () => {
  const instance = new A();
  expect(instance.methodToMock()).toBe('mocked result');
  expect(instance.keepThisMethod()).toBe('original untouched value');
});

这个方式适合需要在多个测试用例中复用同一个mock逻辑的场景。

内容的提问来源于stack exchange，提问作者Javid Asgarov