如何在Async.waterfall数组Node.js中为函数编写单元测试

const async = require('async');
const { expect } = require('chai');

// 被测试函数
function asyncFunction(arg1, arg2, callback) {
  // 异步操作
  setTimeout(() => {
    const result = arg1 + arg2;
    callback(null, result);
  }, 1000);
}

// 测试用例
describe('Async.waterfall unit test', () => {
  it('should return the sum of two numbers', (done) => {
    async.waterfall([
      (callback) => {
        asyncFunction(2, 3, callback);
      },
      (result, callback) => {
        expect(result).to.equal(5); // 断言结果是否符合预期
        callback(null, result);
      }
    ], (err, result) => {
      expect(err).to.be.null; // 断言错误是否为空
      expect(result).to.equal(5); // 断言最终结果是否符合预期
      done();
    });
  });

  it('should handle error correctly', (done) => {
    async.waterfall([
      (callback) => {
        asyncFunction(2, 'abc', callback);
      },
      (result, callback) => {
        // 不应该执行到这里
        callback(null, result);
      }
    ], (err, result) => {
      expect(err).to.be.an('error'); // 断言错误是否存在
      expect(result).to.be.undefined; // 断言结果是否为undefined
      done();
    });
  });
});