如何为AWS SQS编写mocha单元测试？

Question

我的代码是：

  const validateFile = function(file) {
    // DoStuff
  }

  const validatorConsumer = Consumer.create({
    queueUrl: 'https://sqs.eu-west-1.amazonaws.com/account-id/queue-name',
    handleMessage: (message, done) => {
      return FileLib.downloadFromBucket(message)
      .then(validateFile)
      .then(done)
    }
  })

我正在使用sqs-consumer的npm包。如何编写单元测试以查看它是否有效？

Answer 1

我假设您想要对handleMessage函数进行单元测试。

如果在单独的节点模块中定义函数，则可以在单元测试中以及在创建使用者的生产代码中对其进行require。在您的测试中，您只需传入可能出现在队列中的手动创建的消息以及用于进行测试断言的done回调的自定义函数。

// messageHandler.js
module.export = function(message, done) {
    // message handler code...
    return FileLib.downloadFromBucket(message)
        .then(validateFile)
        .then(done)
    }
}

// main.js
const messageHandler = require('./messageHandler');
const validatorConsumer = Consumer.create({
    queueUrl: 'https://sqs.eu-west-1.amazonaws.com/account-id/queue-name',
    handleMessage: messageHandler
})

// messageHandler.test.js
const messageHandler = require('./messageHandler');
const message = '...' // create a sample message here

it('handles message correctly', done => {
    messageHandler(message, result => {
        // assertions go here...
        done();
    })
});