如何Mock TypeScript类的多个函数？Jest测试问题解析

一、为什么仅Mock __functionA 并断言时测试通过？ #

这种情况通常是因为其他未Mock的方法执行了真实逻辑，引入了副作用，导致测试失败。具体来说：

• 当你只Mock __functionA 时，这个方法被替换成了Jest的mock函数，不会执行原有的代码逻辑，也就不会产生任何副作用（比如调用外部API、修改全局状态、抛出异常等），此时断言它是否被调用的逻辑不受其他方法影响，自然能通过。
• 但当你测试其他场景（比如调用类的另一个方法，期望它触发 __functionA），如果其他方法没有被Mock，就会执行真实代码——可能真实代码里并没有调用 __functionA，或者调用过程中因为依赖缺失、状态不对导致流程中断，最终导致断言失败。

另外还有一种可能：你在Mock多个方法时，Mock的方式不正确，导致mock没有生效，或者mock的实现干扰了测试逻辑。

二、如何正确Mock类的多个方法？ #

在Jest中，有两种常见的方式可以Mock类的多个方法，根据你的需求选择：

方式1：使用 jest.spyOn 逐个Mock方法 #

适合需要保留部分原方法逻辑，仅Mock特定方法的场景：

import { YourClass } from './a.ts';

describe('YourClass', () => {
  let instance: YourClass;

  beforeEach(() => {
    instance = new YourClass();
    // Mock多个方法
    jest.spyOn(instance, '__functionA').mockImplementation(() => 'mocked result A');
    jest.spyOn(instance, '__functionB').mockImplementation(() => 'mocked result B');
    jest.spyOn(instance, '__functionC').mockResolvedValue('mocked async result C'); // 异步方法
  });

  afterEach(() => {
    // 清除所有mock，避免污染其他测试
    jest.clearAllMocks();
  });

  test('should call both __functionA and __functionB', () => {
    instance.someMethodThatCallsAB(); // 假设这个方法会调用A和B

    expect(instance.__functionA).toHaveBeenCalled();
    expect(instance.__functionB).toHaveBeenCalled();
  });
});

方式2：使用 jest.mock 直接Mock整个类 #

适合需要完全替换类的所有方法，不需要原逻辑的场景：

// 直接Mock整个类，所有方法都会被替换成jest mock函数
jest.mock('./a.ts', () => {
  return {
    YourClass: jest.fn().mockImplementation(() => {
      return {
        __functionA: jest.fn(),
        __functionB: jest.fn().mockReturnValue('mocked B'),
        __functionC: jest.fn().mockResolvedValue('mocked async C'),
      };
    }),
  };
});

import { YourClass } from './a.ts';

describe('YourClass', () => {
  let instance: jest.Mocked<YourClass>;

  beforeEach(() => {
    instance = new YourClass() as jest.Mocked<YourClass>;
  });

  test('should trigger multiple mocked methods', () => {
    instance.__functionA();
    instance.__functionB();

    expect(instance.__functionA).toHaveBeenCalledTimes(1);
    expect(instance.__functionB).toHaveReturnedWith('mocked B');
  });
});

三、常见错误操作指正 #

1. Mock方法后未正确处理异步逻辑：如果被Mock的方法是异步的，需要用 async/await 或者 .resolves/.rejects 来处理断言，否则测试会在异步操作完成前结束，导致断言失败。
2. Mock的方法名拼写错误：注意类的方法名（比如你的 __functionA 是双下划线开头，别漏写），如果拼写错误，mock不会生效，执行的还是真实方法。
3. 没有在测试前后清除Mock：如果多个测试共用同一个mock实例，前一个测试的调用记录会影响后一个测试，记得用 jest.clearAllMocks() 或者 jest.resetAllMocks() 在 afterEach 中清除。
4. 错误地Mock了类的原型而非实例：如果用 jest.spyOn(YourClass.prototype, '__functionA')，需要确保实例是在mock之后创建的，否则mock不会生效到已创建的实例上。

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