node.js - 使用 aws-sdk-mock 的 Promise 支持和 DocumentClient 进行模拟

Question

我正在尝试使用 aws-sdk-mock 的 promise 支持编写单元测试。我正在使用 DocumentClient。

我的代码如下所示:

const docClient = new AWS.DynamoDB.DocumentClient();

const getItemPromise = docClient.get(params).promise();
   return getItemPromise.then((data) => {
   console.log('Success');
   return data;
}).catch((err) => {
   console.log(err);
});

我的模拟和单元测试如下所示:

const AWS = require('aws-sdk-mock');
AWS.Promise = Promise.Promise;

AWS.mock('DynamoDB.DocumentClient', 'get', function (params, callback)
{
   callback(null, { Item: { Key: 'test value } });
});

dynamoStore.getItems('tableName', 'idName', 'id').then((actualResponse) => {
  // assertions
  done();
});

运行我的单元测试，不会返回我的测试值，它实际上绕过我的模拟，并直接调用 dynamoDb。我究竟做错了什么？如何正确设置我的模拟？

Answer 1

从您的代码中不清楚，但 aws-sdk-mock 有此注释

NB: The AWS Service needs to be initialised inside the function being tested in order for the SDK method to be mocked

因此以下内容将无法正确模拟

var AWS      = require('aws-sdk');
var sns      = AWS.SNS();
var dynamoDb = AWS.DynamoDB();

exports.handler = function(event, context) {
  // do something with the services e.g. sns.publish 
}

但这会

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

exports.handler = function(event, context) {
  var sns      = AWS.SNS();
  var dynamoDb = AWS.DynamoDB();
  // do something with the services e.g. sns.publish 
}

在这里查看更多https://github.com/dwyl/aws-sdk-mock#how-usage