在Jest中，如何在同一测试文件的不同测试中以不同的方式模拟模块？

在Jest中，模拟（mocking）模块是一种常见的做法，用于隔离测试并确保测试的独立性。你可以在同一测试文件的不同测试中以不同的方式模拟同一个模块。以下是一些基础概念和相关方法：

基础概念

模拟模块：使用Jest的模拟功能来替换实际的模块，以便在测试中使用预定义的行为或返回值。
间谍函数（Spies）：用于跟踪函数调用及其参数。
存根（Stubs）：预定义的函数或对象，用于替代实际的功能。

类型与应用场景

手动模拟：适用于简单的模块替换。
自动模拟：Jest会自动模拟整个模块，适用于大多数情况。
部分模拟：只模拟模块中的某些函数或属性。

示例代码

假设我们有一个模块 mathUtils.js，其中包含两个函数 add 和 subtract：

// mathUtils.js
export const add = (a, b) => a + b;
export const subtract = (a, b) => a - b;

我们希望在不同的测试中以不同的方式模拟这个模块。

手动模拟

// __tests__/mathUtils.test.js
import * as mathUtils from '../mathUtils';

jest.mock('../mathUtils', () => {
  return {
    add: jest.fn(() => 10),
    subtract: jest.fn(() => 5)
  };
});

describe('mathUtils', () => {
  beforeEach(() => {
    jest.clearAllMocks();
  });

  it('should mock add function', () => {
    expect(mathUtils.add(2, 3)).toBe(10);
    expect(mathUtils.add).toHaveBeenCalledWith(2, 3);
  });

  it('should mock subtract function differently', () => {
    mathUtils.subtract.mockImplementation((a, b) => a * b);
    expect(mathUtils.subtract(2, 3)).toBe(6);
    expect(mathUtils.subtract).toHaveBeenCalledWith(2, 3);
  });
});

自动模拟

// __tests__/mathUtils.test.js
import * as mathUtils from '../mathUtils';

jest.mock('../mathUtils');

describe('mathUtils', () => {
  beforeEach(() => {
    jest.clearAllMocks();
  });

  it('should auto mock add function', () => {
    mathUtils.add.mockReturnValue(10);
    expect(mathUtils.add(2, 3)).toBe(10);
    expect(mathUtils.add).toHaveBeenCalledWith(2, 3);
  });

  it('should auto mock subtract function differently', () => {
    mathUtils.subtract.mockImplementation((a, b) => a * b);
    expect(mathUtils.subtract(2, 3)).toBe(6);
    expect(mathUtils.subtract).toHaveBeenCalledWith(2, 3);
  });
});