Jest用不同的参数模拟同一个函数两次

基础概念

Jest 是一个流行的 JavaScript 测试框架，广泛用于前端和 Node.js 应用的单元测试。模拟（Mocking）是 Jest 中的一个重要功能，允许你在测试中控制和监视函数的行为。

类型

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

手动模拟：手动创建一个模拟函数并定义其行为。
自动模拟：Jest 会自动模拟导入的模块中的所有函数。
使用 jest.mock：显式地告诉 Jest 模拟某个模块。

应用场景

当你需要测试一个函数的行为，而该函数依赖于其他函数或外部服务时，可以使用 Jest 的模拟功能。例如，测试一个处理用户数据的函数，而该函数依赖于一个数据库查询函数。

示例代码

假设我们有一个函数 getUserData，它依赖于一个 fetchUser 函数来获取用户数据：

// fetchUser.js
export const fetchUser = (userId) => {
  // 实际实现会调用外部服务
};

// userData.js
import { fetchUser } from './fetchUser';

export const getUserData = (userId) => {
  return fetchUser(userId).then(data => {
    return { id: data.id, name: data.name };
  });
};

我们可以使用 Jest 来模拟 fetchUser 函数，并测试 getUserData 函数的行为：

// userData.test.js
import { getUserData } from './userData';
import * as fetchUserModule from './fetchUser';

jest.mock('./fetchUser');

describe('getUserData', () => {
  it('should return user data for a valid user ID', async () => {
    fetchUserModule.fetchUser.mockResolvedValue({ id: 1, name: 'John Doe' });
    const result = await getUserData(1);
    expect(result).toEqual({ id: 1, name: 'John Doe' });
  });

  it('should handle errors for an invalid user ID', async () => {
    fetchUserModule.fetchUser.mockRejectedValue(new Error('User not found'));
    await expect(getUserData(999)).rejects.toThrow('User not found');
  });
});

遇到的问题及解决方法

如果你需要用不同的参数模拟同一个函数两次，可以使用 Jest 的 mockImplementation 或 mockImplementationOnce 方法。例如：

it('should handle different user IDs correctly', async () => {
  fetchUserModule.fetchUser.mockImplementation((userId) => {
    if (userId === 1) {
      return Promise.resolve({ id: 1, name: 'John Doe' });
    } else if (userId === 2) {
      return Promise.resolve({ id: 2, name: 'Jane Doe' });
    } else {
      return Promise.reject(new Error('User not found'));
    }
  });

  const result1 = await getUserData(1);
  expect(result1).toEqual({ id: 1, name: 'John Doe' });

  const result2 = await getUserData(2);
  expect(result2).toEqual({ id: 2, name: 'Jane Doe' });

  await expect(getUserData(999)).rejects.toThrow('User not found');
});

在这个示例中，我们使用 mockImplementation 方法来根据不同的参数返回不同的结果。