Stubber.add_response for query method validate against an invalid schema about botocore

[tim-finnigan]

Thanks for reaching out. It looks like there are issues with how you are using the Stubber class. Here is the documentation for reference: https://botocore.amazonaws.com/v1/documentation/api/latest/reference/stubber.html. I think example below addresses what you're trying to do:

import boto3
from botocore.stub import Stubber

# Set up the test data
document_id = "8048ccd9-3b7e-4bb1-a6e9-ada69204a299"
document_version = 1298093434.865334
document_raw_s3_uri = "http://www.saunders.com/app/list/appcategory.php"

# Create a DynamoDB resource
dynamodb_resource = boto3.resource("dynamodb")

# Create a DynamoDB Stubber
dynamodb_stubber = Stubber(dynamodb_resource.meta.client)

# Set up the expected response for the query command
expected_params = {
    "TableName": "stubber-test",
    "KeyConditionExpression": "partition_key = :partition_key",
    "ExpressionAttributeValues": {
        ":partition_key": {"S": "my-partition-key-value"}
    }
}

response = {
    "Items": [
        {
            "DocumentId": {"S": document_id},
            "DocumentVersion": {"N": str(document_version)},
            "RawDocumentS3Uri": {"S": document_raw_s3_uri}
        }
    ],
    "Count": 1,
    "ScannedCount": 1,
    "LastEvaluatedKey": {
        "DocumentId": {"S": document_id}
    }
}

# Add the expected response to the Stubber
dynamodb_stubber.add_response("query", response, expected_params)

# Activate the Stubber
with dynamodb_stubber:
    # Call the query command
    result = dynamodb_resource.Table("stubber-test").query(
        KeyConditionExpression="partition_key = :partition_key",
        ExpressionAttributeValues={
            ':partition_key': {'S': 'my-partition-key-value'}
        }
    )

    # Assert that the response matches the expected response
    assert result == response

[TiphaineLAURENT]

Oh I see. It seems I was missing the type in the response format ({"<type>": <value>}).
Thank you very much it is very helpful

[github-actions]

This issue is now closed. Comments on closed issues are hard for our team to see.
If you need more assistance, please open a new issue that references this one.