如何让mocked方法为任意参数返回不同的值(从数组)？

// 假设我们要mock的方法是getRandomNumber，它接受一个参数n，并返回一个小于n的随机数

// 创建一个包含不同返回值的数组
const mockValues = [1, 2, 3, 4, 5];

// 创建mock对象并设置方法行为
const mockedObject = {
  getRandomNumber: jest.fn().mockImplementation(() => {
    return mockValues.shift(); // 返回数组中的下一个值
  }),
};

// 测试代码
test('测试getRandomNumber方法', () => {
  expect(mockedObject.getRandomNumber(10)).toBe(1); // 第一次调用返回1
  expect(mockedObject.getRandomNumber(10)).toBe(2); // 第二次调用返回2
  expect(mockedObject.getRandomNumber(10)).toBe(3); // 第三次调用返回3
});