请求协助编写带serviceFunction模拟的AWS Lambda Jest单元测试

没问题，我来帮你搞定这个Jest单元测试的问题！核心就是用Jest的mock功能把dataService.retrieveData替换掉，这样就不会触发真实的服务调用，完全在测试环境里模拟各种返回场景。

核心思路 #

你的Lambda逻辑依赖serviceFunction，而serviceFunction本质是调用dataService.retrieveData。我们不需要测试dataService的真实功能（那是集成测试的范畴），只需要验证：

• 当retrieveData返回200响应时，Lambda是否直接返回该响应
• 当retrieveData返回非200响应时，Lambda是否格式化出正确的错误结构
• retrieveData是否被传入了正确的id参数

完整测试代码示例 #

假设你的Lambda函数文件名为lambda.js，我们创建对应的测试文件lambda.test.js：

const { lambdaService } = require('./lambda');
const dataService = require('../dataService');

// 自动模拟整个dataService模块，替换所有方法为Jest mock函数
jest.mock('../dataService');

describe('lambdaService', () => {
  // 测试成功场景：返回code 200的响应
  it('should return the raw success response when dataService returns code 200', async () => {
    // 模拟retrieveData的成功返回值
    const mockSuccessResponse = { code: 200, data: 'sample data' };
    dataService.retrieveData.mockResolvedValue(mockSuccessResponse);

    // 调用Lambda函数，传入测试用的event
    const result = await lambdaService({ id: 'test-id-123' });

    // 断言结果符合预期
    expect(result).toEqual(mockSuccessResponse);
    // 验证retrieveData被正确调用了一次，参数是传入的id
    expect(dataService.retrieveData).toHaveBeenCalledTimes(1);
    expect(dataService.retrieveData).toHaveBeenCalledWith('test-id-123');
  });

  // 测试错误场景：返回非200的响应
  it('should return formatted error response when dataService returns non-200 code', async () => {
    const mockErrorResponse = { code: 404, message: 'Requested data not found' };
    dataService.retrieveData.mockResolvedValue(mockErrorResponse);

    const result = await lambdaService({ id: 'invalid-id-456' });

    expect(result).toEqual({
      statusCode: 404,
      body: JSON.stringify({ message: 'Requested data not found' })
    });
    expect(dataService.retrieveData).toHaveBeenCalledWith('invalid-id-456');
  });

  // 可选：测试dataService抛出错误的场景
  it('should handle errors thrown by dataService', async () => {
    const mockError = new Error('Database connection failed');
    dataService.retrieveData.mockRejectedValue(mockError);

    // 这里需要用expect().rejects.toThrow()来捕获异步错误
    await expect(lambdaService({ id: 'error-id-789' })).rejects.toThrow(mockError);
  });
});

关键知识点解释 #

1. jest.mock('../dataService')：这行代码会自动生成dataService模块的mock版本，所有方法都会变成Jest的mock function，不会执行真实的服务调用。
2. mockResolvedValue：用来模拟异步函数的成功返回值，对应async/await的成功场景。如果要模拟失败（比如服务抛出错误），就用mockRejectedValue。
3. 私有函数serviceFunction的测试：因为serviceFunction是Lambda文件内部的私有函数，我们不需要单独测试它——通过测试公共的lambdaService，已经间接覆盖了它的逻辑（只要验证retrieveData被正确调用，就说明serviceFunction的逻辑是对的）。如果确实需要单独测试它，可以考虑把它导出为模块的私有成员（比如exports._serviceFunction = serviceFunction），不过一般不推荐，因为私有函数属于实现细节。

运行测试 #

在项目根目录执行npx jest lambda.test.js就能运行这个测试用例啦，所有场景都会在模拟环境下执行，不会触发真实的服务调用。

内容的提问来源于stack exchange，提问作者Prabhakaran