4

我正在尝试用 Sinon 模拟 SES，但面临以下错误。尝试使用 aws-sdk-mock，但它不起作用。

Error: TypeError: Cannot stub non-existent own property sendEmail

测试类代码片段：

import * as AWS from 'aws-sdk';

const sandbox = sinon.createSandbox();
sandbox.stub(AWS.SES, 'sendEmail').returns({promise: () => true});

实际班级：

import * as AWS from 'aws-sdk';
import * as _ from 'lodash';    

export async function sendAlertMailOnFailure(status:any)
{   
    // load AWS SES
    var ses = new AWS.SES();   
    const params = {
        Destination: {
          ToAddresses: <to_address>
        },
        Message: {...},    
        Source: <sender_address>
      }
      ses.sendEmail(params, (err, data) => {
        if (err) {
          log.error("Error sending mail::");
          log.error(err, err.stack);
        }
      })
}

有没有办法用 Sinon 或 aws-sdk-mock 模拟 SES？

4

6 回答 6

1

我在这里的回答SES不是. 也许您可以在单元测试中为其他客户调整我的工作示例。DynamoDB.DocumentClientSQSSESaws-sdk

我只是花了几个小时试图让 AWS SQS 模拟工作，而没有诉诸在函数aws-sdk-mock中导入客户端的要求。aws-sdk

嘲笑 forAWS.DynamoDB.DocumentClient很容易，但AWS.SQS嘲笑让我难住了，直到我遇到使用rewire的建议。

我的 lambda 将错误消息移动到 SQS FailQueue（而不是让 Lambda 失败并将消息返回到常规队列进行重试，然后在 maxRetries 之后返回 DeadLetterQueue）。模拟以下 SQS 方法所需的单元测试：

  • SQS.getQueueUrl
  • SQS.sendMessage
  • SQS.deleteMessage

我将尽量使这个示例代码保持简洁，同时仍然包含所有相关部分：

我的 AWS Lambda (index.js) 的片段：

const AWS = require('aws-sdk');
AWS.config.update({region:'eu-west-1'});
const docClient = new AWS.DynamoDB.DocumentClient();
const sqs = new AWS.SQS({ apiVersion: '2012-11-05' });
// ...snip

简化的 Lambda 事件记录 (event.json)

{
    "valid": {
        "Records": [{
            "messageId": "c292410d-3b27-49ae-8e1f-0eb155f0710b",
            "receiptHandle": "AQEBz5JUoLYsn4dstTAxP7/IF9+T1S994n3FLkMvMmAh1Ut/Elpc0tbNZSaCPYDvP+mBBecVWmAM88SgW7iI8T65Blz3cXshP3keWzCgLCnmkwGvDHBYFVccm93yuMe0i5W02jX0s1LJuNVYI1aVtyz19IbzlVksp+z2RxAX6zMhcTy3VzusIZ6aDORW6yYppIYtKuB2G4Ftf8SE4XPzXo5RCdYirja1aMuh9DluEtSIW+lgDQcHbhIZeJx0eC09KQGJSF2uKk2BqTGvQrknw0EvjNEl6Jv56lWKyFT78K3TLBy2XdGFKQTsSALBNtlwFd8ZzcJoMaUFpbJVkzuLDST1y4nKQi7MK58JMsZ4ujZJnYvKFvgtc6YfWgsEuV0QSL9U5FradtXg4EnaBOnGVTFrbE18DoEuvUUiO7ZQPO9auS4=",
            "body": "{ \"key1\": \"value 1\", \"key2\": \"value 2\", \"key3\": \"value 3\", \"key4\": \"value 4\", \"key5\": \"value 5\" }",
            "attributes": {
                "ApproximateReceiveCount": "1",
                "SentTimestamp": "1536763724607",
                "SenderId": "AROAJAAXYIAN46PWMV46S:steve.goossens@bbc.co.uk",
                "ApproximateFirstReceiveTimestamp": "1536763724618"
            },
            "messageAttributes": {},
            "md5OfBody": "e5b16f3a468e6547785a3454cfb33293",
            "eventSource": "aws:sqs",
            "eventSourceARN": "arn:aws:sqs:eu-west-1:123456789012:sqs-queue-name",
            "awsRegion": "eu-west-1"
        }]
    }
}

