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

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)
    }
  })

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

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();
    })
});