How to test await/async with DynamoDb on nodejs?

Question

I use NodeJs to implement AWS Lambda function and want to clarify what is the right way to test integration with DynamoDB:

Lambda code:

const AWS = require('aws-sdk');

AWS.config.update({region: region});
const dynamoDb = new AWS.DynamoDB();

exports.handler = async function(event, context) {
   ...
   await dynamoDb.deleteItem(item).promise();
}

I was going to use mocha, sinon, chai and aws-sdk-mock for tests:

const expect = require('chai').expect;
const AWS = require('aws-sdk-mock');
const lambda = require('../index.js');
const sinon = require('sinon');

describe('Test', function () {
  let deleteItemSpy;

  beforeEach(function () {
     deleteItemSpy = sinon.spy();
     AWS.mock('DynamoDB', 'deleteItem', deleteItemSpy);
  }
  it('valid call', async function() {
     await lambda.handler({"id":1}, null); 
     expect(deleteItemSpy.calledOnce).to.be.true;
  })
});

But there are two main problems:

Mock doesn't work if dynamoDb is created outside of the handler. What other options do I have? Can I use sinon.stub somehow?
It throws timeout because await never receives the result from lambda. Problem is related to spy itself. I probably can replace it with: AWS.mock('DynamoDB', 'deleteItem', function (params, callback) { });

you need to use sinon.stub(dynamoDb,'deleteItem')

muthukumar selvaraj
– muthukumar selvaraj

2018-07-17 12:13:59 +00:00
Commented Jul 17, 2018 at 12:13 — muthukumar selvaraj
– muthukumar selvaraj, Commented Jul 17, 2018 at 12:13

Orest · Accepted Answer · 2020-01-09 16:49:15Z

5

Ok I figured it out. Not sure if it the best way, but it works.

const expect = require('chai').expect;
const AWS = require('aws-sdk');
const sinon = require('sinon');

describe('Test', function () {
   let deleteItemStub;
   let mockDynamoDb;
   let lambda;

   before(function() {
       deleteItemStub = sinon.stub();
       mockDynamoDb = sinon.stub(AWS, 'DynamoDB').callsFake(function() {
           return {
               deleteItem: deleteItemStub
           };
       });
       lambda = require('../index.js');
   });
   after(function () {
       mockDynamoDb.restore();
   });

   beforeEach(function () {
       deleteItemStub.returns({ promise: () => Promise.resolve() });
   });

   afterEach(function () {
       deleteItemStub.reset();
   });
   it('valid call', async function() {
       await lambda.handler({"id":1}, null); 
       expect(deleteItemStub.calledOnce).to.be.true;
   });
});

edited Jan 9, 2020 at 16:49

answered Jun 26, 2018 at 1:23

Orest

6,77812 gold badges57 silver badges90 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to test await/async with DynamoDb on nodejs?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related