使用mocha测试节点回调

时间:2016-10-31 15:13:35

标签: node.js unit-testing mocha nodes aws-lambda

我尝试从async.waterfall中提取方法。我想分别测试它们。这是第一个在AWS上调用lambda的方法。

const async = require('async');
const AWS = require('aws-sdk');
AWS.config.region = 'eu-west-1';
const lambda = new AWS.Lambda();
const table_name = 'my_table';

exports.handler = function(event, context) {
  async.waterfall([
    increase_io_write_capacity(callback)
  ],
    function (err) {
      if (err) {
        context.fail(err);
      } else {
        context.succeed('Succeed');
      }
    });
};

function increase_io_write_capacity(callback) {
  var payload = JSON.stringify({
    tableName:table_name,
    increaseConsumedWriteCapacityUnits: 5
  });
  var params = {
    FunctionName: 'dynamodb_scaling_locker',
    InvocationType: 'RequestResponse',
    Payload: payload
  };

  lambda.invoke(params, function(error, data) {
    if (error) {
      console.log('Error invoker!' + JSON.stringify(error));
      callback('Invoker error' + JSON.stringify(error));
    } else {
      console.log('Done invoker!' + JSON.stringify(data));
      callback(null);
    }
  });
}

if (typeof exports !== 'undefined') {
  exports.increase_io_write_capacity = increase_io_write_capacity;
}

测试使用mochaaws-sdk-mock

const aws = require('aws-sdk-mock');
const testingAggregate = require('../index.js');
const assert = require('assert');
const expect = require( 'chai' ).expect;
const event = '';

describe('Testing aggregate function', function () {
  afterEach(function (done) {
    aws.restore();
    done();
  });

  describe('increase_io_write_capacity', function() {
    it('fail accessing to the lambda', function(done){
      aws.mock('Lambda', 'invoke', function(params, callback){
        callback('fail', null);
      });
      testingAggregate.increase_io_write_capacity(function(err, data){
        expect(err).to.equal('Error invoker! fail');
        done();
      });
    });
  });
});

问题是它永远不会断言。我正确地进入了方法,但从未进入lambda.invoke。似乎模拟永远不会返回回调。

  Testing aggregate function
    increase_io_write_capacity
      1) fail accessing to the lambda


  0 passing (2s)
  1 failing

  1) Testing aggregate function increase_io_write_capacity fail accessing to the lambda:
     Error: timeout of 2000ms exceeded. Ensure the done() callback is being called in this test.

如果我在aws-sdk-mock代码库中尝试一个简单的例子,我就不会有任何问题。

  t.test('mock function replaces method with replace function with lambda', function(st){
    awsMock.mock('Lambda', 'invoke', function(params, callback){
      callback("error", null);
    });
    var lambda = new AWS.Lambda();
    lambda.invoke({}, function(err, data){
      st.equals(err, "error'");
      awsMock.restore('Lambda');
      st.end();
    })
  })

也许我没有正确断言回调的结果?

提前感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

我使用aws-sdk-mock

失败了

我终于使用了proxyquire。代码不是那么干净,不是DRY。我不是JS开发人员。但它的确有效。随意进行更改。

const proxyquire = require('proxyquire').noCallThru();
const expect = require( 'chai' ).expect;

describe('testingAggregate', function() {
  describe('increase_io_write_capacity', function() {
    it('fail invoke the lambda', function(done){
      let awsSdkMock = {
        config: {
          region: 'eu-west-1'
        },
        Lambda: function() {
          this.invoke = function(params, callback) {
            expect(params.Payload).to.eq('{"tableName": "my_table_name","increaseConsumedWriteCapacityUnits":5}');
            callback(new Error('Lambda not in this region'));
          };
        }
      };
      let testingAggregate = proxyquire('../index', { 'aws-sdk': awsSdkMock });

      testingAggregate.increase_io_write_capacity(function(err, data){
        expect(err).to.equal('Invoker error: Error: Lambda not in this region');
      });
      done();
    });

    it('succeed invoking the lambda', function(done){
      let awsSdkMock = {
        config: {
          region: 'eu-west-1'
        },
        Lambda: function() {
          this.invoke = function(params, callback) {
            callback(null);
          };
        }
      };
      let testingAggregate = proxyquire('../index', { 'aws-sdk': awsSdkMock });

      testingAggregate.increase_io_write_capacity(function(err, data){
        expect(err).to.equal(null); //really not the best test
      });
      done();
    });
  });
});