如何用Jest模拟另一个函数内部的函数_如何用jest模拟saga内部的API函数？_在函数内部使用Jest模拟类 - 腾讯云开发者社区

如何用Jest模拟另一个函数内部的函数

基础概念

Jest 是一个流行的 JavaScript 测试框架，广泛用于前端开发中的单元测试和集成测试。模拟（Mocking）是 Jest 中的一个重要功能，它允许你在测试中替换掉某些函数或对象的行为，以便更好地控制和验证测试环境。

类型

Jest 提供了几种模拟函数的方式：

手动模拟：通过 jest.fn() 手动创建一个模拟函数。
自动模拟：Jest 会自动模拟导入的模块中的函数。
使用 jest.mock()：可以全局或局部地模拟整个模块。

应用场景

当你需要测试一个函数，但该函数内部调用了其他函数，而这些函数的行为会影响测试结果时，可以使用 Jest 模拟这些内部函数。

示例代码

假设我们有一个函数 calculateTotal，它内部调用了另一个函数 getPrice：

// functions.js
export function getPrice(item) {
  // 假设这是一个复杂的计算
  return item.price * item.quantity;
}

export function calculateTotal(items) {
  let total = 0;
  for (const item of items) {
    total += getPrice(item);
  }
  return total;
}

我们可以使用 Jest 来模拟 getPrice 函数，以便测试 calculateTotal 函数：

// functions.test.js
import { calculateTotal, getPrice } from './functions';

jest.mock('./functions', () => {
  return {
    ...jest.requireActual('./functions'),
    getPrice: jest.fn(),
  };
});

describe('calculateTotal', () => {
  it('should calculate the total correctly', () => {
    const items = [
      { price: 10, quantity: 2 },
      { price: 5, quantity: 3 },
    ];

    // 模拟 getPrice 函数的行为
    getPrice.mockImplementation((item) => item.price * item.quantity);

    const result = calculateTotal(items);
    expect(result).toBe(35);

    // 验证 getPrice 函数是否被正确调用
    expect(getPrice).toHaveBeenCalledTimes(2);
    expect(getPrice).toHaveBeenCalledWith({ price: 10, quantity: 2 });
    expect(getPrice).toHaveBeenCalledWith({ price: 5, quantity: 3 });
  });
});