如何对 python 中的嵌套 DynamoDB 调用进行单元测试？

Question

我有一个 DynamoDB 助手 class 定义如下：

class DynamoDbHelper:
    def __init__(self, table_name, key_and_range, AttributeDefinitions, GlobalSecondaryIndexes):
        self.get_ddb_conn(table_name, key_and_range, AttributeDefinitions, GlobalSecondaryIndexes)

    def get_ddb_conn(self, table_name, key_and_range, AttributeDefinitions, GlobalSecondaryIndexes):
        """
        Set client and resource for dynamoDB
        client: Get list of tables
        resource: Create and get tables
        """
        self.resource = boto3.resource('dynamodb', aws_access_key_id=Storage_Credentials.access_key,
                                       aws_secret_access_key=Storage_Credentials.secret_access_key,
                                       region_name="us-east-1")
        self.client = boto3.client('dynamodb', aws_access_key_id=Storage_Credentials.access_key,
                                   aws_secret_access_key=Storage_Credentials.secret_access_key,
                                   region_name="us-east-1")

        tables_list = self.client.list_tables()['TableNames']
        try:
            # First we look if table of the given table name already exists in dynamo_db, if not attempt to create one.
            # Else updates are done to existing table
            if table_name in tables_list:
                self.table = self.resource.Table(table_name)
                self.logger.info('Successfully found table: {}'.format(table_name))
            else:
                self.logger.info('Attempting to create missing dynamoDB table: {}'.format(table_name))
                table = self.resource.create_table(GlobalSecondaryIndexes=GlobalSecondaryIndexes,
                                                   TableName=table_name,
                                                   KeySchema=key_and_range,
                                                   AttributeDefinitions=AttributeDefinitions,
                                                   ProvisionedThroughput=ProvisionedThroughput)
                # Wait for the table to be created in dynamodb.
                table.meta.client.get_waiter('table_exists').wait(TableName=table_name)
                self.logger.info('Successfully created table: {}'.format(table_name))
                self.table = self.resource.Table(table_name)

        except Exception as e:
            raise Exception(e)

在这个 class 中，还有其他简单的函数，如get_table 、 put_record 、 update_record等可以使用。

在这个 DynamoDb 助手 class 中，在init中调用 ddb 连接，在get_db_conn中，我们使用存储访问密钥和密钥的存储凭证

我面临的问题是我没有正确的 mocking 凭据。 我还尝试直接传递存储凭据和 mocking 它们也失败了。 所以每次

前测试-->

class DynamoDBTest(unittest.TestCase):
    """
    Defines class for all Dynamodb tests
    """
    def setUp(self):
 
        self.ddb_helper = DynamoDbHelper(table_name='TEST_TABLE', logger=logger, key_and_range=MagicMock(),
                                         AttributeDefinitions=MagicMock(),
                                         GlobalSecondaryIndexes=MagicMock())
        DynamoDbHelper.get_db_conn = MagicMock()

每次初始化DynamoDbHelper时，都会使用真实凭据调用get_ddb_conn 。 我不希望有真正的电话。 只想测试此 class 中的方法，可能使用虚拟凭据，或者在初始化 DynamoDbHelper 时是否有办法修补或模拟 get_ddb_conn 调用或模拟凭据。

====简单测试记录====

def test_dynamo_db_activity(self):


        attrs = {
            "name": "ABC",
             "status": "testing",    
        }
        result = self.ddb_helper.put_record('TEST-ID', attrs)
        self.assertTrue(result)

Answer 1

在setUp中，您正在初始化ddb_helper 。 在该代码中，您可以使用unittest mocking 来修补您不想测试的 function。 在该代码中，您可以使用补丁语句包装它初始化的位置，然后设置您自己的模拟客户端和模拟资源（以及 get_ddb_conn function 可以做的任何其他事情，例如：

with patch('your_module.DynamoDbHelper.get_ddb_conn') as mock_get_ddb_conn:
    self.ddb_helper = DynamoDbHelper(table_name='TEST_TABLE', logger=logger, key_and_range=MagicMock(),
                                     AttributeDefinitions=MagicMock(),
                                     GlobalSecondaryIndexes=MagicMock())
    mock_client = MagicMock()
    ddb_helper.client = mock_client

To test the get_ddb_conn function itself, you'd do a similar approach where you mock the __init__ function instead when initializing the DynamoDbHelper and then call the real get_ddb_conn function in your test (you'd likely want to patch other things like boto3 usage in that case 所以它不会进行任何真正的 boto3 调用）。