1

我正在尝试为与 DynamoDB 通信的 Lambda 函数编写单元测试。我正在使用moto，但它没有嘲笑任何东西。每当我在 boto3 中调用某些内容时，它都会使用我的 AWS CLI 配置文件与实际 API 进行通信，而不是模拟 API。为什么会这样？

这是代码：

### Unit test for the visitorCounterLambda function

from visitorCounterLambda import handler
import boto3
from moto import mock_dynamodb2


def setUp(self):
  #pass
  self.region = 'us-east-2'

@mock_dynamodb2
def test_handler():
  dynamodb = boto3.client('dynamodb')
  ddbTableName = "myDDBtable"

  # table = dynamodb.create_table(
  #   TableName = ddbTableName,
  #   BillingMode='PAY_PER_REQUEST',
  #   AttributeDefinitions=[
  #       {
  #           'AttributeName': 'id',
  #           'AttributeType': 'S'
  #       },
  #   ],
  #   KeySchema=[
  #       {
  #           'AttributeName': 'id',
  #           'KeyType': 'HASH'
  #       },
  #   ]
  # )
  tablesListed = dynamodb.list_tables()
  print(tablesListed)



if __name__ == '__main__':
    test_handler()

print(tablesListed)从我的实际帐户返回我的实际表格。如果我取消注释该create_table命令，它也会在我的 AWS 账户中创建表。

我在这里想念什么？谢谢

4

1 回答 1

1

我发现问题出在from visitorCounterLambda import handler部件上，因为该脚本在导入时已经建立了一个 boto3 客户端，因此mock无法破坏它。正确的做法在Moto 文档中的“非常重要 - 推荐用法”下进行了概述。你应该先建立@mock_dynamodb2然后import你的外部资源进入功能。

例子：

import boto3
from moto import mock_dynamodb2

@mock_dynamodb2
def test_handler():
  from visitorCounterLambda import handler
  dynamodb = boto3.client('dynamodb')

  ## do your magic here

  tablesListed = dynamodb.list_tables()
  print(tablesListed)
于 2020-06-07T02:04:02.377 回答