Question

能否请您查看下面的代码并告诉我它有什么问题？该代码会在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();
});

Answer 1

我终于找到了可行的变体：

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');
  });
});

使用Jest模拟AWS-SDK-mock与AWS-SDK-Mock

1 个答案: