从RSpec转用Jest：如何取消单个类实例方法的模拟？

我明白你从RSpec转Jest时遇到的困惑——两者的Mock思路确实有差异，Jest没有像RSpec那样直接提供“一键取消单个方法Mock”的API，但我们可以通过手动替换原型方法的方式实现你想要的效果：先整体Mock整个类，再选择性恢复特定方法来单独测试。

核心思路 #

Jest的jest.mock()会把整个模块替换成自动生成的Mock类，所有原型方法都会变成jest.fn()。要恢复单个方法，我们需要：

1. 通过jest.requireActual()获取原模块的真实类/方法
2. 将Mock类原型上的目标方法替换为原方法，其他方法保持Mock状态

具体实现 #

假设你的foo.js是CoffeeScript代码：

class Foo
  constructor: ->
    @bar()
    @baz()
  bar: -> 'bar'
  baz: -> 'baz'

对应的测试文件foo_test.js可以这样写：

// 1. 先整体Mock整个Foo模块
jest.mock('./foo');

// 2. 获取原模块的真实Foo类，以及Mock后的Foo类
const OriginalFoo = jest.requireActual('./foo').default;
const Foo = require('./foo').default;

describe('Foo类的部分Mock测试', () => {
  // 每次测试前重置所有Mock的调用记录，避免测试污染
  beforeEach(() => {
    jest.clearAllMocks();
  });

  test('恢复bar方法，保持baz为Mock状态', () => {
    // 将Mock类的bar方法替换为原方法
    Foo.prototype.bar = OriginalFoo.prototype.bar;

    const foo = new Foo();

    // 验证bar返回真实值
    expect(foo.bar()).toBe('bar');
    // 验证baz是Mock函数，默认返回undefined，且被构造函数调用过
    expect(foo.baz()).toBeUndefined();
    expect(foo.baz).toHaveBeenCalled();

    // 如果需要验证bar的调用次数（原方法本身没有Mock能力），可以用包裹的方式：
    const originalBar = OriginalFoo.prototype.bar;
    Foo.prototype.bar = jest.fn(originalBar); // 用Mock函数包裹原方法
    const foo2 = new Foo();
    expect(foo2.bar).toHaveBeenCalled(); // 现在可以验证调用了
    expect(foo2.bar()).toBe('bar'); // 依然返回真实值
  });

  test('恢复baz方法，保持bar为Mock状态', () => {
    // 替换baz方法为原方法
    Foo.prototype.baz = OriginalFoo.prototype.baz;

    const foo = new Foo();
    expect(foo.baz()).toBe('baz');
    expect(foo.bar()).toBeUndefined();
    expect(foo.bar).toHaveBeenCalled();
  });

  test('测试后恢复Mock方法（可选）', () => {
    // 先保存原来的Mock方法
    const originalMockBar = Foo.prototype.bar;
    // 替换为原方法
    Foo.prototype.bar = OriginalFoo.prototype.bar;

    // 执行测试逻辑...

    // 恢复成原来的Mock方法，避免影响其他测试
    Foo.prototype.bar = originalMockBar;
    const foo = new Foo();
    expect(foo.bar()).toBeUndefined(); // 回到Mock状态
  });
});

关键注意点 #

• jest.unmock()是用来取消整个模块的Mock，恢复原模块，不能用来单独恢复某个方法，这和你之前尝试的用法不同。
• 如果需要保留原构造函数的逻辑，只Mock部分方法，可以手动编写Mock工厂函数：

  jest.mock('./foo', () => {
    const OriginalFoo = jest.requireActual('./foo').default;
    return {
      default: jest.fn().mockImplementation(() => {
        const instance = new OriginalFoo();
        // 仅Mock baz方法，保留bar方法
        instance.baz = jest.fn();
        return instance;
      })
    };
  });
  

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