使用Moto +无服务器模拟DynamoDB

时间:2018-09-07 09:26:04

标签: amazon-web-services amazon-dynamodb boto3 serverless-framework moto

我正在尝试使用AWS无服务器框架为无服务器应用程序编写测试。我面临一个奇怪的问题。每当我尝试使用moto模拟S3或DynamoDB时,它都将不起作用。实际上,boto3调用不是去模拟,而是转到我的AWS账户并尝试在其中执行操作。

这不是可取的行为。你能帮忙吗?

示例代码:

import datetime
import boto3
import uuid
import os
from moto import mock_dynamodb2
from unittest import mock, TestCase
from JobEngine.job_engine import check_duplicate


class TestJobEngine(TestCase):
    @mock.patch.dict(os.environ, {'IN_QUEUE_URL': 'mytemp'})
    @mock.patch('JobEngine.job_engine.logger')
    @mock_dynamodb2
    def test_check_duplicate(self, mock_logger):
        id = 'ABCD123'  

        db = boto3.resource('dynamodb', 'us-east-1')

        table = db.create_table(
            TableName='my_table',
            KeySchema=[
                {
                    'AttributeName': 'id',
                    'KeyType': 'HASH'
                }
            ],
            AttributeDefinitions=[
                {
                    'AttributeName': 'id',
                    'AttributeType': 'S'
                }
            ],
            ProvisionedThroughput={
                'ReadCapacityUnits': 1,
                'WriteCapacityUnits': 1
            }
        )


    table.meta.client.get_waiter('table_exists').wait(TableName='my_table')

        table.put_item(
        Item={
            'id': {'S': id},
           ... other data ...
        }
    )

    res = check_duplicate(id)
    self.assertTrue(mock_logger.info.called)
    self.assertEqual(res, True, 'True')

请参见上面的代码,我试图将记录插入表中,然后调用一个函数,该函数将验证表中是否已存在指定的ID。在这里,运行此代码时我得到一个错误表。

如果我禁用网络,则会收到错误消息:

botocore.exceptions.EndpointConnectionError: Could not connect to the endpoint URL: "https://dynamodb.us-east-1.amazonaws.com/"

如果我们尝试模拟,我无法理解为什么要尝试连接到AWS。

1 个答案:

答案 0 :(得分:2)

我做了一些挖掘,终于设法解决了这个问题。

请参见https://github.com/spulec/moto/issues/1793

此问题是由于boto和moto之间不兼容。转过来,当我们将botocore降级到1.10.84时,一切正常