删节的单元测试文件（test/index.test.js）：

const AWS = require('aws-sdk');
const expect = require('chai').expect;
const LamdbaTester = require('lambda-tester');
const rewire = require('rewire');
const sinon = require('sinon');

const event = require('./event');
const lambda = rewire('../index');

let sinonSandbox;

function mockGoodSqsMove() {
    const promiseStubSqs = sinonSandbox.stub().resolves({});
    const sqsMock = {
        getQueueUrl: () => ({ promise: sinonSandbox.stub().resolves({ QueueUrl: 'queue-url' }) }),
        sendMessage: () => ({ promise: promiseStubSqs }),
        deleteMessage: () => ({ promise: promiseStubSqs })
    }
    lambda.__set__('sqs', sqsMock);
}

describe('handler', function () {
    beforeEach(() => {
        sinonSandbox = sinon.createSandbox();
    });

    afterEach(() => {
        sinonSandbox.restore();
    });

    describe('when SQS message is in dedupe cache', function () {
        beforeEach(() => {
            // mock SQS
            mockGoodSqsMove();
            // mock DynamoDBClient
            const promiseStub = sinonSandbox.stub().resolves({'Item': 'something'});
            sinonSandbox.stub(AWS.DynamoDB.DocumentClient.prototype, 'get').returns({ promise: promiseStub });
        });

        it('should return an error for a duplicate message', function () {
            return LamdbaTester(lambda.handler)
                .event(event.valid)
                .expectReject((err, additional) => {
                    expect(err).to.have.property('message', 'Duplicate message: {"Item":"something"}');
                });
        });
    });
});
于 2019-06-17T15:23:54.673 回答
0

您需要使用prototypeinAWS来存根它：

import AWS from 'aws-sdk';

const sandbox = sinon.createSandbox();
sandbox.stub(AWS.prototype, 'SES').returns({
  sendEmail: () => {
    return true;
  }
});
于 2019-03-22T12:10:46.407 回答
0

该错误似乎表明AWS正在导入为undefined.

可能是您的 ES6 编译器没有自动转换这一行：

import AWS from 'aws-sdk';

...进入 into 中的所有内容的aws-sdk导入AWS。

将其更改为：

import * as AWS from 'aws-sdk';

...这可能会解决问题。

（免责声明：我无法在使用Babelv7 编译并自动处理任一方法的环境中重现该错误）

于 2019-03-22T18:15:14.370 回答
0

使用 require & 而不使用原型。这对我模拟 DynamoDB 很有用。

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

const sandbox = sinon.createSandbox();

this.awsStub = sandbox.stub(aws, 'DynamoDB').returns({
  query: function() {
    return {
      promise: function() {
        return {
          Items: []
        };
      }
    };
  }
});

包：

“aws-sdk”：“^2.453.0”

“诗乃”：“^7.3.2”

于 2019-05-14T12:25:09.987 回答
0

通过执行以下操作，我能够使用 awk-sdk-mock：

测试班

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

...

    AWSMock.mock('SES', 'sendRawEmail', mockSendEmail);
// call method that needs to mock send an email goes below
    sendEmail(to, from, subject, body, callback);

function mockSendEmail(params, callback) {
    console.log('mock email');
    return callback({
        MessageId: '1234567',
    });
}

实际班级

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


function sendEmail(to, from, subject, body, callback) {
    let addresses = to;
    if (!Array.isArray(addresses)) {
        addresses = [addresses];
    }
    let replyTo = [];
    if (from) {
        replyTo.push(from);
    }

    let data = {
        to: addresses,
        replyTo,
        subject,
        text: body,
    };

    nodemailer.createTransport({ SES: new aws.SES({ apiVersion: '2010-12-01' }) }).sendMail(data, callback);
}
于 2020-06-16T16:19:24.593 回答
0 
const AWS = require('aws-sdk');
...
const sandbox = sinon.createSandbox();
sandbox.stub(AWS, 'SES').returns({
    sendRawEmail: () => {
        console.log("My sendRawEmail");
        return {
            promise: function () {
                return {
                    MessageId: '987654321'
                };
            }
        };
    }
});
let ses = new AWS.SES({ region: 'us-east-1' });
let result = ses.sendRawEmail(params).promise();
于 2021-04-01T22:53:57.333 回答