Насмешка над обещаниями aws-sdk с помощью aws-sdk-mock с использованием jest - PullRequest
0 голосов
/ 17 мая 2019

пожалуйста, посмотрите код ниже и скажите, что с ним не так?Код истекает через 5 секунд, но я ожидаю, что он будет работать нормально, согласно официальному описанию.

Кто-нибудь видит, что в корне неправильно?

import * as AWS from "aws-sdk-mock";
import * as _AWS from "aws-sdk"; 

beforeAll(async (done) => {
  //get requires env vars
 });

describe("the module", () => {

  it("should read from the database", async () => {
    AWS.mock('DynamoDB.DocumentClient', 'get', (error, callback) => { callback(null, "got it")});
    expect(await (new _AWS.DynamoDB.DocumentClient()).get({TableName:"", Key: {pk: "foo", sk: "bar"}}).promise()).toBe("got it");
  });
});

afterAll(() => {
  AWS.restore();
});

1 Ответ

0 голосов
/ 21 мая 2019

я наконец нашел рабочий вариант:

import * as AWSMock from "aws-sdk-mock";
import * as AWS from "aws-sdk"; 
import { GetItemInput } from "aws-sdk/clients/dynamodb";

beforeAll(async (done) => {
  //get requires env vars
  done();
 });

describe("the module", () => {

  it("should mock getItem from DynamoDB", async () => {

    AWSMock.setSDKInstance(AWS);
    AWSMock.mock('DynamoDB', 'getItem', (params: GetItemInput, callback: Function) => {
      console.log('DynamoDB', 'getItem', 'mock called');
      callback(null, {pk: "foo", sk: "bar"});
    })

    let input:GetItemInput = { TableName: '', Key: {} };
    const dynamodb = new AWS.DynamoDB({apiVersion: '2012-08-10'});
    expect(await dynamodb.getItem(input).promise()).toStrictEqual( { pk: 'foo', sk: 'bar' });

    AWSMock.restore('DynamoDB');
  });

  it("should mock reading from DocumentClient", async () => {

    AWSMock.setSDKInstance(AWS);
    AWSMock.mock('DynamoDB.DocumentClient', 'get', (params: GetItemInput, callback: Function) => {
      console.log('DynamoDB.DocumentClient', 'get', 'mock called');
      callback(null, {pk: "foo", sk: "bar"});
    })

    let input:GetItemInput = { TableName: '', Key: {} };
    const client = new AWS.DynamoDB.DocumentClient({apiVersion: '2012-08-10'});
    expect(await client.get(input).promise()).toStrictEqual( { pk: 'foo', sk: 'bar' });

    AWSMock.restore('DynamoDB.DocumentClient');
  });
});
...