如何测试被Middy包装的Handler函数（无需导出main）

方法1：直接导出main函数（最推荐） #

修改原代码，把const main改成export const main，测试代码就能直接导入并调用核心逻辑：

export const main = async (event: APIGatewayProxyEvent): Promise<APIGatewayProxyResult> => {
  // 核心逻辑代码
};

export const handle = middy(main)
  .use(httpJsonBodyParser())
  .use(customMiddleWare(ID))
  .use(httpErrorHandler());

测试文件示例：

import { main } from './your-lambda-file';
import { APIGatewayProxyEvent } from 'aws-lambda';

test('测试main函数逻辑', async () => {
  // 构造符合格式的模拟事件
  const mockEvent: APIGatewayProxyEvent = {
    body: JSON.stringify({ key: 'test-value' }),
    requestContext: {},
    headers: {},
    httpMethod: 'POST',
    path: '/api/test',
    isBase64Encoded: false,
    resource: '',
    queryStringParameters: null,
    multiValueQueryStringParameters: null,
    pathParameters: null,
    stageVariables: null,
    multiValueHeaders: {}
  };

  const result = await main(mockEvent);
  // 根据业务需求编写断言，比如检查返回状态码、响应体内容
  expect(result.statusCode).toBe(200);
});

方法2：仅在测试环境导出main（避免生产暴露） #

如果不想在生产代码中暴露main函数，可以通过环境变量做条件导出：

const main = async (event: APIGatewayProxyEvent): Promise<APIGatewayProxyResult> => {
  // 核心逻辑代码
};

export const handle = middy(main)
  .use(httpJsonBodyParser())
  .use(customMiddleWare(ID))
  .use(httpErrorHandler());

// 只在测试环境导出main
if (process.env.NODE_ENV === 'test') {
  export { main };
}

注意：Jest等测试框架会自动将NODE_ENV设为test，TypeScript配置需支持这种条件导出语法。

方法3：测试工具直接访问私有成员（不推荐） #

部分测试框架（如Jest）可以通过模块对象强制访问未导出的变量，但这种方式依赖代码实现细节，后续重构时容易失效：

import * as lambdaModule from './your-lambda-file';

test('测试main函数逻辑', async () => {
  // 强制获取未导出的main函数
  const main = (lambdaModule as any).main;
  const mockEvent = {} as APIGatewayProxyEvent;
  
  const result = await main(mockEvent);
  // 编写断言逻辑
});

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