Stubing AWS SQS 客户端

Question

我正在尝试打断 aws-sdk/client-sqs 电话。我对 aws 和存根都是新手，我在文档中找不到很多关于使用 mocha/chai

存根的信息

我的档案

import { SQSClient, ReceiveMessageCommand, DeleteMessageCommand } from '@aws-sdk/client-sqs'
const sqsClient = new SQSClient()

const params = {
    AttributeNames: ['SentTimestamp'],
    MaxNumberOfMessages: 10,
    QueueUrl: paymentsUrl,
    //....more params
}

const getPolledMessages = async sqsClient => {

    const data = await sqsClient.send(new ReceiveMessageCommand(params))
    //....continue to do stuff with the data
}

我的测试如下。我大量借鉴了这个 test 存根 @aws-sdk/client-ses

import { ReceiveMessageCommand, SQSClient } from '@aws-sdk/client-sqs'
import { expect } from 'chai'


describe('mock sqs', () => {
  let stub;
  before(() => {
    stub = sinon.stub(SQSClient.prototype, 'ReceiveMessageCommand')
  })
  after(() => stub.restore())

  it('can send', async () => {
    await SQSClient.send(new ReceiveMessageCommand(params))
    expect(stub.calledOnce).to.be.true
  })
})

当前出现以下错误

 TypeError: Cannot stub non-existent property ReceiveMessageCommand
 at Function.stub (node_modules/sinon/lib/sinon/stub.js:73:15)
 at Sandbox.stub (node_modules/sinon/lib/sinon/sandbox.js:333:37)

Answer 1

您可以 stub out dependencies with link seams.. We should use proxyquire 打包来做到这一点。

例如

index.ts:

import { SQSClient, ReceiveMessageCommand } from '@aws-sdk/client-sqs';
const sqsClient = new SQSClient({ region: 'REGION' });

const params = {
  AttributeNames: ['SentTimestamp'],
  MaxNumberOfMessages: 10,
  QueueUrl: 'paymentsUrl',
};

export const getPolledMessages = async () => {
  const data = await sqsClient.send(new ReceiveMessageCommand(params));
};

index.test.ts:

import proxyquire from 'proxyquire';
import sinon from 'sinon';

describe('68017252', () => {
  it('should pass', async () => {
    const sqsClientInstance = {
      send: sinon.stub(),
    };
    const SQSClient = sinon.stub().returns(sqsClientInstance);
    const ReceiveMessageCommand = sinon.stub();
    const { getPolledMessages } = proxyquire('./', {
      '@aws-sdk/client-sqs': {
        SQSClient,
        ReceiveMessageCommand,
      },
    });
    await getPolledMessages();
    sinon.assert.calledWithExactly(SQSClient, { region: 'REGION' });
    sinon.assert.calledOnce(sqsClientInstance.send);
    sinon.assert.calledWithExactly(ReceiveMessageCommand, {
      AttributeNames: ['SentTimestamp'],
      MaxNumberOfMessages: 10,
      QueueUrl: 'paymentsUrl',
    });
  });
});

测试结果：

  68017252
    ✓ should pass (3331ms)


  1 passing (3s)

----------|---------|----------|---------|---------|-------------------
File      | % Stmts | % Branch | % Funcs | % Lines | Uncovered Line #s 
----------|---------|----------|---------|---------|-------------------
All files |     100 |      100 |     100 |     100 |                   
 index.ts |     100 |      100 |     100 |     100 |                   
----------|---------|----------|---------|---------|-------------------

Stubing AWS SQS 客户端

Stubing AWS SQS client

javascript

mocha.js

sinon

chai

aws-sdk-js