Node.js Lambda函数Mocha/Chai单元测试异常问题排查

问题背景 #

你有一个Node.js环境下的Lambda函数，成功响应格式如下：

{"moduleName":"getVVTags","message":"[\"6.9.0\",\"6.8.0\",...]","moduleType":"Lambda","moduleVersion":"1.0","language":"en","memberId":"NA","invokedFunctionArn":"arn:aws:lambda:us-east-1:280362372798:function:handler:1.0","clientIP":"NA","awsRequestId":"8d9db1be-04e1-26e5-01b0-2cff55576062","errorDescription":"","logEvent":"Response","httpStatusCode":200}

使用Mocha/Chai编写的单元测试代码如下：

const { describe } = require("mocha");
const chai = require("chai");
const { expect } = chai;
const getVVTagsLambdaHandler = require("../src/api/getVVTags/handler/GetVVTags");
const getVVTagsMockEvent = require("../mock/GetVVTags.json");

describe('getVVTags', () => {
    it('shoud return', () => {
        try {
            getVVTagsLambdaHandler(getVVTagsMockEvent, {
                succeed: result => {
                    console.log({result});
                    expect(result).to.be.a('object');
                    expect(result).to.not.be.empty;
                    expect(result.statusCode).to.be.equal(200);
                    expect(result.message).to.be.a("array");
                },
                fail: result => {
                    expect(result).to.have.property('httpStatusCode');
                    expect(result.httpStatusCode).to.not.be.equal(200);
                }
            });
        } catch (error) {
            // done(error.message);
        }
    });
});

运行测试后遇到两个问题：

• 无论断言是否正确，测试始终通过
• 控制台看不到console.log的输出

问题根源 #

1. 测试未等待异步回调执行：你的Lambda handler是基于succed/fail回调的异步逻辑，但Mocha默认按同步测试执行，会在handler调用完成后立刻结束测试，根本不会等待回调里的断言执行，所以无论断言对错，测试都会直接标记为通过。
2. 日志输出被遗漏：同理，console.log在异步回调中，测试结束后Mocha可能已经清理了输出流，导致日志无法显示。
3. 断言本身存在错误：Lambda响应里是httpStatusCode而非statusCode，且message是字符串化的数组而非真正的数组，这些错误断言也会导致测试逻辑不符合预期。

修复方案 #

方案1：包装handler为Promise，使用async/await测试（推荐） #

将回调式的Lambda handler包装成返回Promise的函数，配合Mocha的async/await语法实现异步测试，确保测试等待handler执行完成：

const { describe, it } = require("mocha");
const chai = require("chai");
const { expect } = chai;
const getVVTagsLambdaHandler = require("../src/api/getVVTags/handler/GetVVTags");
const getVVTagsMockEvent = require("../mock/GetVVTags.json");

// 把回调式handler包装为Promise
const wrappedHandler = (event) => {
  return new Promise((resolve, reject) => {
    getVVTagsLambdaHandler(event, {
      succeed: resolve,
      fail: reject
    });
  });
};

describe('getVVTags', () => {
  it('should return valid success response', async () => {
    const result = await wrappedHandler(getVVTagsMockEvent);
    console.log({result});
    
    // 修正断言逻辑
    expect(result).to.be.an('object');
    expect(result).to.not.be.empty;
    expect(result.httpStatusCode).to.equal(200);
    // 解析字符串化的message后断言为数组
    expect(JSON.parse(result.message)).to.be.an('array');
  });

  it('should handle errors correctly', async () => {
    // 构造触发错误的mock事件（根据业务逻辑修改）
    const errorEvent = {...getVVTagsMockEvent, invalidParam: true};
    
    try {
      await wrappedHandler(errorEvent);
      // 如果handler没有抛出错误，手动标记测试失败
      expect.fail('Expected handler to reject with error');
    } catch (error) {
      expect(error).to.have.property('httpStatusCode');
      expect(error.httpStatusCode).to.not.equal(200);
    }
  });
});

方案2：使用Mocha的done回调 #

如果不想用Promise，也可以直接使用Mocha的done回调，在succed/fail中手动控制测试结束时机：

const { describe, it } = require("mocha");
const chai = require("chai");
const { expect } = chai;
const getVVTagsLambdaHandler = require("../src/api/getVVTags/handler/GetVVTags");
const getVVTagsMockEvent = require("../mock/GetVVTags.json");

describe('getVVTags', () => {
  it('should return valid success response', (done) => {
    getVVTagsLambdaHandler(getVVTagsMockEvent, {
      succeed: result => {
        console.log({result});
        try {
          expect(result).to.be.an('object');
          expect(result).to.not.be.empty;
          expect(result.httpStatusCode).to.equal(200);
          expect(JSON.parse(result.message)).to.be.an('array');
          done(); // 断言成功，通知Mocha测试结束
        } catch (err) {
          done(err); // 断言失败，把错误传给Mocha
        }
      },
      fail: result => {
        try {
          expect(result).to.have.property('httpStatusCode');
          expect(result.httpStatusCode).to.not.equal(200);
          done();
        } catch (err) {
          done(err);
        }
      }
    });
  });
});

关键注意事项 #

• 修正断言字段：Lambda响应中是httpStatusCode，不是你测试里写的statusCode，必须匹配字段名。
• 处理字符串化数组：message字段是JSON字符串，需要用JSON.parse解析后再断言数组类型。
• 异步测试必须显式等待：无论是用Promise/async还是done回调，都要确保Mocha等待异步逻辑完成后再结束测试。

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