Jest测试中的Mock函数返回值控制

在Jest测试中，我们可以使用jest.fn()来创建一个mock函数。这个函数可以接收一个实现作为参数，然后它会返回一个新的函数，这个新函数可以跟踪它被调用的情况。我们还可以设置这个mock函数的返回值。

下面是一个简单的例子：

const myFunction = jest.fn();

// 设置 mock 函数的返回值
myFunction.mockReturnValue(42);

// 调用 mock 函数
const result = myFunction();

// 断言函数被调用
expect(myFunction).toHaveBeenCalled();

// 断言函数的返回值
expect(result).toBe(42);

你也可以链式地设置多个返回值：

const myFunction = jest.fn()
  .mockReturnValueOnce(1)
  .mockReturnValueOnce(2)
  .mockReturnValue(3);

console.log(myFunction()); // 输出 1
console.log(myFunction()); // 输出 2
console.log(myFunction()); // 输出 3
console.log(myFunction()); // 输出 3，因为这是链式设置的最后一个返回值

如果你想要在每次调用时返回不同的值，你可以使用.mockImplementation()方法：

let count = 0;
const myFunction = jest.fn().mockImplementation(() => {
  count++;
  return count * 2;
});

console.log(myFunction()); // 输出 2
console.log(myFunction()); // 输出 4
console.log(myFunction()); // 输出 6

这些方法可以让你更好地控制mock函数的行为，从而使你的测试更加灵活和可靠。