from __future__ import unicode_literals, print_function from datetime import datetime from decimal import Decimal import boto import boto3 from boto3.dynamodb.conditions import Attr, Key import re import sure # noqa from moto import mock_dynamodb2, mock_dynamodb2_deprecated from moto.dynamodb2 import dynamodb_backend2, dynamodb_backends2 from boto.exception import JSONResponseError from botocore.exceptions import ClientError, ParamValidationError from tests.helpers import requires_boto_gte import moto.dynamodb2.comparisons import moto.dynamodb2.models from nose.tools import assert_raises try: import boto.dynamodb2 except ImportError: print("This boto version is not supported") @requires_boto_gte("2.9") @mock_dynamodb2_deprecated def test_list_tables(): name = "TestTable" # Should make tables properly with boto dynamodb_backend2.create_table( name, schema=[ {"KeyType": "HASH", "AttributeName": "forum_name"}, {"KeyType": "RANGE", "AttributeName": "subject"}, ], ) conn = boto.dynamodb2.connect_to_region( "us-east-1", aws_access_key_id="ak", aws_secret_access_key="sk" ) assert conn.list_tables()["TableNames"] == [name] @requires_boto_gte("2.9") @mock_dynamodb2_deprecated def test_list_tables_layer_1(): # Should make tables properly with boto dynamodb_backend2.create_table( "test_1", schema=[{"KeyType": "HASH", "AttributeName": "name"}] ) dynamodb_backend2.create_table( "test_2", schema=[{"KeyType": "HASH", "AttributeName": "name"}] ) conn = boto.dynamodb2.connect_to_region( "us-east-1", aws_access_key_id="ak", aws_secret_access_key="sk" ) res = conn.list_tables(limit=1) expected = {"TableNames": ["test_1"], "LastEvaluatedTableName": "test_1"} res.should.equal(expected) res = conn.list_tables(limit=1, exclusive_start_table_name="test_1") expected = {"TableNames": ["test_2"]} res.should.equal(expected) @requires_boto_gte("2.9") @mock_dynamodb2_deprecated def test_describe_missing_table(): conn = boto.dynamodb2.connect_to_region( "us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk" ) with assert_raises(JSONResponseError): conn.describe_table("messages") @requires_boto_gte("2.9") @mock_dynamodb2 def test_list_table_tags(): name = "TestTable" conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) conn.create_table( TableName=name, KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table_description = conn.describe_table(TableName=name) arn = table_description["Table"]["TableArn"] # Tag table tags = [ {"Key": "TestTag", "Value": "TestValue"}, {"Key": "TestTag2", "Value": "TestValue2"}, ] conn.tag_resource(ResourceArn=arn, Tags=tags) # Check tags resp = conn.list_tags_of_resource(ResourceArn=arn) assert resp["Tags"] == tags # Remove 1 tag conn.untag_resource(ResourceArn=arn, TagKeys=["TestTag"]) # Check tags resp = conn.list_tags_of_resource(ResourceArn=arn) assert resp["Tags"] == [{"Key": "TestTag2", "Value": "TestValue2"}] @requires_boto_gte("2.9") @mock_dynamodb2 def test_list_table_tags_empty(): name = "TestTable" conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) conn.create_table( TableName=name, KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table_description = conn.describe_table(TableName=name) arn = table_description["Table"]["TableArn"] tags = [{"Key": "TestTag", "Value": "TestValue"}] # conn.tag_resource(ResourceArn=arn, # Tags=tags) resp = conn.list_tags_of_resource(ResourceArn=arn) assert resp["Tags"] == [] @requires_boto_gte("2.9") @mock_dynamodb2 def test_list_table_tags_paginated(): name = "TestTable" conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) conn.create_table( TableName=name, KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table_description = conn.describe_table(TableName=name) arn = table_description["Table"]["TableArn"] for i in range(11): tags = [{"Key": "TestTag%d" % i, "Value": "TestValue"}] conn.tag_resource(ResourceArn=arn, Tags=tags) resp = conn.list_tags_of_resource(ResourceArn=arn) assert len(resp["Tags"]) == 10 assert "NextToken" in resp.keys() resp2 = conn.list_tags_of_resource(ResourceArn=arn, NextToken=resp["NextToken"]) assert len(resp2["Tags"]) == 1 assert "NextToken" not in resp2.keys() @requires_boto_gte("2.9") @mock_dynamodb2 def test_list_not_found_table_tags(): conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) arn = "DymmyArn" try: conn.list_tags_of_resource(ResourceArn=arn) except ClientError as exception: assert exception.response["Error"]["Code"] == "ResourceNotFoundException" @requires_boto_gte("2.9") @mock_dynamodb2 def test_item_add_empty_string_exception(): name = "TestTable" conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) conn.create_table( TableName=name, KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) with assert_raises(ClientError) as ex: conn.put_item( TableName=name, Item={ "forum_name": {"S": "LOLCat Forum"}, "subject": {"S": "Check this out!"}, "Body": {"S": "http://url_to_lolcat.gif"}, "SentBy": {"S": ""}, "ReceivedTime": {"S": "12/9/2011 11:36:03 PM"}, }, ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.exception.response["Error"]["Message"].should.equal( "One or more parameter values were invalid: An AttributeValue may not contain an empty string" ) @requires_boto_gte("2.9") @mock_dynamodb2 def test_update_item_with_empty_string_exception(): name = "TestTable" conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) conn.create_table( TableName=name, KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) conn.put_item( TableName=name, Item={ "forum_name": {"S": "LOLCat Forum"}, "subject": {"S": "Check this out!"}, "Body": {"S": "http://url_to_lolcat.gif"}, "SentBy": {"S": "test"}, "ReceivedTime": {"S": "12/9/2011 11:36:03 PM"}, }, ) with assert_raises(ClientError) as ex: conn.update_item( TableName=name, Key={"forum_name": {"S": "LOLCat Forum"}}, UpdateExpression="set Body=:Body", ExpressionAttributeValues={":Body": {"S": ""}}, ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.exception.response["Error"]["Message"].should.equal( "One or more parameter values were invalid: An AttributeValue may not contain an empty string" ) @requires_boto_gte("2.9") @mock_dynamodb2 def test_query_invalid_table(): conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) try: conn.query( TableName="invalid_table", KeyConditionExpression="index1 = :partitionkeyval", ExpressionAttributeValues={":partitionkeyval": {"S": "test"}}, ) except ClientError as exception: assert exception.response["Error"]["Code"] == "ResourceNotFoundException" @requires_boto_gte("2.9") @mock_dynamodb2 def test_scan_returns_consumed_capacity(): name = "TestTable" conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) conn.create_table( TableName=name, KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) conn.put_item( TableName=name, Item={ "forum_name": {"S": "LOLCat Forum"}, "subject": {"S": "Check this out!"}, "Body": {"S": "http://url_to_lolcat.gif"}, "SentBy": {"S": "test"}, "ReceivedTime": {"S": "12/9/2011 11:36:03 PM"}, }, ) response = conn.scan(TableName=name) assert "ConsumedCapacity" in response assert "CapacityUnits" in response["ConsumedCapacity"] assert response["ConsumedCapacity"]["TableName"] == name @requires_boto_gte("2.9") @mock_dynamodb2 def test_put_item_with_special_chars(): name = "TestTable" conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) conn.create_table( TableName=name, KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) conn.put_item( TableName=name, Item={ "forum_name": {"S": "LOLCat Forum"}, "subject": {"S": "Check this out!"}, "Body": {"S": "http://url_to_lolcat.gif"}, "SentBy": {"S": "test"}, "ReceivedTime": {"S": "12/9/2011 11:36:03 PM"}, '"': {"S": "foo"}, }, ) @requires_boto_gte("2.9") @mock_dynamodb2 def test_put_item_with_streams(): name = "TestTable" conn = boto3.client( "dynamodb", region_name="us-west-2", aws_access_key_id="ak", aws_secret_access_key="sk", ) conn.create_table( TableName=name, KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], StreamSpecification={ "StreamEnabled": True, "StreamViewType": "NEW_AND_OLD_IMAGES", }, ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) conn.put_item( TableName=name, Item={ "forum_name": {"S": "LOLCat Forum"}, "subject": {"S": "Check this out!"}, "Body": {"S": "http://url_to_lolcat.gif"}, "SentBy": {"S": "test"}, "Data": {"M": {"Key1": {"S": "Value1"}, "Key2": {"S": "Value2"}}}, }, ) result = conn.get_item(TableName=name, Key={"forum_name": {"S": "LOLCat Forum"}}) result["Item"].should.be.equal( { "forum_name": {"S": "LOLCat Forum"}, "subject": {"S": "Check this out!"}, "Body": {"S": "http://url_to_lolcat.gif"}, "SentBy": {"S": "test"}, "Data": {"M": {"Key1": {"S": "Value1"}, "Key2": {"S": "Value2"}}}, } ) table = dynamodb_backends2["us-west-2"].get_table(name) if not table: # There is no way to access stream data over the API, so this part can't run in server-tests mode. return len(table.stream_shard.items).should.be.equal(1) stream_record = table.stream_shard.items[0].record stream_record["eventName"].should.be.equal("INSERT") stream_record["dynamodb"]["SizeBytes"].should.be.equal(447) @requires_boto_gte("2.9") @mock_dynamodb2 def test_query_returns_consumed_capacity(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={"forum_name": "the-key", "subject": "123", "body": "some test message"} ) results = table.query(KeyConditionExpression=Key("forum_name").eq("the-key")) assert "ConsumedCapacity" in results assert "CapacityUnits" in results["ConsumedCapacity"] assert results["ConsumedCapacity"]["CapacityUnits"] == 1 @mock_dynamodb2 def test_basic_projection_expression_using_get_item(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={"forum_name": "the-key", "subject": "123", "body": "some test message"} ) table.put_item( Item={ "forum_name": "not-the-key", "subject": "123", "body": "some other test message", } ) result = table.get_item( Key={"forum_name": "the-key", "subject": "123"}, ProjectionExpression="body, subject", ) result["Item"].should.be.equal({"subject": "123", "body": "some test message"}) # The projection expression should not remove data from storage result = table.get_item(Key={"forum_name": "the-key", "subject": "123"}) result["Item"].should.be.equal( {"forum_name": "the-key", "subject": "123", "body": "some test message"} ) @mock_dynamodb2 def test_basic_projection_expressions_using_query(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={"forum_name": "the-key", "subject": "123", "body": "some test message"} ) table.put_item( Item={ "forum_name": "not-the-key", "subject": "123", "body": "some other test message", } ) # Test a query returning all items results = table.query( KeyConditionExpression=Key("forum_name").eq("the-key"), ProjectionExpression="body, subject", ) assert "body" in results["Items"][0] assert results["Items"][0]["body"] == "some test message" assert "subject" in results["Items"][0] table.put_item( Item={ "forum_name": "the-key", "subject": "1234", "body": "yet another test message", } ) results = table.query( KeyConditionExpression=Key("forum_name").eq("the-key"), ProjectionExpression="body", ) assert "body" in results["Items"][0] assert "subject" not in results["Items"][0] assert results["Items"][0]["body"] == "some test message" assert "body" in results["Items"][1] assert "subject" not in results["Items"][1] assert results["Items"][1]["body"] == "yet another test message" # The projection expression should not remove data from storage results = table.query(KeyConditionExpression=Key("forum_name").eq("the-key")) assert "subject" in results["Items"][0] assert "body" in results["Items"][1] assert "forum_name" in results["Items"][1] @mock_dynamodb2 def test_basic_projection_expressions_using_scan(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={"forum_name": "the-key", "subject": "123", "body": "some test message"} ) table.put_item( Item={ "forum_name": "not-the-key", "subject": "123", "body": "some other test message", } ) # Test a scan returning all items results = table.scan( FilterExpression=Key("forum_name").eq("the-key"), ProjectionExpression="body, subject", ) assert "body" in results["Items"][0] assert results["Items"][0]["body"] == "some test message" assert "subject" in results["Items"][0] table.put_item( Item={ "forum_name": "the-key", "subject": "1234", "body": "yet another test message", } ) results = table.scan( FilterExpression=Key("forum_name").eq("the-key"), ProjectionExpression="body" ) assert "body" in results["Items"][0] assert "subject" not in results["Items"][0] assert "forum_name" not in results["Items"][0] assert "body" in results["Items"][1] assert "subject" not in results["Items"][1] assert "forum_name" not in results["Items"][1] # The projection expression should not remove data from storage results = table.query(KeyConditionExpression=Key("forum_name").eq("the-key")) assert "subject" in results["Items"][0] assert "body" in results["Items"][1] assert "forum_name" in results["Items"][1] @mock_dynamodb2 def test_nested_projection_expression_using_get_item(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, "foo": "bar", } ) table.put_item( Item={ "forum_name": "key2", "nested": {"id": "id2", "incode": "code2"}, "foo": "bar", } ) # Test a get_item returning all items result = table.get_item( Key={"forum_name": "key1"}, ProjectionExpression="nested.level1.id, nested.level2", )["Item"] result.should.equal( {"nested": {"level1": {"id": "id1"}, "level2": {"id": "id2", "include": "all"}}} ) # Assert actual data has not been deleted result = table.get_item(Key={"forum_name": "key1"})["Item"] result.should.equal( { "foo": "bar", "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, } ) @mock_dynamodb2 def test_basic_projection_expressions_using_query(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={"forum_name": "the-key", "subject": "123", "body": "some test message"} ) table.put_item( Item={ "forum_name": "not-the-key", "subject": "123", "body": "some other test message", } ) # Test a query returning all items result = table.query( KeyConditionExpression=Key("forum_name").eq("the-key"), ProjectionExpression="body, subject", )["Items"][0] assert "body" in result assert result["body"] == "some test message" assert "subject" in result assert "forum_name" not in result table.put_item( Item={ "forum_name": "the-key", "subject": "1234", "body": "yet another test message", } ) items = table.query( KeyConditionExpression=Key("forum_name").eq("the-key"), ProjectionExpression="body", )["Items"] assert "body" in items[0] assert "subject" not in items[0] assert items[0]["body"] == "some test message" assert "body" in items[1] assert "subject" not in items[1] assert items[1]["body"] == "yet another test message" # The projection expression should not remove data from storage items = table.query(KeyConditionExpression=Key("forum_name").eq("the-key"))["Items"] assert "subject" in items[0] assert "body" in items[1] assert "forum_name" in items[1] @mock_dynamodb2 def test_nested_projection_expression_using_query(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, "foo": "bar", } ) table.put_item( Item={ "forum_name": "key2", "nested": {"id": "id2", "incode": "code2"}, "foo": "bar", } ) # Test a query returning all items result = table.query( KeyConditionExpression=Key("forum_name").eq("key1"), ProjectionExpression="nested.level1.id, nested.level2", )["Items"][0] assert "nested" in result result["nested"].should.equal( {"level1": {"id": "id1"}, "level2": {"id": "id2", "include": "all"}} ) assert "foo" not in result # Assert actual data has not been deleted result = table.query(KeyConditionExpression=Key("forum_name").eq("key1"))["Items"][ 0 ] result.should.equal( { "foo": "bar", "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, } ) @mock_dynamodb2 def test_basic_projection_expressions_using_scan(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={"forum_name": "the-key", "subject": "123", "body": "some test message"} ) table.put_item( Item={ "forum_name": "not-the-key", "subject": "123", "body": "some other test message", } ) # Test a scan returning all items results = table.scan( FilterExpression=Key("forum_name").eq("the-key"), ProjectionExpression="body, subject", )["Items"] results.should.equal([{"body": "some test message", "subject": "123"}]) table.put_item( Item={ "forum_name": "the-key", "subject": "1234", "body": "yet another test message", } ) results = table.scan( FilterExpression=Key("forum_name").eq("the-key"), ProjectionExpression="body" )["Items"] assert {"body": "some test message"} in results assert {"body": "yet another test message"} in results # The projection expression should not remove data from storage results = table.query(KeyConditionExpression=Key("forum_name").eq("the-key")) assert "subject" in results["Items"][0] assert "body" in results["Items"][1] assert "forum_name" in results["Items"][1] @mock_dynamodb2 def test_nested_projection_expression_using_scan(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, "foo": "bar", } ) table.put_item( Item={ "forum_name": "key2", "nested": {"id": "id2", "incode": "code2"}, "foo": "bar", } ) # Test a scan results = table.scan( FilterExpression=Key("forum_name").eq("key1"), ProjectionExpression="nested.level1.id, nested.level2", )["Items"] results.should.equal( [ { "nested": { "level1": {"id": "id1"}, "level2": {"include": "all", "id": "id2"}, } } ] ) # Assert original data is still there results = table.scan(FilterExpression=Key("forum_name").eq("key1"))["Items"] results.should.equal( [ { "forum_name": "key1", "foo": "bar", "nested": { "level1": {"att": "irrelevant", "id": "id1"}, "level2": {"include": "all", "id": "id2"}, "level3": {"id": "irrelevant"}, }, } ] ) @mock_dynamodb2 def test_basic_projection_expression_using_get_item_with_attr_expression_names(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "the-key", "subject": "123", "body": "some test message", "attachment": "something", } ) table.put_item( Item={ "forum_name": "not-the-key", "subject": "123", "body": "some other test message", "attachment": "something", } ) result = table.get_item( Key={"forum_name": "the-key", "subject": "123"}, ProjectionExpression="#rl, #rt, subject", ExpressionAttributeNames={"#rl": "body", "#rt": "attachment"}, ) result["Item"].should.be.equal( {"subject": "123", "body": "some test message", "attachment": "something"} ) @mock_dynamodb2 def test_basic_projection_expressions_using_query_with_attr_expression_names(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "the-key", "subject": "123", "body": "some test message", "attachment": "something", } ) table.put_item( Item={ "forum_name": "not-the-key", "subject": "123", "body": "some other test message", "attachment": "something", } ) # Test a query returning all items results = table.query( KeyConditionExpression=Key("forum_name").eq("the-key"), ProjectionExpression="#rl, #rt, subject", ExpressionAttributeNames={"#rl": "body", "#rt": "attachment"}, ) assert "body" in results["Items"][0] assert results["Items"][0]["body"] == "some test message" assert "subject" in results["Items"][0] assert results["Items"][0]["subject"] == "123" assert "attachment" in results["Items"][0] assert results["Items"][0]["attachment"] == "something" @mock_dynamodb2 def test_nested_projection_expression_using_get_item_with_attr_expression(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, "foo": "bar", } ) table.put_item( Item={ "forum_name": "key2", "nested": {"id": "id2", "incode": "code2"}, "foo": "bar", } ) # Test a get_item returning all items result = table.get_item( Key={"forum_name": "key1"}, ProjectionExpression="#nst.level1.id, #nst.#lvl2", ExpressionAttributeNames={"#nst": "nested", "#lvl2": "level2"}, )["Item"] result.should.equal( {"nested": {"level1": {"id": "id1"}, "level2": {"id": "id2", "include": "all"}}} ) # Assert actual data has not been deleted result = table.get_item(Key={"forum_name": "key1"})["Item"] result.should.equal( { "foo": "bar", "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, } ) @mock_dynamodb2 def test_nested_projection_expression_using_query_with_attr_expression_names(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, "foo": "bar", } ) table.put_item( Item={ "forum_name": "key2", "nested": {"id": "id2", "incode": "code2"}, "foo": "bar", } ) # Test a query returning all items result = table.query( KeyConditionExpression=Key("forum_name").eq("key1"), ProjectionExpression="#nst.level1.id, #nst.#lvl2", ExpressionAttributeNames={"#nst": "nested", "#lvl2": "level2"}, )["Items"][0] assert "nested" in result result["nested"].should.equal( {"level1": {"id": "id1"}, "level2": {"id": "id2", "include": "all"}} ) assert "foo" not in result # Assert actual data has not been deleted result = table.query(KeyConditionExpression=Key("forum_name").eq("key1"))["Items"][ 0 ] result.should.equal( { "foo": "bar", "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, } ) @mock_dynamodb2 def test_basic_projection_expressions_using_scan_with_attr_expression_names(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "the-key", "subject": "123", "body": "some test message", "attachment": "something", } ) table.put_item( Item={ "forum_name": "not-the-key", "subject": "123", "body": "some other test message", "attachment": "something", } ) # Test a scan returning all items results = table.scan( FilterExpression=Key("forum_name").eq("the-key"), ProjectionExpression="#rl, #rt, subject", ExpressionAttributeNames={"#rl": "body", "#rt": "attachment"}, ) assert "body" in results["Items"][0] assert "attachment" in results["Items"][0] assert "subject" in results["Items"][0] assert "form_name" not in results["Items"][0] # Test without a FilterExpression results = table.scan( ProjectionExpression="#rl, #rt, subject", ExpressionAttributeNames={"#rl": "body", "#rt": "attachment"}, ) assert "body" in results["Items"][0] assert "attachment" in results["Items"][0] assert "subject" in results["Items"][0] assert "form_name" not in results["Items"][0] @mock_dynamodb2 def test_nested_projection_expression_using_scan_with_attr_expression_names(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "key1", "nested": { "level1": {"id": "id1", "att": "irrelevant"}, "level2": {"id": "id2", "include": "all"}, "level3": {"id": "irrelevant"}, }, "foo": "bar", } ) table.put_item( Item={ "forum_name": "key2", "nested": {"id": "id2", "incode": "code2"}, "foo": "bar", } ) # Test a scan results = table.scan( FilterExpression=Key("forum_name").eq("key1"), ProjectionExpression="nested.level1.id, nested.level2", ExpressionAttributeNames={"#nst": "nested", "#lvl2": "level2"}, )["Items"] results.should.equal( [ { "nested": { "level1": {"id": "id1"}, "level2": {"include": "all", "id": "id2"}, } } ] ) # Assert original data is still there results = table.scan(FilterExpression=Key("forum_name").eq("key1"))["Items"] results.should.equal( [ { "forum_name": "key1", "foo": "bar", "nested": { "level1": {"att": "irrelevant", "id": "id1"}, "level2": {"include": "all", "id": "id2"}, "level3": {"id": "irrelevant"}, }, } ] ) @mock_dynamodb2 def test_put_item_returns_consumed_capacity(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") response = table.put_item( Item={"forum_name": "the-key", "subject": "123", "body": "some test message"} ) assert "ConsumedCapacity" in response @mock_dynamodb2 def test_update_item_returns_consumed_capacity(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={"forum_name": "the-key", "subject": "123", "body": "some test message"} ) response = table.update_item( Key={"forum_name": "the-key", "subject": "123"}, UpdateExpression="set body=:tb", ExpressionAttributeValues={":tb": "a new message"}, ) assert "ConsumedCapacity" in response assert "CapacityUnits" in response["ConsumedCapacity"] assert "TableName" in response["ConsumedCapacity"] @mock_dynamodb2 def test_get_item_returns_consumed_capacity(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. table = dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={"forum_name": "the-key", "subject": "123", "body": "some test message"} ) response = table.get_item(Key={"forum_name": "the-key", "subject": "123"}) assert "ConsumedCapacity" in response assert "CapacityUnits" in response["ConsumedCapacity"] assert "TableName" in response["ConsumedCapacity"] @mock_dynamodb2 def test_put_empty_item(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( AttributeDefinitions=[{"AttributeName": "structure_id", "AttributeType": "S"},], TableName="test", KeySchema=[{"AttributeName": "structure_id", "KeyType": "HASH"},], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) table = dynamodb.Table("test") with assert_raises(ClientError) as ex: table.put_item(Item={}) ex.exception.response["Error"]["Message"].should.equal( "One or more parameter values were invalid: Missing the key structure_id in the item" ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") @mock_dynamodb2 def test_put_item_nonexisting_hash_key(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( AttributeDefinitions=[{"AttributeName": "structure_id", "AttributeType": "S"},], TableName="test", KeySchema=[{"AttributeName": "structure_id", "KeyType": "HASH"},], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) table = dynamodb.Table("test") with assert_raises(ClientError) as ex: table.put_item(Item={"a_terribly_misguided_id_attribute": "abcdef"}) ex.exception.response["Error"]["Message"].should.equal( "One or more parameter values were invalid: Missing the key structure_id in the item" ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") @mock_dynamodb2 def test_put_item_nonexisting_range_key(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( AttributeDefinitions=[ {"AttributeName": "structure_id", "AttributeType": "S"}, {"AttributeName": "added_at", "AttributeType": "N"}, ], TableName="test", KeySchema=[ {"AttributeName": "structure_id", "KeyType": "HASH"}, {"AttributeName": "added_at", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) table = dynamodb.Table("test") with assert_raises(ClientError) as ex: table.put_item(Item={"structure_id": "abcdef"}) ex.exception.response["Error"]["Message"].should.equal( "One or more parameter values were invalid: Missing the key added_at in the item" ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") def test_filter_expression(): row1 = moto.dynamodb2.models.Item( None, None, None, None, { "Id": {"N": "8"}, "Subs": {"N": "5"}, "Desc": {"S": "Some description"}, "KV": {"SS": ["test1", "test2"]}, }, ) row2 = moto.dynamodb2.models.Item( None, None, None, None, { "Id": {"N": "8"}, "Subs": {"N": "10"}, "Desc": {"S": "A description"}, "KV": {"SS": ["test3", "test4"]}, }, ) # NOT test 1 filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "NOT attribute_not_exists(Id)", {}, {} ) filter_expr.expr(row1).should.be(True) # NOT test 2 filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "NOT (Id = :v0)", {}, {":v0": {"N": "8"}} ) filter_expr.expr(row1).should.be(False) # Id = 8 so should be false # AND test filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "Id > :v0 AND Subs < :v1", {}, {":v0": {"N": "5"}, ":v1": {"N": "7"}} ) filter_expr.expr(row1).should.be(True) filter_expr.expr(row2).should.be(False) # lowercase AND test filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "Id > :v0 and Subs < :v1", {}, {":v0": {"N": "5"}, ":v1": {"N": "7"}} ) filter_expr.expr(row1).should.be(True) filter_expr.expr(row2).should.be(False) # OR test filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "Id = :v0 OR Id=:v1", {}, {":v0": {"N": "5"}, ":v1": {"N": "8"}} ) filter_expr.expr(row1).should.be(True) # BETWEEN test filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "Id BETWEEN :v0 AND :v1", {}, {":v0": {"N": "5"}, ":v1": {"N": "10"}} ) filter_expr.expr(row1).should.be(True) # PAREN test filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "Id = :v0 AND (Subs = :v0 OR Subs = :v1)", {}, {":v0": {"N": "8"}, ":v1": {"N": "5"}}, ) filter_expr.expr(row1).should.be(True) # IN test filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "Id IN (:v0, :v1, :v2)", {}, {":v0": {"N": "7"}, ":v1": {"N": "8"}, ":v2": {"N": "9"}}, ) filter_expr.expr(row1).should.be(True) # attribute function tests (with extra spaces) filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "attribute_exists(Id) AND attribute_not_exists (User)", {}, {} ) filter_expr.expr(row1).should.be(True) filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "attribute_type(Id, :v0)", {}, {":v0": {"S": "N"}} ) filter_expr.expr(row1).should.be(True) # beginswith function test filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "begins_with(Desc, :v0)", {}, {":v0": {"S": "Some"}} ) filter_expr.expr(row1).should.be(True) filter_expr.expr(row2).should.be(False) # contains function test filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "contains(KV, :v0)", {}, {":v0": {"S": "test1"}} ) filter_expr.expr(row1).should.be(True) filter_expr.expr(row2).should.be(False) # size function test filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "size(Desc) > size(KV)", {}, {} ) filter_expr.expr(row1).should.be(True) # Expression from @batkuip filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "(#n0 < :v0 AND attribute_not_exists(#n1))", {"#n0": "Subs", "#n1": "fanout_ts"}, {":v0": {"N": "7"}}, ) filter_expr.expr(row1).should.be(True) # Expression from to check contains on string value filter_expr = moto.dynamodb2.comparisons.get_filter_expression( "contains(#n0, :v0)", {"#n0": "Desc"}, {":v0": {"S": "Some"}} ) filter_expr.expr(row1).should.be(True) filter_expr.expr(row2).should.be(False) @mock_dynamodb2 def test_query_filter(): client = boto3.client("dynamodb", region_name="us-east-1") dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "nested": { "M": { "version": {"S": "version1"}, "contents": {"L": [{"S": "value1"}, {"S": "value2"}]}, } }, }, ) client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app2"}, "nested": { "M": { "version": {"S": "version2"}, "contents": {"L": [{"S": "value1"}, {"S": "value2"}]}, } }, }, ) table = dynamodb.Table("test1") response = table.query(KeyConditionExpression=Key("client").eq("client1")) assert response["Count"] == 2 response = table.query( KeyConditionExpression=Key("client").eq("client1"), FilterExpression=Attr("app").eq("app2"), ) assert response["Count"] == 1 assert response["Items"][0]["app"] == "app2" response = table.query( KeyConditionExpression=Key("client").eq("client1"), FilterExpression=Attr("app").contains("app"), ) assert response["Count"] == 2 response = table.query( KeyConditionExpression=Key("client").eq("client1"), FilterExpression=Attr("nested.version").contains("version"), ) assert response["Count"] == 2 response = table.query( KeyConditionExpression=Key("client").eq("client1"), FilterExpression=Attr("nested.contents[0]").eq("value1"), ) assert response["Count"] == 2 @mock_dynamodb2 def test_query_filter_overlapping_expression_prefixes(): client = boto3.client("dynamodb", region_name="us-east-1") dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "nested": { "M": { "version": {"S": "version1"}, "contents": {"L": [{"S": "value1"}, {"S": "value2"}]}, } }, }, ) table = dynamodb.Table("test1") response = table.query( KeyConditionExpression=Key("client").eq("client1") & Key("app").eq("app1"), ProjectionExpression="#1, #10, nested", ExpressionAttributeNames={"#1": "client", "#10": "app"}, ) assert response["Count"] == 1 assert response["Items"][0] == { "client": "client1", "app": "app1", "nested": {"version": "version1", "contents": ["value1", "value2"]}, } @mock_dynamodb2 def test_scan_filter(): client = boto3.client("dynamodb", region_name="us-east-1") dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.put_item( TableName="test1", Item={"client": {"S": "client1"}, "app": {"S": "app1"}} ) table = dynamodb.Table("test1") response = table.scan(FilterExpression=Attr("app").eq("app2")) assert response["Count"] == 0 response = table.scan(FilterExpression=Attr("app").eq("app1")) assert response["Count"] == 1 response = table.scan(FilterExpression=Attr("app").ne("app2")) assert response["Count"] == 1 response = table.scan(FilterExpression=Attr("app").ne("app1")) assert response["Count"] == 0 @mock_dynamodb2 def test_scan_filter2(): client = boto3.client("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "N"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.put_item( TableName="test1", Item={"client": {"S": "client1"}, "app": {"N": "1"}} ) response = client.scan( TableName="test1", Select="ALL_ATTRIBUTES", FilterExpression="#tb >= :dt", ExpressionAttributeNames={"#tb": "app"}, ExpressionAttributeValues={":dt": {"N": str(1)}}, ) assert response["Count"] == 1 @mock_dynamodb2 def test_scan_filter3(): client = boto3.client("dynamodb", region_name="us-east-1") dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "N"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.put_item( TableName="test1", Item={"client": {"S": "client1"}, "app": {"N": "1"}, "active": {"BOOL": True}}, ) table = dynamodb.Table("test1") response = table.scan(FilterExpression=Attr("active").eq(True)) assert response["Count"] == 1 response = table.scan(FilterExpression=Attr("active").ne(True)) assert response["Count"] == 0 response = table.scan(FilterExpression=Attr("active").ne(False)) assert response["Count"] == 1 response = table.scan(FilterExpression=Attr("app").ne(1)) assert response["Count"] == 0 response = table.scan(FilterExpression=Attr("app").ne(2)) assert response["Count"] == 1 @mock_dynamodb2 def test_scan_filter4(): client = boto3.client("dynamodb", region_name="us-east-1") dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "N"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) table = dynamodb.Table("test1") response = table.scan( FilterExpression=Attr("epoch_ts").lt(7) & Attr("fanout_ts").not_exists() ) # Just testing assert response["Count"] == 0 @mock_dynamodb2 def test_scan_filter_should_not_return_non_existing_attributes(): table_name = "my-table" item = {"partitionKey": "pk-2", "my-attr": 42} # Create table res = boto3.resource("dynamodb", region_name="us-east-1") res.create_table( TableName=table_name, KeySchema=[{"AttributeName": "partitionKey", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "partitionKey", "AttributeType": "S"}], BillingMode="PAY_PER_REQUEST", ) table = res.Table(table_name) # Insert items table.put_item(Item={"partitionKey": "pk-1"}) table.put_item(Item=item) # Verify a few operations # Assert we only find the item that has this attribute table.scan(FilterExpression=Attr("my-attr").lt(43))["Items"].should.equal([item]) table.scan(FilterExpression=Attr("my-attr").lte(42))["Items"].should.equal([item]) table.scan(FilterExpression=Attr("my-attr").gte(42))["Items"].should.equal([item]) table.scan(FilterExpression=Attr("my-attr").gt(41))["Items"].should.equal([item]) # Sanity check that we can't find the item if the FE is wrong table.scan(FilterExpression=Attr("my-attr").gt(43))["Items"].should.equal([]) @mock_dynamodb2 def test_bad_scan_filter(): client = boto3.client("dynamodb", region_name="us-east-1") dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) table = dynamodb.Table("test1") # Bad expression try: table.scan(FilterExpression="client test") except ClientError as err: err.response["Error"]["Code"].should.equal("ValidationError") else: raise RuntimeError("Should have raised ResourceInUseException") @mock_dynamodb2 def test_create_table_pay_per_request(): client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], BillingMode="PAY_PER_REQUEST", ) @mock_dynamodb2 def test_create_table_error_pay_per_request_with_provisioned_param(): client = boto3.client("dynamodb", region_name="us-east-1") try: client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, BillingMode="PAY_PER_REQUEST", ) except ClientError as err: err.response["Error"]["Code"].should.equal("ValidationException") @mock_dynamodb2 def test_duplicate_create(): client = boto3.client("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) try: client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) except ClientError as err: err.response["Error"]["Code"].should.equal("ResourceInUseException") else: raise RuntimeError("Should have raised ResourceInUseException") @mock_dynamodb2 def test_delete_table(): client = boto3.client("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.delete_table(TableName="test1") resp = client.list_tables() len(resp["TableNames"]).should.equal(0) try: client.delete_table(TableName="test1") except ClientError as err: err.response["Error"]["Code"].should.equal("ResourceNotFoundException") else: raise RuntimeError("Should have raised ResourceNotFoundException") @mock_dynamodb2 def test_delete_item(): client = boto3.client("dynamodb", region_name="us-east-1") dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.put_item( TableName="test1", Item={"client": {"S": "client1"}, "app": {"S": "app1"}} ) client.put_item( TableName="test1", Item={"client": {"S": "client1"}, "app": {"S": "app2"}} ) table = dynamodb.Table("test1") response = table.scan() assert response["Count"] == 2 # Test ReturnValues validation with assert_raises(ClientError) as ex: table.delete_item( Key={"client": "client1", "app": "app1"}, ReturnValues="ALL_NEW" ) # Test deletion and returning old value response = table.delete_item( Key={"client": "client1", "app": "app1"}, ReturnValues="ALL_OLD" ) response["Attributes"].should.contain("client") response["Attributes"].should.contain("app") response = table.scan() assert response["Count"] == 1 # Test deletion returning nothing response = table.delete_item(Key={"client": "client1", "app": "app2"}) len(response["Attributes"]).should.equal(0) response = table.scan() assert response["Count"] == 0 @mock_dynamodb2 def test_describe_limits(): client = boto3.client("dynamodb", region_name="eu-central-1") resp = client.describe_limits() resp["AccountMaxReadCapacityUnits"].should.equal(20000) resp["AccountMaxWriteCapacityUnits"].should.equal(20000) resp["TableMaxWriteCapacityUnits"].should.equal(10000) resp["TableMaxReadCapacityUnits"].should.equal(10000) @mock_dynamodb2 def test_set_ttl(): client = boto3.client("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.update_time_to_live( TableName="test1", TimeToLiveSpecification={"Enabled": True, "AttributeName": "expire"}, ) resp = client.describe_time_to_live(TableName="test1") resp["TimeToLiveDescription"]["TimeToLiveStatus"].should.equal("ENABLED") resp["TimeToLiveDescription"]["AttributeName"].should.equal("expire") client.update_time_to_live( TableName="test1", TimeToLiveSpecification={"Enabled": False, "AttributeName": "expire"}, ) resp = client.describe_time_to_live(TableName="test1") resp["TimeToLiveDescription"]["TimeToLiveStatus"].should.equal("DISABLED") @mock_dynamodb2 def test_describe_continuous_backups(): # given client = boto3.client("dynamodb", region_name="us-east-1") table_name = client.create_table( TableName="test", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], BillingMode="PAY_PER_REQUEST", )["TableDescription"]["TableName"] # when response = client.describe_continuous_backups(TableName=table_name) # then response["ContinuousBackupsDescription"].should.equal( { "ContinuousBackupsStatus": "ENABLED", "PointInTimeRecoveryDescription": {"PointInTimeRecoveryStatus": "DISABLED"}, } ) @mock_dynamodb2 def test_describe_continuous_backups_errors(): # given client = boto3.client("dynamodb", region_name="us-east-1") # when with assert_raises(Exception) as e: client.describe_continuous_backups(TableName="not-existing-table") # then ex = e.exception ex.operation_name.should.equal("DescribeContinuousBackups") ex.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.response["Error"]["Code"].should.contain("TableNotFoundException") ex.response["Error"]["Message"].should.equal("Table not found: not-existing-table") @mock_dynamodb2 def test_update_continuous_backups(): # given client = boto3.client("dynamodb", region_name="us-east-1") table_name = client.create_table( TableName="test", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], BillingMode="PAY_PER_REQUEST", )["TableDescription"]["TableName"] # when response = client.update_continuous_backups( TableName=table_name, PointInTimeRecoverySpecification={"PointInTimeRecoveryEnabled": True}, ) # then response["ContinuousBackupsDescription"]["ContinuousBackupsStatus"].should.equal( "ENABLED" ) point_in_time = response["ContinuousBackupsDescription"][ "PointInTimeRecoveryDescription" ] earliest_datetime = point_in_time["EarliestRestorableDateTime"] earliest_datetime.should.be.a(datetime) latest_datetime = point_in_time["LatestRestorableDateTime"] latest_datetime.should.be.a(datetime) point_in_time["PointInTimeRecoveryStatus"].should.equal("ENABLED") # when # a second update should not change anything response = client.update_continuous_backups( TableName=table_name, PointInTimeRecoverySpecification={"PointInTimeRecoveryEnabled": True}, ) # then response["ContinuousBackupsDescription"]["ContinuousBackupsStatus"].should.equal( "ENABLED" ) point_in_time = response["ContinuousBackupsDescription"][ "PointInTimeRecoveryDescription" ] point_in_time["EarliestRestorableDateTime"].should.equal(earliest_datetime) point_in_time["LatestRestorableDateTime"].should.equal(latest_datetime) point_in_time["PointInTimeRecoveryStatus"].should.equal("ENABLED") # when response = client.update_continuous_backups( TableName=table_name, PointInTimeRecoverySpecification={"PointInTimeRecoveryEnabled": False}, ) # then response["ContinuousBackupsDescription"].should.equal( { "ContinuousBackupsStatus": "ENABLED", "PointInTimeRecoveryDescription": {"PointInTimeRecoveryStatus": "DISABLED"}, } ) @mock_dynamodb2 def test_update_continuous_backups_errors(): # given client = boto3.client("dynamodb", region_name="us-east-1") # when with assert_raises(Exception) as e: client.update_continuous_backups( TableName="not-existing-table", PointInTimeRecoverySpecification={"PointInTimeRecoveryEnabled": True}, ) # then ex = e.exception ex.operation_name.should.equal("UpdateContinuousBackups") ex.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.response["Error"]["Code"].should.contain("TableNotFoundException") ex.response["Error"]["Message"].should.equal("Table not found: not-existing-table") # https://github.com/spulec/moto/issues/1043 @mock_dynamodb2 def test_query_missing_expr_names(): client = boto3.client("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.put_item( TableName="test1", Item={"client": {"S": "test1"}, "app": {"S": "test1"}} ) client.put_item( TableName="test1", Item={"client": {"S": "test2"}, "app": {"S": "test2"}} ) resp = client.query( TableName="test1", KeyConditionExpression="client=:client", ExpressionAttributeValues={":client": {"S": "test1"}}, ) resp["Count"].should.equal(1) resp["Items"][0]["client"]["S"].should.equal("test1") resp = client.query( TableName="test1", KeyConditionExpression=":name=test2", ExpressionAttributeNames={":name": "client"}, ) resp["Count"].should.equal(1) resp["Items"][0]["client"]["S"].should.equal("test2") # https://github.com/spulec/moto/issues/2328 @mock_dynamodb2 def test_update_item_with_list(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="Table", KeySchema=[{"AttributeName": "key", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "key", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) table = dynamodb.Table("Table") table.update_item( Key={"key": "the-key"}, AttributeUpdates={"list": {"Value": [1, 2], "Action": "PUT"}}, ) resp = table.get_item(Key={"key": "the-key"}) resp["Item"].should.equal({"key": "the-key", "list": [1, 2]}) # https://github.com/spulec/moto/issues/1342 @mock_dynamodb2 def test_update_item_on_map(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") client = boto3.client("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item( Item={ "forum_name": "the-key", "subject": "123", "body": {"nested": {"data": "test"}}, } ) resp = table.scan() resp["Items"][0]["body"].should.equal({"nested": {"data": "test"}}) # Nonexistent nested attributes are supported for existing top-level attributes. table.update_item( Key={"forum_name": "the-key", "subject": "123"}, UpdateExpression="SET body.#nested.#data = :tb", ExpressionAttributeNames={"#nested": "nested", "#data": "data",}, ExpressionAttributeValues={":tb": "new_value"}, ) # Running this against AWS DDB gives an exception so make sure it also fails.: with assert_raises(client.exceptions.ClientError): # botocore.exceptions.ClientError: An error occurred (ValidationException) when calling the UpdateItem # operation: The document path provided in the update expression is invalid for update table.update_item( Key={"forum_name": "the-key", "subject": "123"}, UpdateExpression="SET body.#nested.#nonexistentnested.#data = :tb2", ExpressionAttributeNames={ "#nested": "nested", "#nonexistentnested": "nonexistentnested", "#data": "data", }, ExpressionAttributeValues={":tb2": "other_value"}, ) table.update_item( Key={"forum_name": "the-key", "subject": "123"}, UpdateExpression="SET body.#nested.#nonexistentnested = :tb2", ExpressionAttributeNames={ "#nested": "nested", "#nonexistentnested": "nonexistentnested", }, ExpressionAttributeValues={":tb2": {"data": "other_value"}}, ) resp = table.scan() resp["Items"][0]["body"].should.equal( {"nested": {"data": "new_value", "nonexistentnested": {"data": "other_value"}}} ) # Test nested value for a nonexistent attribute throws a ClientError. with assert_raises(client.exceptions.ClientError): table.update_item( Key={"forum_name": "the-key", "subject": "123"}, UpdateExpression="SET nonexistent.#nested = :tb", ExpressionAttributeNames={"#nested": "nested"}, ExpressionAttributeValues={":tb": "new_value"}, ) # https://github.com/spulec/moto/issues/1358 @mock_dynamodb2 def test_update_if_not_exists(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[ {"AttributeName": "forum_name", "KeyType": "HASH"}, {"AttributeName": "subject", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "forum_name", "AttributeType": "S"}, {"AttributeName": "subject", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.put_item(Item={"forum_name": "the-key", "subject": "123"}) table.update_item( Key={"forum_name": "the-key", "subject": "123"}, # if_not_exists without space UpdateExpression="SET created_at=if_not_exists(created_at,:created_at)", ExpressionAttributeValues={":created_at": 123}, ) resp = table.scan() assert resp["Items"][0]["created_at"] == 123 table.update_item( Key={"forum_name": "the-key", "subject": "123"}, # if_not_exists with space UpdateExpression="SET created_at = if_not_exists (created_at, :created_at)", ExpressionAttributeValues={":created_at": 456}, ) resp = table.scan() # Still the original value assert resp["Items"][0]["created_at"] == 123 # https://github.com/spulec/moto/issues/1937 @mock_dynamodb2 def test_update_return_attributes(): dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="moto-test", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) def update(col, to, rv): return dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "foo"}}, AttributeUpdates={col: {"Value": {"S": to}, "Action": "PUT"}}, ReturnValues=rv, ) r = update("col1", "val1", "ALL_NEW") assert r["Attributes"] == {"id": {"S": "foo"}, "col1": {"S": "val1"}} r = update("col1", "val2", "ALL_OLD") assert r["Attributes"] == {"id": {"S": "foo"}, "col1": {"S": "val1"}} r = update("col2", "val3", "UPDATED_NEW") assert r["Attributes"] == {"col2": {"S": "val3"}} r = update("col2", "val4", "UPDATED_OLD") assert r["Attributes"] == {"col2": {"S": "val3"}} r = update("col1", "val5", "NONE") assert r["Attributes"] == {} with assert_raises(ClientError) as ex: r = update("col1", "val6", "WRONG") @mock_dynamodb2 def test_put_return_attributes(): dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="moto-test", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) r = dynamodb.put_item( TableName="moto-test", Item={"id": {"S": "foo"}, "col1": {"S": "val1"}}, ReturnValues="NONE", ) assert "Attributes" not in r r = dynamodb.put_item( TableName="moto-test", Item={"id": {"S": "foo"}, "col1": {"S": "val2"}}, ReturnValues="ALL_OLD", ) assert r["Attributes"] == {"id": {"S": "foo"}, "col1": {"S": "val1"}} with assert_raises(ClientError) as ex: dynamodb.put_item( TableName="moto-test", Item={"id": {"S": "foo"}, "col1": {"S": "val3"}}, ReturnValues="ALL_NEW", ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.exception.response["Error"]["Message"].should.equal( "Return values set to invalid value" ) @mock_dynamodb2 def test_query_global_secondary_index_when_created_via_update_table_resource(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") # Create the DynamoDB table. dynamodb.create_table( TableName="users", KeySchema=[{"AttributeName": "user_id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "user_id", "AttributeType": "N"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("users") table.update( AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], GlobalSecondaryIndexUpdates=[ { "Create": { "IndexName": "forum_name_index", "KeySchema": [{"AttributeName": "forum_name", "KeyType": "HASH"}], "Projection": {"ProjectionType": "ALL"}, "ProvisionedThroughput": { "ReadCapacityUnits": 5, "WriteCapacityUnits": 5, }, } } ], ) next_user_id = 1 for my_forum_name in ["cats", "dogs"]: for my_subject in [ "my pet is the cutest", "wow look at what my pet did", "don't you love my pet?", ]: table.put_item( Item={ "user_id": next_user_id, "forum_name": my_forum_name, "subject": my_subject, } ) next_user_id += 1 # get all the cat users forum_only_query_response = table.query( IndexName="forum_name_index", Select="ALL_ATTRIBUTES", KeyConditionExpression=Key("forum_name").eq("cats"), ) forum_only_items = forum_only_query_response["Items"] assert len(forum_only_items) == 3 for item in forum_only_items: assert item["forum_name"] == "cats" # query all cat users with a particular subject forum_and_subject_query_results = table.query( IndexName="forum_name_index", Select="ALL_ATTRIBUTES", KeyConditionExpression=Key("forum_name").eq("cats"), FilterExpression=Attr("subject").eq("my pet is the cutest"), ) forum_and_subject_items = forum_and_subject_query_results["Items"] assert len(forum_and_subject_items) == 1 assert forum_and_subject_items[0] == { "user_id": Decimal("1"), "forum_name": "cats", "subject": "my pet is the cutest", } @mock_dynamodb2 def test_dynamodb_streams_1(): conn = boto3.client("dynamodb", region_name="us-east-1") resp = conn.create_table( TableName="test-streams", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, StreamSpecification={ "StreamEnabled": True, "StreamViewType": "NEW_AND_OLD_IMAGES", }, ) assert "StreamSpecification" in resp["TableDescription"] assert resp["TableDescription"]["StreamSpecification"] == { "StreamEnabled": True, "StreamViewType": "NEW_AND_OLD_IMAGES", } assert "LatestStreamLabel" in resp["TableDescription"] assert "LatestStreamArn" in resp["TableDescription"] resp = conn.delete_table(TableName="test-streams") assert "StreamSpecification" in resp["TableDescription"] @mock_dynamodb2 def test_dynamodb_streams_2(): conn = boto3.client("dynamodb", region_name="us-east-1") resp = conn.create_table( TableName="test-stream-update", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) assert "StreamSpecification" not in resp["TableDescription"] resp = conn.update_table( TableName="test-stream-update", StreamSpecification={"StreamEnabled": True, "StreamViewType": "NEW_IMAGE"}, ) assert "StreamSpecification" in resp["TableDescription"] assert resp["TableDescription"]["StreamSpecification"] == { "StreamEnabled": True, "StreamViewType": "NEW_IMAGE", } assert "LatestStreamLabel" in resp["TableDescription"] assert "LatestStreamArn" in resp["TableDescription"] @mock_dynamodb2 def test_condition_expressions(): client = boto3.client("dynamodb", region_name="us-east-1") # Create the DynamoDB table. client.create_table( TableName="test1", AttributeDefinitions=[ {"AttributeName": "client", "AttributeType": "S"}, {"AttributeName": "app", "AttributeType": "S"}, ], KeySchema=[ {"AttributeName": "client", "KeyType": "HASH"}, {"AttributeName": "app", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 123, "WriteCapacityUnits": 123}, ) client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "match": {"S": "match"}, "existing": {"S": "existing"}, }, ) client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "match": {"S": "match"}, "existing": {"S": "existing"}, }, ConditionExpression="attribute_exists(#existing) AND attribute_not_exists(#nonexistent) AND #match = :match", ExpressionAttributeNames={ "#existing": "existing", "#nonexistent": "nope", "#match": "match", }, ExpressionAttributeValues={":match": {"S": "match"}}, ) client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "match": {"S": "match"}, "existing": {"S": "existing"}, }, ConditionExpression="NOT(attribute_exists(#nonexistent1) AND attribute_exists(#nonexistent2))", ExpressionAttributeNames={"#nonexistent1": "nope", "#nonexistent2": "nope2"}, ) client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "match": {"S": "match"}, "existing": {"S": "existing"}, }, ConditionExpression="attribute_exists(#nonexistent) OR attribute_exists(#existing)", ExpressionAttributeNames={"#nonexistent": "nope", "#existing": "existing"}, ) client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "match": {"S": "match"}, "existing": {"S": "existing"}, }, ConditionExpression="#client BETWEEN :a AND :z", ExpressionAttributeNames={"#client": "client"}, ExpressionAttributeValues={":a": {"S": "a"}, ":z": {"S": "z"}}, ) client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "match": {"S": "match"}, "existing": {"S": "existing"}, }, ConditionExpression="#client IN (:client1, :client2)", ExpressionAttributeNames={"#client": "client"}, ExpressionAttributeValues={ ":client1": {"S": "client1"}, ":client2": {"S": "client2"}, }, ) with assert_raises(client.exceptions.ConditionalCheckFailedException): client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "match": {"S": "match"}, "existing": {"S": "existing"}, }, ConditionExpression="attribute_exists(#nonexistent1) AND attribute_exists(#nonexistent2)", ExpressionAttributeNames={ "#nonexistent1": "nope", "#nonexistent2": "nope2", }, ) with assert_raises(client.exceptions.ConditionalCheckFailedException): client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "match": {"S": "match"}, "existing": {"S": "existing"}, }, ConditionExpression="NOT(attribute_not_exists(#nonexistent1) AND attribute_not_exists(#nonexistent2))", ExpressionAttributeNames={ "#nonexistent1": "nope", "#nonexistent2": "nope2", }, ) with assert_raises(client.exceptions.ConditionalCheckFailedException): client.put_item( TableName="test1", Item={ "client": {"S": "client1"}, "app": {"S": "app1"}, "match": {"S": "match"}, "existing": {"S": "existing"}, }, ConditionExpression="attribute_exists(#existing) AND attribute_not_exists(#nonexistent) AND #match = :match", ExpressionAttributeNames={ "#existing": "existing", "#nonexistent": "nope", "#match": "match", }, ExpressionAttributeValues={":match": {"S": "match2"}}, ) # Make sure update_item honors ConditionExpression as well client.update_item( TableName="test1", Key={"client": {"S": "client1"}, "app": {"S": "app1"}}, UpdateExpression="set #match=:match", ConditionExpression="attribute_exists(#existing)", ExpressionAttributeNames={"#existing": "existing", "#match": "match"}, ExpressionAttributeValues={":match": {"S": "match"}}, ) with assert_raises(client.exceptions.ConditionalCheckFailedException): client.update_item( TableName="test1", Key={"client": {"S": "client1"}, "app": {"S": "app1"}}, UpdateExpression="set #match=:match", ConditionExpression="attribute_not_exists(#existing)", ExpressionAttributeValues={":match": {"S": "match"}}, ExpressionAttributeNames={"#existing": "existing", "#match": "match"}, ) with assert_raises(client.exceptions.ConditionalCheckFailedException): client.delete_item( TableName="test1", Key={"client": {"S": "client1"}, "app": {"S": "app1"}}, ConditionExpression="attribute_not_exists(#existing)", ExpressionAttributeValues={":match": {"S": "match"}}, ExpressionAttributeNames={"#existing": "existing", "#match": "match"}, ) @mock_dynamodb2 def test_condition_expression_numerical_attribute(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="my-table", KeySchema=[{"AttributeName": "partitionKey", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "partitionKey", "AttributeType": "S"}], ) table = dynamodb.Table("my-table") table.put_item(Item={"partitionKey": "pk-pos", "myAttr": 5}) table.put_item(Item={"partitionKey": "pk-neg", "myAttr": -5}) # try to update the item we put in the table using numerical condition expression # Specifically, verify that we can compare with a zero-value # First verify that > and >= work on positive numbers update_numerical_con_expr( key="pk-pos", con_expr="myAttr > :zero", res="6", table=table ) update_numerical_con_expr( key="pk-pos", con_expr="myAttr >= :zero", res="7", table=table ) # Second verify that < and <= work on negative numbers update_numerical_con_expr( key="pk-neg", con_expr="myAttr < :zero", res="-4", table=table ) update_numerical_con_expr( key="pk-neg", con_expr="myAttr <= :zero", res="-3", table=table ) def update_numerical_con_expr(key, con_expr, res, table): table.update_item( Key={"partitionKey": key}, UpdateExpression="ADD myAttr :one", ExpressionAttributeValues={":zero": 0, ":one": 1}, ConditionExpression=con_expr, ) table.get_item(Key={"partitionKey": key})["Item"]["myAttr"].should.equal( Decimal(res) ) @mock_dynamodb2 def test_condition_expression__attr_doesnt_exist(): client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( TableName="test", KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) client.put_item( TableName="test", Item={"forum_name": {"S": "foo"}, "ttl": {"N": "bar"}} ) def update_if_attr_doesnt_exist(): # Test nonexistent top-level attribute. client.update_item( TableName="test", Key={"forum_name": {"S": "the-key"}, "subject": {"S": "the-subject"}}, UpdateExpression="set #new_state=:new_state, #ttl=:ttl", ConditionExpression="attribute_not_exists(#new_state)", ExpressionAttributeNames={"#new_state": "foobar", "#ttl": "ttl"}, ExpressionAttributeValues={ ":new_state": {"S": "some-value"}, ":ttl": {"N": "12345.67"}, }, ReturnValues="ALL_NEW", ) update_if_attr_doesnt_exist() # Second time should fail with assert_raises(client.exceptions.ConditionalCheckFailedException): update_if_attr_doesnt_exist() @mock_dynamodb2 def test_condition_expression__or_order(): client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( TableName="test", KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) # ensure that the RHS of the OR expression is not evaluated if the LHS # returns true (as it would result an error) client.update_item( TableName="test", Key={"forum_name": {"S": "the-key"}}, UpdateExpression="set #ttl=:ttl", ConditionExpression="attribute_not_exists(#ttl) OR #ttl <= :old_ttl", ExpressionAttributeNames={"#ttl": "ttl"}, ExpressionAttributeValues={":ttl": {"N": "6"}, ":old_ttl": {"N": "5"}}, ) @mock_dynamodb2 def test_condition_expression__and_order(): client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( TableName="test", KeySchema=[{"AttributeName": "forum_name", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "forum_name", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) # ensure that the RHS of the AND expression is not evaluated if the LHS # returns true (as it would result an error) with assert_raises(client.exceptions.ConditionalCheckFailedException): client.update_item( TableName="test", Key={"forum_name": {"S": "the-key"}}, UpdateExpression="set #ttl=:ttl", ConditionExpression="attribute_exists(#ttl) AND #ttl <= :old_ttl", ExpressionAttributeNames={"#ttl": "ttl"}, ExpressionAttributeValues={":ttl": {"N": "6"}, ":old_ttl": {"N": "5"}}, ) @mock_dynamodb2 def test_query_gsi_with_range_key(): dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[ {"AttributeName": "id", "AttributeType": "S"}, {"AttributeName": "gsi_hash_key", "AttributeType": "S"}, {"AttributeName": "gsi_range_key", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, GlobalSecondaryIndexes=[ { "IndexName": "test_gsi", "KeySchema": [ {"AttributeName": "gsi_hash_key", "KeyType": "HASH"}, {"AttributeName": "gsi_range_key", "KeyType": "RANGE"}, ], "Projection": {"ProjectionType": "ALL"}, "ProvisionedThroughput": { "ReadCapacityUnits": 1, "WriteCapacityUnits": 1, }, } ], ) dynamodb.put_item( TableName="test", Item={ "id": {"S": "test1"}, "gsi_hash_key": {"S": "key1"}, "gsi_range_key": {"S": "range1"}, }, ) dynamodb.put_item( TableName="test", Item={"id": {"S": "test2"}, "gsi_hash_key": {"S": "key1"}} ) res = dynamodb.query( TableName="test", IndexName="test_gsi", KeyConditionExpression="gsi_hash_key = :gsi_hash_key and gsi_range_key = :gsi_range_key", ExpressionAttributeValues={ ":gsi_hash_key": {"S": "key1"}, ":gsi_range_key": {"S": "range1"}, }, ) res.should.have.key("Count").equal(1) res.should.have.key("Items") res["Items"][0].should.equal( { "id": {"S": "test1"}, "gsi_hash_key": {"S": "key1"}, "gsi_range_key": {"S": "range1"}, } ) @mock_dynamodb2 def test_scan_by_non_exists_index(): dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[ {"AttributeName": "id", "AttributeType": "S"}, {"AttributeName": "gsi_col", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, GlobalSecondaryIndexes=[ { "IndexName": "test_gsi", "KeySchema": [{"AttributeName": "gsi_col", "KeyType": "HASH"}], "Projection": {"ProjectionType": "ALL"}, "ProvisionedThroughput": { "ReadCapacityUnits": 1, "WriteCapacityUnits": 1, }, } ], ) with assert_raises(ClientError) as ex: dynamodb.scan(TableName="test", IndexName="non_exists_index") ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.exception.response["Error"]["Message"].should.equal( "The table does not have the specified index: non_exists_index" ) @mock_dynamodb2 def test_query_by_non_exists_index(): dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[ {"AttributeName": "id", "AttributeType": "S"}, {"AttributeName": "gsi_col", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, GlobalSecondaryIndexes=[ { "IndexName": "test_gsi", "KeySchema": [{"AttributeName": "gsi_col", "KeyType": "HASH"}], "Projection": {"ProjectionType": "ALL"}, "ProvisionedThroughput": { "ReadCapacityUnits": 1, "WriteCapacityUnits": 1, }, } ], ) with assert_raises(ClientError) as ex: dynamodb.query( TableName="test", IndexName="non_exists_index", KeyConditionExpression="CarModel=M", ) ex.exception.response["Error"]["Code"].should.equal("ResourceNotFoundException") ex.exception.response["Error"]["Message"].should.equal( "Invalid index: non_exists_index for table: test. Available indexes are: test_gsi" ) @mock_dynamodb2 def test_batch_items_returns_all(): dynamodb = _create_user_table() returned_items = dynamodb.batch_get_item( RequestItems={ "users": { "Keys": [ {"username": {"S": "user0"}}, {"username": {"S": "user1"}}, {"username": {"S": "user2"}}, {"username": {"S": "user3"}}, ], "ConsistentRead": True, } } )["Responses"]["users"] assert len(returned_items) == 3 assert [item["username"]["S"] for item in returned_items] == [ "user1", "user2", "user3", ] @mock_dynamodb2 def test_batch_items_throws_exception_when_requesting_100_items_for_single_table(): dynamodb = _create_user_table() with assert_raises(ClientError) as ex: dynamodb.batch_get_item( RequestItems={ "users": { "Keys": [ {"username": {"S": "user" + str(i)}} for i in range(0, 104) ], "ConsistentRead": True, } } ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") msg = ex.exception.response["Error"]["Message"] msg.should.contain("1 validation error detected: Value") msg.should.contain( "at 'requestItems.users.member.keys' failed to satisfy constraint: Member must have length less than or equal to 100" ) @mock_dynamodb2 def test_batch_items_throws_exception_when_requesting_100_items_across_all_tables(): dynamodb = _create_user_table() with assert_raises(ClientError) as ex: dynamodb.batch_get_item( RequestItems={ "users": { "Keys": [ {"username": {"S": "user" + str(i)}} for i in range(0, 75) ], "ConsistentRead": True, }, "users2": { "Keys": [ {"username": {"S": "user" + str(i)}} for i in range(0, 75) ], "ConsistentRead": True, }, } ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["Error"]["Message"].should.equal( "Too many items requested for the BatchGetItem call" ) @mock_dynamodb2 def test_batch_items_with_basic_projection_expression(): dynamodb = _create_user_table() returned_items = dynamodb.batch_get_item( RequestItems={ "users": { "Keys": [ {"username": {"S": "user0"}}, {"username": {"S": "user1"}}, {"username": {"S": "user2"}}, {"username": {"S": "user3"}}, ], "ConsistentRead": True, "ProjectionExpression": "username", } } )["Responses"]["users"] returned_items.should.have.length_of(3) [item["username"]["S"] for item in returned_items].should.be.equal( ["user1", "user2", "user3"] ) [item.get("foo") for item in returned_items].should.be.equal([None, None, None]) # The projection expression should not remove data from storage returned_items = dynamodb.batch_get_item( RequestItems={ "users": { "Keys": [ {"username": {"S": "user0"}}, {"username": {"S": "user1"}}, {"username": {"S": "user2"}}, {"username": {"S": "user3"}}, ], "ConsistentRead": True, } } )["Responses"]["users"] [item["username"]["S"] for item in returned_items].should.be.equal( ["user1", "user2", "user3"] ) [item["foo"]["S"] for item in returned_items].should.be.equal(["bar", "bar", "bar"]) @mock_dynamodb2 def test_batch_items_with_basic_projection_expression_and_attr_expression_names(): dynamodb = _create_user_table() returned_items = dynamodb.batch_get_item( RequestItems={ "users": { "Keys": [ {"username": {"S": "user0"}}, {"username": {"S": "user1"}}, {"username": {"S": "user2"}}, {"username": {"S": "user3"}}, ], "ConsistentRead": True, "ProjectionExpression": "#rl", "ExpressionAttributeNames": {"#rl": "username"}, } } )["Responses"]["users"] returned_items.should.have.length_of(3) [item["username"]["S"] for item in returned_items].should.be.equal( ["user1", "user2", "user3"] ) [item.get("foo") for item in returned_items].should.be.equal([None, None, None]) @mock_dynamodb2 def test_batch_items_should_throw_exception_for_duplicate_request(): client = _create_user_table() with assert_raises(ClientError) as ex: client.batch_get_item( RequestItems={ "users": { "Keys": [ {"username": {"S": "user0"}}, {"username": {"S": "user0"}}, ], "ConsistentRead": True, } } ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["Error"]["Message"].should.equal( "Provided list of item keys contains duplicates" ) @mock_dynamodb2 def test_index_with_unknown_attributes_should_fail(): dynamodb = boto3.client("dynamodb", region_name="us-east-1") expected_exception = ( "Some index key attributes are not defined in AttributeDefinitions." ) with assert_raises(ClientError) as ex: dynamodb.create_table( AttributeDefinitions=[ {"AttributeName": "customer_nr", "AttributeType": "S"}, {"AttributeName": "last_name", "AttributeType": "S"}, ], TableName="table_with_missing_attribute_definitions", KeySchema=[ {"AttributeName": "customer_nr", "KeyType": "HASH"}, {"AttributeName": "last_name", "KeyType": "RANGE"}, ], LocalSecondaryIndexes=[ { "IndexName": "indexthataddsanadditionalattribute", "KeySchema": [ {"AttributeName": "customer_nr", "KeyType": "HASH"}, {"AttributeName": "postcode", "KeyType": "RANGE"}, ], "Projection": {"ProjectionType": "ALL"}, } ], BillingMode="PAY_PER_REQUEST", ) ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["Error"]["Message"].should.contain(expected_exception) @mock_dynamodb2 def test_update_list_index__set_existing_index(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={ "id": {"S": "foo"}, "itemlist": {"L": [{"S": "bar1"}, {"S": "bar2"}, {"S": "bar3"}]}, }, ) client.update_item( TableName=table_name, Key={"id": {"S": "foo"}}, UpdateExpression="set itemlist[1]=:Item", ExpressionAttributeValues={":Item": {"S": "bar2_update"}}, ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo"}})["Item"] result["id"].should.equal({"S": "foo"}) result["itemlist"].should.equal( {"L": [{"S": "bar1"}, {"S": "bar2_update"}, {"S": "bar3"}]} ) @mock_dynamodb2 def test_update_list_index__set_existing_nested_index(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={ "id": {"S": "foo2"}, "itemmap": { "M": {"itemlist": {"L": [{"S": "bar1"}, {"S": "bar2"}, {"S": "bar3"}]}} }, }, ) client.update_item( TableName=table_name, Key={"id": {"S": "foo2"}}, UpdateExpression="set itemmap.itemlist[1]=:Item", ExpressionAttributeValues={":Item": {"S": "bar2_update"}}, ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo2"}})["Item"] result["id"].should.equal({"S": "foo2"}) result["itemmap"]["M"]["itemlist"]["L"].should.equal( [{"S": "bar1"}, {"S": "bar2_update"}, {"S": "bar3"}] ) @mock_dynamodb2 def test_update_list_index__set_index_out_of_range(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={ "id": {"S": "foo"}, "itemlist": {"L": [{"S": "bar1"}, {"S": "bar2"}, {"S": "bar3"}]}, }, ) client.update_item( TableName=table_name, Key={"id": {"S": "foo"}}, UpdateExpression="set itemlist[10]=:Item", ExpressionAttributeValues={":Item": {"S": "bar10"}}, ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo"}})["Item"] assert result["id"] == {"S": "foo"} assert result["itemlist"] == { "L": [{"S": "bar1"}, {"S": "bar2"}, {"S": "bar3"}, {"S": "bar10"}] } @mock_dynamodb2 def test_update_list_index__set_nested_index_out_of_range(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={ "id": {"S": "foo2"}, "itemmap": { "M": {"itemlist": {"L": [{"S": "bar1"}, {"S": "bar2"}, {"S": "bar3"}]}} }, }, ) client.update_item( TableName=table_name, Key={"id": {"S": "foo2"}}, UpdateExpression="set itemmap.itemlist[10]=:Item", ExpressionAttributeValues={":Item": {"S": "bar10"}}, ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo2"}})["Item"] assert result["id"] == {"S": "foo2"} assert result["itemmap"]["M"]["itemlist"]["L"] == [ {"S": "bar1"}, {"S": "bar2"}, {"S": "bar3"}, {"S": "bar10"}, ] @mock_dynamodb2 def test_update_list_index__set_double_nested_index(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={ "id": {"S": "foo2"}, "itemmap": { "M": { "itemlist": { "L": [ {"M": {"foo": {"S": "bar11"}, "foos": {"S": "bar12"}}}, {"M": {"foo": {"S": "bar21"}, "foos": {"S": "bar21"}}}, ] } } }, }, ) client.update_item( TableName=table_name, Key={"id": {"S": "foo2"}}, UpdateExpression="set itemmap.itemlist[1].foos=:Item", ExpressionAttributeValues={":Item": {"S": "bar22"}}, ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo2"}})["Item"] assert result["id"] == {"S": "foo2"} len(result["itemmap"]["M"]["itemlist"]["L"]).should.equal(2) result["itemmap"]["M"]["itemlist"]["L"][0].should.equal( {"M": {"foo": {"S": "bar11"}, "foos": {"S": "bar12"}}} ) # unchanged result["itemmap"]["M"]["itemlist"]["L"][1].should.equal( {"M": {"foo": {"S": "bar21"}, "foos": {"S": "bar22"}}} ) # updated @mock_dynamodb2 def test_update_list_index__set_index_of_a_string(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={"id": {"S": "foo2"}, "itemstr": {"S": "somestring"}} ) with assert_raises(ClientError) as ex: client.update_item( TableName=table_name, Key={"id": {"S": "foo2"}}, UpdateExpression="set itemstr[1]=:Item", ExpressionAttributeValues={":Item": {"S": "string_update"}}, ) result = client.get_item(TableName=table_name, Key={"id": {"S": "foo2"}})[ "Item" ] ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["Error"]["Message"].should.equal( "The document path provided in the update expression is invalid for update" ) @mock_dynamodb2 def test_remove_top_level_attribute(): table_name = "test_remove" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={"id": {"S": "foo"}, "item": {"S": "bar"}} ) client.update_item( TableName=table_name, Key={"id": {"S": "foo"}}, UpdateExpression="REMOVE #i", ExpressionAttributeNames={"#i": "item"}, ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo"}})["Item"] result.should.equal({"id": {"S": "foo"}}) @mock_dynamodb2 def test_remove_top_level_attribute_non_existent(): """ Remove statements do not require attribute to exist they silently pass """ table_name = "test_remove" client = create_table_with_list(table_name) ddb_item = {"id": {"S": "foo"}, "item": {"S": "bar"}} client.put_item(TableName=table_name, Item=ddb_item) client.update_item( TableName=table_name, Key={"id": {"S": "foo"}}, UpdateExpression="REMOVE non_existent_attribute", ExpressionAttributeNames={"#i": "item"}, ) result = client.get_item(TableName=table_name, Key={"id": {"S": "foo"}})["Item"] result.should.equal(ddb_item) @mock_dynamodb2 def test_remove_list_index__remove_existing_index(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={ "id": {"S": "foo"}, "itemlist": {"L": [{"S": "bar1"}, {"S": "bar2"}, {"S": "bar3"}]}, }, ) client.update_item( TableName=table_name, Key={"id": {"S": "foo"}}, UpdateExpression="REMOVE itemlist[1]", ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo"}})["Item"] result["id"].should.equal({"S": "foo"}) result["itemlist"].should.equal({"L": [{"S": "bar1"}, {"S": "bar3"}]}) @mock_dynamodb2 def test_remove_list_index__remove_existing_nested_index(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={ "id": {"S": "foo2"}, "itemmap": {"M": {"itemlist": {"L": [{"S": "bar1"}, {"S": "bar2"}]}}}, }, ) client.update_item( TableName=table_name, Key={"id": {"S": "foo2"}}, UpdateExpression="REMOVE itemmap.itemlist[1]", ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo2"}})["Item"] result["id"].should.equal({"S": "foo2"}) result["itemmap"]["M"]["itemlist"]["L"].should.equal([{"S": "bar1"}]) @mock_dynamodb2 def test_remove_list_index__remove_existing_double_nested_index(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={ "id": {"S": "foo2"}, "itemmap": { "M": { "itemlist": { "L": [ {"M": {"foo00": {"S": "bar1"}, "foo01": {"S": "bar2"}}}, {"M": {"foo10": {"S": "bar1"}, "foo11": {"S": "bar2"}}}, ] } } }, }, ) client.update_item( TableName=table_name, Key={"id": {"S": "foo2"}}, UpdateExpression="REMOVE itemmap.itemlist[1].foo10", ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo2"}})["Item"] assert result["id"] == {"S": "foo2"} assert result["itemmap"]["M"]["itemlist"]["L"][0]["M"].should.equal( {"foo00": {"S": "bar1"}, "foo01": {"S": "bar2"}} ) # untouched assert result["itemmap"]["M"]["itemlist"]["L"][1]["M"].should.equal( {"foo11": {"S": "bar2"}} ) # changed @mock_dynamodb2 def test_remove_list_index__remove_index_out_of_range(): table_name = "test_list_index_access" client = create_table_with_list(table_name) client.put_item( TableName=table_name, Item={ "id": {"S": "foo"}, "itemlist": {"L": [{"S": "bar1"}, {"S": "bar2"}, {"S": "bar3"}]}, }, ) client.update_item( TableName=table_name, Key={"id": {"S": "foo"}}, UpdateExpression="REMOVE itemlist[10]", ) # result = client.get_item(TableName=table_name, Key={"id": {"S": "foo"}})["Item"] assert result["id"] == {"S": "foo"} assert result["itemlist"] == {"L": [{"S": "bar1"}, {"S": "bar2"}, {"S": "bar3"}]} def create_table_with_list(table_name): client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( TableName=table_name, KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], BillingMode="PAY_PER_REQUEST", ) return client @mock_dynamodb2 def test_sorted_query_with_numerical_sort_key(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="CarCollection", KeySchema=[ {"AttributeName": "CarModel", "KeyType": "HASH"}, {"AttributeName": "CarPrice", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "CarModel", "AttributeType": "S"}, {"AttributeName": "CarPrice", "AttributeType": "N"}, ], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) def create_item(price): return {"CarModel": "M", "CarPrice": price} table = dynamodb.Table("CarCollection") items = list(map(create_item, [2, 1, 10, 3])) for item in items: table.put_item(Item=item) response = table.query(KeyConditionExpression=Key("CarModel").eq("M")) response_items = response["Items"] assert len(items) == len(response_items) assert all(isinstance(item["CarPrice"], Decimal) for item in response_items) response_prices = [item["CarPrice"] for item in response_items] expected_prices = [Decimal(item["CarPrice"]) for item in items] expected_prices.sort() assert ( expected_prices == response_prices ), "result items are not sorted by numerical value" # https://github.com/spulec/moto/issues/1874 @mock_dynamodb2 def test_item_size_is_under_400KB(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") client = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="moto-test", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) table = dynamodb.Table("moto-test") large_item = "x" * 410 * 1000 assert_failure_due_to_item_size( func=client.put_item, TableName="moto-test", Item={"id": {"S": "foo"}, "cont": {"S": large_item}}, ) assert_failure_due_to_item_size( func=table.put_item, Item={"id": "bar", "cont": large_item} ) assert_failure_due_to_item_size_to_update( func=client.update_item, TableName="moto-test", Key={"id": {"S": "foo2"}}, UpdateExpression="set cont=:Item", ExpressionAttributeValues={":Item": {"S": large_item}}, ) # Assert op fails when updating a nested item assert_failure_due_to_item_size( func=table.put_item, Item={"id": "bar", "itemlist": [{"cont": large_item}]} ) assert_failure_due_to_item_size( func=client.put_item, TableName="moto-test", Item={ "id": {"S": "foo"}, "itemlist": {"L": [{"M": {"item1": {"S": large_item}}}]}, }, ) def assert_failure_due_to_item_size(func, **kwargs): with assert_raises(ClientError) as ex: func(**kwargs) ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["Error"]["Message"].should.equal( "Item size has exceeded the maximum allowed size" ) def assert_failure_due_to_item_size_to_update(func, **kwargs): with assert_raises(ClientError) as ex: func(**kwargs) ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["Error"]["Message"].should.equal( "Item size to update has exceeded the maximum allowed size" ) @mock_dynamodb2 # https://docs.aws.amazon.com/amazondynamodb/latest/APIReference/API_Query.html#DDB-Query-request-KeyConditionExpression def test_hash_key_cannot_use_begins_with_operations(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") table = dynamodb.create_table( TableName="test-table", KeySchema=[{"AttributeName": "key", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "key", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) items = [ {"key": "prefix-$LATEST", "value": "$LATEST"}, {"key": "prefix-DEV", "value": "DEV"}, {"key": "prefix-PROD", "value": "PROD"}, ] with table.batch_writer() as batch: for item in items: batch.put_item(Item=item) table = dynamodb.Table("test-table") with assert_raises(ClientError) as ex: table.query(KeyConditionExpression=Key("key").begins_with("prefix-")) ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["Error"]["Message"].should.equal( "Query key condition not supported" ) @mock_dynamodb2 def test_update_supports_complex_expression_attribute_values(): client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( AttributeDefinitions=[{"AttributeName": "SHA256", "AttributeType": "S"}], TableName="TestTable", KeySchema=[{"AttributeName": "SHA256", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) client.update_item( TableName="TestTable", Key={"SHA256": {"S": "sha-of-file"}}, UpdateExpression=( "SET MD5 = :md5," "MyStringSet = :string_set," "MyMap = :map" ), ExpressionAttributeValues={ ":md5": {"S": "md5-of-file"}, ":string_set": {"SS": ["string1", "string2"]}, ":map": {"M": {"EntryKey": {"SS": ["thing1", "thing2"]}}}, }, ) result = client.get_item( TableName="TestTable", Key={"SHA256": {"S": "sha-of-file"}} )["Item"] result.should.equal( { "MyStringSet": {"SS": ["string1", "string2"]}, "MyMap": {"M": {"EntryKey": {"SS": ["thing1", "thing2"]}}}, "SHA256": {"S": "sha-of-file"}, "MD5": {"S": "md5-of-file"}, } ) @mock_dynamodb2 def test_update_supports_list_append(): # Verify whether the list_append operation works as expected client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( AttributeDefinitions=[{"AttributeName": "SHA256", "AttributeType": "S"}], TableName="TestTable", KeySchema=[{"AttributeName": "SHA256", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) client.put_item( TableName="TestTable", Item={"SHA256": {"S": "sha-of-file"}, "crontab": {"L": [{"S": "bar1"}]}}, ) # Update item using list_append expression updated_item = client.update_item( TableName="TestTable", Key={"SHA256": {"S": "sha-of-file"}}, UpdateExpression="SET crontab = list_append(crontab, :i)", ExpressionAttributeValues={":i": {"L": [{"S": "bar2"}]}}, ReturnValues="UPDATED_NEW", ) # Verify updated item is correct updated_item["Attributes"].should.equal( {"crontab": {"L": [{"S": "bar1"}, {"S": "bar2"}]}} ) # Verify item is appended to the existing list result = client.get_item( TableName="TestTable", Key={"SHA256": {"S": "sha-of-file"}} )["Item"] result.should.equal( { "SHA256": {"S": "sha-of-file"}, "crontab": {"L": [{"S": "bar1"}, {"S": "bar2"}]}, } ) @mock_dynamodb2 def test_update_supports_nested_list_append(): # Verify whether we can append a list that's inside a map client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], TableName="TestTable", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) client.put_item( TableName="TestTable", Item={ "id": {"S": "nested_list_append"}, "a": {"M": {"b": {"L": [{"S": "bar1"}]}}}, }, ) # Update item using list_append expression updated_item = client.update_item( TableName="TestTable", Key={"id": {"S": "nested_list_append"}}, UpdateExpression="SET a.#b = list_append(a.#b, :i)", ExpressionAttributeValues={":i": {"L": [{"S": "bar2"}]}}, ExpressionAttributeNames={"#b": "b"}, ReturnValues="UPDATED_NEW", ) # Verify updated item is correct updated_item["Attributes"].should.equal( {"a": {"M": {"b": {"L": [{"S": "bar1"}, {"S": "bar2"}]}}}} ) result = client.get_item( TableName="TestTable", Key={"id": {"S": "nested_list_append"}} )["Item"] result.should.equal( { "id": {"S": "nested_list_append"}, "a": {"M": {"b": {"L": [{"S": "bar1"}, {"S": "bar2"}]}}}, } ) @mock_dynamodb2 def test_update_supports_multiple_levels_nested_list_append(): # Verify whether we can append a list that's inside a map that's inside a map (Inception!) client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], TableName="TestTable", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) client.put_item( TableName="TestTable", Item={ "id": {"S": "nested_list_append"}, "a": {"M": {"b": {"M": {"c": {"L": [{"S": "bar1"}]}}}}}, }, ) # Update item using list_append expression updated_item = client.update_item( TableName="TestTable", Key={"id": {"S": "nested_list_append"}}, UpdateExpression="SET a.#b.c = list_append(a.#b.#c, :i)", ExpressionAttributeValues={":i": {"L": [{"S": "bar2"}]}}, ExpressionAttributeNames={"#b": "b", "#c": "c"}, ReturnValues="UPDATED_NEW", ) # Verify updated item is correct updated_item["Attributes"].should.equal( {"a": {"M": {"b": {"M": {"c": {"L": [{"S": "bar1"}, {"S": "bar2"}]}}}}}} ) # Verify item is appended to the existing list result = client.get_item( TableName="TestTable", Key={"id": {"S": "nested_list_append"}} )["Item"] result.should.equal( { "id": {"S": "nested_list_append"}, "a": {"M": {"b": {"M": {"c": {"L": [{"S": "bar1"}, {"S": "bar2"}]}}}}}, } ) @mock_dynamodb2 def test_update_supports_nested_list_append_onto_another_list(): # Verify whether we can take the contents of one list, and use that to fill another list # Note that the contents of the other list is completely overwritten client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], TableName="TestTable", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) client.put_item( TableName="TestTable", Item={ "id": {"S": "list_append_another"}, "a": {"M": {"b": {"L": [{"S": "bar1"}]}, "c": {"L": [{"S": "car1"}]}}}, }, ) # Update item using list_append expression updated_item = client.update_item( TableName="TestTable", Key={"id": {"S": "list_append_another"}}, UpdateExpression="SET a.#c = list_append(a.#b, :i)", ExpressionAttributeValues={":i": {"L": [{"S": "bar2"}]}}, ExpressionAttributeNames={"#b": "b", "#c": "c"}, ReturnValues="UPDATED_NEW", ) # Verify updated item is correct updated_item["Attributes"].should.equal( {"a": {"M": {"c": {"L": [{"S": "bar1"}, {"S": "bar2"}]}}}} ) # Verify item is appended to the existing list result = client.get_item( TableName="TestTable", Key={"id": {"S": "list_append_another"}} )["Item"] result.should.equal( { "id": {"S": "list_append_another"}, "a": { "M": { "b": {"L": [{"S": "bar1"}]}, "c": {"L": [{"S": "bar1"}, {"S": "bar2"}]}, } }, } ) @mock_dynamodb2 def test_update_supports_list_append_maps(): client = boto3.client("dynamodb", region_name="us-west-1") client.create_table( AttributeDefinitions=[ {"AttributeName": "id", "AttributeType": "S"}, {"AttributeName": "rid", "AttributeType": "S"}, ], TableName="TestTable", KeySchema=[ {"AttributeName": "id", "KeyType": "HASH"}, {"AttributeName": "rid", "KeyType": "RANGE"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) client.put_item( TableName="TestTable", Item={ "id": {"S": "nested_list_append"}, "rid": {"S": "range_key"}, "a": {"L": [{"M": {"b": {"S": "bar1"}}}]}, }, ) # Update item using list_append expression updated_item = client.update_item( TableName="TestTable", Key={"id": {"S": "nested_list_append"}, "rid": {"S": "range_key"}}, UpdateExpression="SET a = list_append(a, :i)", ExpressionAttributeValues={":i": {"L": [{"M": {"b": {"S": "bar2"}}}]}}, ReturnValues="UPDATED_NEW", ) # Verify updated item is correct updated_item["Attributes"].should.equal( {"a": {"L": [{"M": {"b": {"S": "bar1"}}}, {"M": {"b": {"S": "bar2"}}}]}} ) # Verify item is appended to the existing list result = client.query( TableName="TestTable", KeyConditionExpression="id = :i AND begins_with(rid, :r)", ExpressionAttributeValues={ ":i": {"S": "nested_list_append"}, ":r": {"S": "range_key"}, }, )["Items"] result.should.equal( [ { "a": {"L": [{"M": {"b": {"S": "bar1"}}}, {"M": {"b": {"S": "bar2"}}}]}, "rid": {"S": "range_key"}, "id": {"S": "nested_list_append"}, } ] ) @mock_dynamodb2 def test_update_supports_list_append_with_nested_if_not_exists_operation(): dynamo = boto3.resource("dynamodb", region_name="us-west-1") table_name = "test" dynamo.create_table( TableName=table_name, AttributeDefinitions=[{"AttributeName": "Id", "AttributeType": "S"}], KeySchema=[{"AttributeName": "Id", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 20, "WriteCapacityUnits": 20}, ) table = dynamo.Table(table_name) table.put_item(Item={"Id": "item-id", "nest1": {"nest2": {}}}) updated_item = table.update_item( Key={"Id": "item-id"}, UpdateExpression="SET nest1.nest2.event_history = list_append(if_not_exists(nest1.nest2.event_history, :empty_list), :new_value)", ExpressionAttributeValues={":empty_list": [], ":new_value": ["some_value"]}, ReturnValues="UPDATED_NEW", ) # Verify updated item is correct updated_item["Attributes"].should.equal( {"nest1": {"nest2": {"event_history": ["some_value"]}}} ) table.get_item(Key={"Id": "item-id"})["Item"].should.equal( {"Id": "item-id", "nest1": {"nest2": {"event_history": ["some_value"]}}} ) @mock_dynamodb2 def test_update_supports_list_append_with_nested_if_not_exists_operation_and_property_already_exists(): dynamo = boto3.resource("dynamodb", region_name="us-west-1") table_name = "test" dynamo.create_table( TableName=table_name, AttributeDefinitions=[{"AttributeName": "Id", "AttributeType": "S"}], KeySchema=[{"AttributeName": "Id", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 20, "WriteCapacityUnits": 20}, ) table = dynamo.Table(table_name) table.put_item(Item={"Id": "item-id", "event_history": ["other_value"]}) updated_item = table.update_item( Key={"Id": "item-id"}, UpdateExpression="SET event_history = list_append(if_not_exists(event_history, :empty_list), :new_value)", ExpressionAttributeValues={":empty_list": [], ":new_value": ["some_value"]}, ReturnValues="UPDATED_NEW", ) # Verify updated item is correct updated_item["Attributes"].should.equal( {"event_history": ["other_value", "some_value"]} ) table.get_item(Key={"Id": "item-id"})["Item"].should.equal( {"Id": "item-id", "event_history": ["other_value", "some_value"]} ) @mock_dynamodb2 def test_update_catches_invalid_list_append_operation(): client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( AttributeDefinitions=[{"AttributeName": "SHA256", "AttributeType": "S"}], TableName="TestTable", KeySchema=[{"AttributeName": "SHA256", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) client.put_item( TableName="TestTable", Item={"SHA256": {"S": "sha-of-file"}, "crontab": {"L": [{"S": "bar1"}]}}, ) # Update item using invalid list_append expression with assert_raises(ParamValidationError) as ex: client.update_item( TableName="TestTable", Key={"SHA256": {"S": "sha-of-file"}}, UpdateExpression="SET crontab = list_append(crontab, :i)", ExpressionAttributeValues={":i": [{"S": "bar2"}]}, ) # Verify correct error is returned str(ex.exception).should.match("Parameter validation failed:") str(ex.exception).should.match( "Invalid type for parameter ExpressionAttributeValues." ) def _create_user_table(): client = boto3.client("dynamodb", region_name="us-east-1") client.create_table( TableName="users", KeySchema=[{"AttributeName": "username", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "username", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) client.put_item( TableName="users", Item={"username": {"S": "user1"}, "foo": {"S": "bar"}} ) client.put_item( TableName="users", Item={"username": {"S": "user2"}, "foo": {"S": "bar"}} ) client.put_item( TableName="users", Item={"username": {"S": "user3"}, "foo": {"S": "bar"}} ) return client @mock_dynamodb2 def test_update_item_if_original_value_is_none(): dynamo = boto3.resource("dynamodb", region_name="eu-central-1") dynamo.create_table( AttributeDefinitions=[{"AttributeName": "job_id", "AttributeType": "S"}], TableName="origin-rbu-dev", KeySchema=[{"AttributeName": "job_id", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) table = dynamo.Table("origin-rbu-dev") table.put_item(Item={"job_id": "a", "job_name": None}) table.update_item( Key={"job_id": "a"}, UpdateExpression="SET job_name = :output", ExpressionAttributeValues={":output": "updated"}, ) table.scan()["Items"][0]["job_name"].should.equal("updated") @mock_dynamodb2 def test_update_nested_item_if_original_value_is_none(): dynamo = boto3.resource("dynamodb", region_name="eu-central-1") dynamo.create_table( AttributeDefinitions=[{"AttributeName": "job_id", "AttributeType": "S"}], TableName="origin-rbu-dev", KeySchema=[{"AttributeName": "job_id", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) table = dynamo.Table("origin-rbu-dev") table.put_item(Item={"job_id": "a", "job_details": {"job_name": None}}) updated_item = table.update_item( Key={"job_id": "a"}, UpdateExpression="SET job_details.job_name = :output", ExpressionAttributeValues={":output": "updated"}, ReturnValues="UPDATED_NEW", ) # Verify updated item is correct updated_item["Attributes"].should.equal({"job_details": {"job_name": "updated"}}) table.scan()["Items"][0]["job_details"]["job_name"].should.equal("updated") @mock_dynamodb2 def test_allow_update_to_item_with_different_type(): dynamo = boto3.resource("dynamodb", region_name="eu-central-1") dynamo.create_table( AttributeDefinitions=[{"AttributeName": "job_id", "AttributeType": "S"}], TableName="origin-rbu-dev", KeySchema=[{"AttributeName": "job_id", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) table = dynamo.Table("origin-rbu-dev") table.put_item(Item={"job_id": "a", "job_details": {"job_name": {"nested": "yes"}}}) table.put_item(Item={"job_id": "b", "job_details": {"job_name": {"nested": "yes"}}}) updated_item = table.update_item( Key={"job_id": "a"}, UpdateExpression="SET job_details.job_name = :output", ExpressionAttributeValues={":output": "updated"}, ReturnValues="UPDATED_NEW", ) # Verify updated item is correct updated_item["Attributes"].should.equal({"job_details": {"job_name": "updated"}}) table.get_item(Key={"job_id": "a"})["Item"]["job_details"][ "job_name" ].should.be.equal("updated") table.get_item(Key={"job_id": "b"})["Item"]["job_details"][ "job_name" ].should.be.equal({"nested": "yes"}) @mock_dynamodb2 def test_query_catches_when_no_filters(): dynamo = boto3.resource("dynamodb", region_name="eu-central-1") dynamo.create_table( AttributeDefinitions=[{"AttributeName": "job_id", "AttributeType": "S"}], TableName="origin-rbu-dev", KeySchema=[{"AttributeName": "job_id", "KeyType": "HASH"}], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) table = dynamo.Table("origin-rbu-dev") with assert_raises(ClientError) as ex: table.query(TableName="original-rbu-dev") ex.exception.response["Error"]["Code"].should.equal("ValidationException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.exception.response["Error"]["Message"].should.equal( "Either KeyConditions or QueryFilter should be present" ) @mock_dynamodb2 def test_invalid_transact_get_items(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test1", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table = dynamodb.Table("test1") table.put_item( Item={"id": "1", "val": "1",} ) table.put_item( Item={"id": "1", "val": "2",} ) client = boto3.client("dynamodb", region_name="us-east-1") with assert_raises(ClientError) as ex: client.transact_get_items( TransactItems=[ {"Get": {"Key": {"id": {"S": "1"}}, "TableName": "test1"}} for i in range(26) ] ) ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.exception.response["Error"]["Message"].should.match( r"failed to satisfy constraint: Member must have length less than or equal to 25", re.I, ) with assert_raises(ClientError) as ex: client.transact_get_items( TransactItems=[ {"Get": {"Key": {"id": {"S": "1"},}, "TableName": "test1"}}, {"Get": {"Key": {"id": {"S": "1"},}, "TableName": "non_exists_table"}}, ] ) ex.exception.response["Error"]["Code"].should.equal("ResourceNotFoundException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.exception.response["Error"]["Message"].should.equal( "Requested resource not found" ) @mock_dynamodb2 def test_valid_transact_get_items(): dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test1", KeySchema=[ {"AttributeName": "id", "KeyType": "HASH"}, {"AttributeName": "sort_key", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "id", "AttributeType": "S"}, {"AttributeName": "sort_key", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table1 = dynamodb.Table("test1") table1.put_item( Item={"id": "1", "sort_key": "1",} ) table1.put_item( Item={"id": "1", "sort_key": "2",} ) dynamodb.create_table( TableName="test2", KeySchema=[ {"AttributeName": "id", "KeyType": "HASH"}, {"AttributeName": "sort_key", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "id", "AttributeType": "S"}, {"AttributeName": "sort_key", "AttributeType": "S"}, ], ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5}, ) table2 = dynamodb.Table("test2") table2.put_item( Item={"id": "1", "sort_key": "1",} ) client = boto3.client("dynamodb", region_name="us-east-1") res = client.transact_get_items( TransactItems=[ { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "1"}}, "TableName": "test1", } }, { "Get": { "Key": {"id": {"S": "non_exists_key"}, "sort_key": {"S": "2"}}, "TableName": "test1", } }, ] ) res["Responses"][0]["Item"].should.equal({"id": {"S": "1"}, "sort_key": {"S": "1"}}) len(res["Responses"]).should.equal(1) res = client.transact_get_items( TransactItems=[ { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "1"}}, "TableName": "test1", } }, { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "2"}}, "TableName": "test1", } }, { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "1"}}, "TableName": "test2", } }, ] ) res["Responses"][0]["Item"].should.equal({"id": {"S": "1"}, "sort_key": {"S": "1"}}) res["Responses"][1]["Item"].should.equal({"id": {"S": "1"}, "sort_key": {"S": "2"}}) res["Responses"][2]["Item"].should.equal({"id": {"S": "1"}, "sort_key": {"S": "1"}}) res = client.transact_get_items( TransactItems=[ { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "1"}}, "TableName": "test1", } }, { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "2"}}, "TableName": "test1", } }, { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "1"}}, "TableName": "test2", } }, ], ReturnConsumedCapacity="TOTAL", ) res["ConsumedCapacity"][0].should.equal( {"TableName": "test1", "CapacityUnits": 4.0, "ReadCapacityUnits": 4.0} ) res["ConsumedCapacity"][1].should.equal( {"TableName": "test2", "CapacityUnits": 2.0, "ReadCapacityUnits": 2.0} ) res = client.transact_get_items( TransactItems=[ { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "1"}}, "TableName": "test1", } }, { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "2"}}, "TableName": "test1", } }, { "Get": { "Key": {"id": {"S": "1"}, "sort_key": {"S": "1"}}, "TableName": "test2", } }, ], ReturnConsumedCapacity="INDEXES", ) res["ConsumedCapacity"][0].should.equal( { "TableName": "test1", "CapacityUnits": 4.0, "ReadCapacityUnits": 4.0, "Table": {"CapacityUnits": 4.0, "ReadCapacityUnits": 4.0,}, } ) res["ConsumedCapacity"][1].should.equal( { "TableName": "test2", "CapacityUnits": 2.0, "ReadCapacityUnits": 2.0, "Table": {"CapacityUnits": 2.0, "ReadCapacityUnits": 2.0,}, } ) @mock_dynamodb2 def test_gsi_verify_negative_number_order(): table_schema = { "KeySchema": [{"AttributeName": "partitionKey", "KeyType": "HASH"}], "GlobalSecondaryIndexes": [ { "IndexName": "GSI-K1", "KeySchema": [ {"AttributeName": "gsiK1PartitionKey", "KeyType": "HASH"}, {"AttributeName": "gsiK1SortKey", "KeyType": "RANGE"}, ], "Projection": {"ProjectionType": "KEYS_ONLY",}, } ], "AttributeDefinitions": [ {"AttributeName": "partitionKey", "AttributeType": "S"}, {"AttributeName": "gsiK1PartitionKey", "AttributeType": "S"}, {"AttributeName": "gsiK1SortKey", "AttributeType": "N"}, ], } item1 = { "partitionKey": "pk-1", "gsiK1PartitionKey": "gsi-k1", "gsiK1SortKey": Decimal("-0.6"), } item2 = { "partitionKey": "pk-2", "gsiK1PartitionKey": "gsi-k1", "gsiK1SortKey": Decimal("-0.7"), } item3 = { "partitionKey": "pk-3", "gsiK1PartitionKey": "gsi-k1", "gsiK1SortKey": Decimal("0.7"), } dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) table = dynamodb.Table("test-table") table.put_item(Item=item3) table.put_item(Item=item1) table.put_item(Item=item2) resp = table.query( KeyConditionExpression=Key("gsiK1PartitionKey").eq("gsi-k1"), IndexName="GSI-K1", ) # Items should be ordered with the lowest number first [float(item["gsiK1SortKey"]) for item in resp["Items"]].should.equal( [-0.7, -0.6, 0.7] ) @mock_dynamodb2 def test_transact_write_items_put(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) # Put multiple items dynamodb.transact_write_items( TransactItems=[ { "Put": { "Item": {"id": {"S": "foo{}".format(str(i))}, "foo": {"S": "bar"},}, "TableName": "test-table", } } for i in range(0, 5) ] ) # Assert all are present items = dynamodb.scan(TableName="test-table")["Items"] items.should.have.length_of(5) @mock_dynamodb2 def test_transact_write_items_put_conditional_expressions(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) dynamodb.put_item( TableName="test-table", Item={"id": {"S": "foo2"},}, ) # Put multiple items with assert_raises(ClientError) as ex: dynamodb.transact_write_items( TransactItems=[ { "Put": { "Item": { "id": {"S": "foo{}".format(str(i))}, "foo": {"S": "bar"}, }, "TableName": "test-table", "ConditionExpression": "#i <> :i", "ExpressionAttributeNames": {"#i": "id"}, "ExpressionAttributeValues": { ":i": { "S": "foo2" } # This item already exist, so the ConditionExpression should fail }, } } for i in range(0, 5) ] ) # Assert the exception is correct ex.exception.response["Error"]["Code"].should.equal("TransactionCanceledException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) # Assert all are present items = dynamodb.scan(TableName="test-table")["Items"] items.should.have.length_of(1) items[0].should.equal({"id": {"S": "foo2"}}) @mock_dynamodb2 def test_transact_write_items_conditioncheck_passes(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) # Insert an item without email address dynamodb.put_item( TableName="test-table", Item={"id": {"S": "foo"},}, ) # Put an email address, after verifying it doesn't exist yet dynamodb.transact_write_items( TransactItems=[ { "ConditionCheck": { "Key": {"id": {"S": "foo"}}, "TableName": "test-table", "ConditionExpression": "attribute_not_exists(#e)", "ExpressionAttributeNames": {"#e": "email_address"}, } }, { "Put": { "Item": { "id": {"S": "foo"}, "email_address": {"S": "test@moto.com"}, }, "TableName": "test-table", } }, ] ) # Assert all are present items = dynamodb.scan(TableName="test-table")["Items"] items.should.have.length_of(1) items[0].should.equal({"email_address": {"S": "test@moto.com"}, "id": {"S": "foo"}}) @mock_dynamodb2 def test_transact_write_items_conditioncheck_fails(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) # Insert an item with email address dynamodb.put_item( TableName="test-table", Item={"id": {"S": "foo"}, "email_address": {"S": "test@moto.com"}}, ) # Try to put an email address, but verify whether it exists # ConditionCheck should fail with assert_raises(ClientError) as ex: dynamodb.transact_write_items( TransactItems=[ { "ConditionCheck": { "Key": {"id": {"S": "foo"}}, "TableName": "test-table", "ConditionExpression": "attribute_not_exists(#e)", "ExpressionAttributeNames": {"#e": "email_address"}, } }, { "Put": { "Item": { "id": {"S": "foo"}, "email_address": {"S": "update@moto.com"}, }, "TableName": "test-table", } }, ] ) # Assert the exception is correct ex.exception.response["Error"]["Code"].should.equal("TransactionCanceledException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) # Assert the original email address is still present items = dynamodb.scan(TableName="test-table")["Items"] items.should.have.length_of(1) items[0].should.equal({"email_address": {"S": "test@moto.com"}, "id": {"S": "foo"}}) @mock_dynamodb2 def test_transact_write_items_delete(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) # Insert an item dynamodb.put_item( TableName="test-table", Item={"id": {"S": "foo"},}, ) # Delete the item dynamodb.transact_write_items( TransactItems=[ {"Delete": {"Key": {"id": {"S": "foo"}}, "TableName": "test-table",}} ] ) # Assert the item is deleted items = dynamodb.scan(TableName="test-table")["Items"] items.should.have.length_of(0) @mock_dynamodb2 def test_transact_write_items_delete_with_successful_condition_expression(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) # Insert an item without email address dynamodb.put_item( TableName="test-table", Item={"id": {"S": "foo"},}, ) # ConditionExpression will pass - no email address has been specified yet dynamodb.transact_write_items( TransactItems=[ { "Delete": { "Key": {"id": {"S": "foo"},}, "TableName": "test-table", "ConditionExpression": "attribute_not_exists(#e)", "ExpressionAttributeNames": {"#e": "email_address"}, } } ] ) # Assert the item is deleted items = dynamodb.scan(TableName="test-table")["Items"] items.should.have.length_of(0) @mock_dynamodb2 def test_transact_write_items_delete_with_failed_condition_expression(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) # Insert an item with email address dynamodb.put_item( TableName="test-table", Item={"id": {"S": "foo"}, "email_address": {"S": "test@moto.com"}}, ) # Try to delete an item that does not have an email address # ConditionCheck should fail with assert_raises(ClientError) as ex: dynamodb.transact_write_items( TransactItems=[ { "Delete": { "Key": {"id": {"S": "foo"},}, "TableName": "test-table", "ConditionExpression": "attribute_not_exists(#e)", "ExpressionAttributeNames": {"#e": "email_address"}, } } ] ) # Assert the exception is correct ex.exception.response["Error"]["Code"].should.equal("TransactionCanceledException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) # Assert the original item is still present items = dynamodb.scan(TableName="test-table")["Items"] items.should.have.length_of(1) items[0].should.equal({"email_address": {"S": "test@moto.com"}, "id": {"S": "foo"}}) @mock_dynamodb2 def test_transact_write_items_update(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) # Insert an item dynamodb.put_item(TableName="test-table", Item={"id": {"S": "foo"}}) # Update the item dynamodb.transact_write_items( TransactItems=[ { "Update": { "Key": {"id": {"S": "foo"}}, "TableName": "test-table", "UpdateExpression": "SET #e = :v", "ExpressionAttributeNames": {"#e": "email_address"}, "ExpressionAttributeValues": {":v": {"S": "test@moto.com"}}, } } ] ) # Assert the item is updated items = dynamodb.scan(TableName="test-table")["Items"] items.should.have.length_of(1) items[0].should.equal({"id": {"S": "foo"}, "email_address": {"S": "test@moto.com"}}) @mock_dynamodb2 def test_transact_write_items_update_with_failed_condition_expression(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) # Insert an item with email address dynamodb.put_item( TableName="test-table", Item={"id": {"S": "foo"}, "email_address": {"S": "test@moto.com"}}, ) # Try to update an item that does not have an email address # ConditionCheck should fail with assert_raises(ClientError) as ex: dynamodb.transact_write_items( TransactItems=[ { "Update": { "Key": {"id": {"S": "foo"}}, "TableName": "test-table", "UpdateExpression": "SET #e = :v", "ConditionExpression": "attribute_not_exists(#e)", "ExpressionAttributeNames": {"#e": "email_address"}, "ExpressionAttributeValues": {":v": {"S": "update@moto.com"}}, } } ] ) # Assert the exception is correct ex.exception.response["Error"]["Code"].should.equal("TransactionCanceledException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) # Assert the original item is still present items = dynamodb.scan(TableName="test-table")["Items"] items.should.have.length_of(1) items[0].should.equal({"email_address": {"S": "test@moto.com"}, "id": {"S": "foo"}}) @mock_dynamodb2 def test_dynamodb_max_1mb_limit(): ddb = boto3.resource("dynamodb", region_name="eu-west-1") table_name = "populated-mock-table" table = ddb.create_table( TableName=table_name, KeySchema=[ {"AttributeName": "partition_key", "KeyType": "HASH"}, {"AttributeName": "sort_key", "KeyType": "RANGE"}, ], AttributeDefinitions=[ {"AttributeName": "partition_key", "AttributeType": "S"}, {"AttributeName": "sort_key", "AttributeType": "S"}, ], BillingMode="PAY_PER_REQUEST", ) # Populate the table items = [ { "partition_key": "partition_key_val", # size=30 "sort_key": "sort_key_value____" + str(i), # size=30 } for i in range(10000, 29999) ] with table.batch_writer() as batch: for item in items: batch.put_item(Item=item) response = table.query( KeyConditionExpression=Key("partition_key").eq("partition_key_val") ) # We shouldn't get everything back - the total result set is well over 1MB len(items).should.be.greater_than(response["Count"]) response["LastEvaluatedKey"].shouldnt.be(None) def assert_raise_syntax_error(client_error, token, near): """ Assert whether a client_error is as expected Syntax error. Syntax error looks like: `syntax_error_template` Args: client_error(ClientError): The ClientError exception that was raised token(str): The token that ws unexpected near(str): The part in the expression that shows where the error occurs it generally has the preceding token the optional separation and the problematic token. """ syntax_error_template = ( 'Invalid UpdateExpression: Syntax error; token: "{token}", near: "{near}"' ) expected_syntax_error = syntax_error_template.format(token=token, near=near) assert client_error.response["Error"]["Code"] == "ValidationException" assert expected_syntax_error == client_error.response["Error"]["Message"] @mock_dynamodb2 def test_update_expression_with_numeric_literal_instead_of_value(): """ DynamoDB requires literals to be passed in as values. If they are put literally in the expression a token error will be raised """ dynamodb = boto3.client("dynamodb", region_name="eu-west-1") dynamodb.create_table( TableName="moto-test", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ) try: dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET MyStr = myNum + 1", ) assert False, "Validation exception not thrown" except dynamodb.exceptions.ClientError as e: assert_raise_syntax_error(e, "1", "+ 1") @mock_dynamodb2 def test_update_expression_with_multiple_set_clauses_must_be_comma_separated(): """ An UpdateExpression can have multiple set clauses but if they are passed in without the separating comma. """ dynamodb = boto3.client("dynamodb", region_name="eu-west-1") dynamodb.create_table( TableName="moto-test", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"}], ) try: dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET MyStr = myNum Mystr2 myNum2", ) assert False, "Validation exception not thrown" except dynamodb.exceptions.ClientError as e: assert_raise_syntax_error(e, "Mystr2", "myNum Mystr2 myNum2") @mock_dynamodb2 def test_list_tables_exclusive_start_table_name_empty(): client = boto3.client("dynamodb", region_name="us-east-1") resp = client.list_tables(Limit=1, ExclusiveStartTableName="whatever") len(resp["TableNames"]).should.equal(0) def assert_correct_client_error( client_error, code, message_template, message_values=None, braces=None ): """ Assert whether a client_error is as expected. Allow for a list of values to be passed into the message Args: client_error(ClientError): The ClientError exception that was raised code(str): The code for the error (e.g. ValidationException) message_template(str): Error message template. if message_values is not None then this template has a {values} as placeholder. For example: 'Value provided in ExpressionAttributeValues unused in expressions: keys: {values}' message_values(list of str|None): The values that are passed in the error message braces(list of str|None): List of length 2 with opening and closing brace for the values. By default it will be surrounded by curly brackets """ braces = braces or ["{", "}"] assert client_error.response["Error"]["Code"] == code if message_values is not None: values_string = "{open_brace}(?P.*){close_brace}".format( open_brace=braces[0], close_brace=braces[1] ) re_msg = re.compile(message_template.format(values=values_string)) match_result = re_msg.match(client_error.response["Error"]["Message"]) assert match_result is not None values_string = match_result.groupdict()["values"] values = [key for key in values_string.split(", ")] assert len(message_values) == len(values) for value in message_values: assert value in values else: assert client_error.response["Error"]["Message"] == message_template def create_simple_table_and_return_client(): dynamodb = boto3.client("dynamodb", region_name="eu-west-1") dynamodb.create_table( TableName="moto-test", KeySchema=[{"AttributeName": "id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "id", "AttributeType": "S"},], ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 1}, ) dynamodb.put_item( TableName="moto-test", Item={"id": {"S": "1"}, "myNum": {"N": "1"}, "MyStr": {"S": "1"},}, ) return dynamodb # https://github.com/spulec/moto/issues/2806 # https://docs.aws.amazon.com/amazondynamodb/latest/APIReference/API_UpdateItem.html # #DDB-UpdateItem-request-UpdateExpression @mock_dynamodb2 def test_update_item_with_attribute_in_right_hand_side_and_operation(): dynamodb = create_simple_table_and_return_client() dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET myNum = myNum+:val", ExpressionAttributeValues={":val": {"N": "3"}}, ) result = dynamodb.get_item(TableName="moto-test", Key={"id": {"S": "1"}}) assert result["Item"]["myNum"]["N"] == "4" dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET myNum = myNum - :val", ExpressionAttributeValues={":val": {"N": "1"}}, ) result = dynamodb.get_item(TableName="moto-test", Key={"id": {"S": "1"}}) assert result["Item"]["myNum"]["N"] == "3" @mock_dynamodb2 def test_non_existing_attribute_should_raise_exception(): """ Does error message get correctly raised if attribute is referenced but it does not exist for the item. """ dynamodb = create_simple_table_and_return_client() try: dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET MyStr = no_attr + MyStr", ) assert False, "Validation exception not thrown" except dynamodb.exceptions.ClientError as e: assert_correct_client_error( e, "ValidationException", "The provided expression refers to an attribute that does not exist in the item", ) @mock_dynamodb2 def test_update_expression_with_plus_in_attribute_name(): """ Does error message get correctly raised if attribute contains a plus and is passed in without an AttributeName. And lhs & rhs are not attribute IDs by themselve. """ dynamodb = create_simple_table_and_return_client() dynamodb.put_item( TableName="moto-test", Item={"id": {"S": "1"}, "my+Num": {"S": "1"}, "MyStr": {"S": "aaa"},}, ) try: dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET MyStr = my+Num", ) assert False, "Validation exception not thrown" except dynamodb.exceptions.ClientError as e: assert_correct_client_error( e, "ValidationException", "The provided expression refers to an attribute that does not exist in the item", ) @mock_dynamodb2 def test_update_expression_with_minus_in_attribute_name(): """ Does error message get correctly raised if attribute contains a minus and is passed in without an AttributeName. And lhs & rhs are not attribute IDs by themselve. """ dynamodb = create_simple_table_and_return_client() dynamodb.put_item( TableName="moto-test", Item={"id": {"S": "1"}, "my-Num": {"S": "1"}, "MyStr": {"S": "aaa"},}, ) try: dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET MyStr = my-Num", ) assert False, "Validation exception not thrown" except dynamodb.exceptions.ClientError as e: assert_correct_client_error( e, "ValidationException", "The provided expression refers to an attribute that does not exist in the item", ) @mock_dynamodb2 def test_update_expression_with_space_in_attribute_name(): """ Does error message get correctly raised if attribute contains a space and is passed in without an AttributeName. And lhs & rhs are not attribute IDs by themselves. """ dynamodb = create_simple_table_and_return_client() dynamodb.put_item( TableName="moto-test", Item={"id": {"S": "1"}, "my Num": {"S": "1"}, "MyStr": {"S": "aaa"},}, ) try: dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET MyStr = my Num", ) assert False, "Validation exception not thrown" except dynamodb.exceptions.ClientError as e: assert_raise_syntax_error(e, "Num", "my Num") @mock_dynamodb2 def test_summing_up_2_strings_raises_exception(): """ Update set supports different DynamoDB types but some operations are not supported. For example summing up 2 strings raises an exception. It results in ClientError with code ValidationException: Saying An operand in the update expression has an incorrect data type """ dynamodb = create_simple_table_and_return_client() try: dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET MyStr = MyStr + MyStr", ) assert False, "Validation exception not thrown" except dynamodb.exceptions.ClientError as e: assert_correct_client_error( e, "ValidationException", "An operand in the update expression has an incorrect data type", ) # https://github.com/spulec/moto/issues/2806 @mock_dynamodb2 def test_update_item_with_attribute_in_right_hand_side(): """ After tokenization and building expression make sure referenced attributes are replaced with their current value """ dynamodb = create_simple_table_and_return_client() # Make sure there are 2 values dynamodb.put_item( TableName="moto-test", Item={"id": {"S": "1"}, "myVal1": {"S": "Value1"}, "myVal2": {"S": "Value2"}}, ) dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET myVal1 = myVal2", ) result = dynamodb.get_item(TableName="moto-test", Key={"id": {"S": "1"}}) assert result["Item"]["myVal1"]["S"] == result["Item"]["myVal2"]["S"] == "Value2" @mock_dynamodb2 def test_multiple_updates(): dynamodb = create_simple_table_and_return_client() dynamodb.put_item( TableName="moto-test", Item={"id": {"S": "1"}, "myNum": {"N": "1"}, "path": {"N": "6"}}, ) dynamodb.update_item( TableName="moto-test", Key={"id": {"S": "1"}}, UpdateExpression="SET myNum = #p + :val, newAttr = myNum", ExpressionAttributeValues={":val": {"N": "1"}}, ExpressionAttributeNames={"#p": "path"}, ) result = dynamodb.get_item(TableName="moto-test", Key={"id": {"S": "1"}})["Item"] expected_result = { "myNum": {"N": "7"}, "newAttr": {"N": "1"}, "path": {"N": "6"}, "id": {"S": "1"}, } assert result == expected_result @mock_dynamodb2 def test_update_item_atomic_counter(): table = "table_t" ddb_mock = boto3.client("dynamodb", region_name="eu-west-3") ddb_mock.create_table( TableName=table, KeySchema=[{"AttributeName": "t_id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "t_id", "AttributeType": "S"}], BillingMode="PAY_PER_REQUEST", ) key = {"t_id": {"S": "item1"}} ddb_mock.put_item( TableName=table, Item={"t_id": {"S": "item1"}, "n_i": {"N": "5"}, "n_f": {"N": "5.3"}}, ) ddb_mock.update_item( TableName=table, Key=key, UpdateExpression="set n_i = n_i + :inc1, n_f = n_f + :inc2", ExpressionAttributeValues={":inc1": {"N": "1.2"}, ":inc2": {"N": "0.05"}}, ) updated_item = ddb_mock.get_item(TableName=table, Key=key)["Item"] updated_item["n_i"]["N"].should.equal("6.2") updated_item["n_f"]["N"].should.equal("5.35") @mock_dynamodb2 def test_update_item_atomic_counter_return_values(): table = "table_t" ddb_mock = boto3.client("dynamodb", region_name="eu-west-3") ddb_mock.create_table( TableName=table, KeySchema=[{"AttributeName": "t_id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "t_id", "AttributeType": "S"}], BillingMode="PAY_PER_REQUEST", ) key = {"t_id": {"S": "item1"}} ddb_mock.put_item(TableName=table, Item={"t_id": {"S": "item1"}, "v": {"N": "5"}}) response = ddb_mock.update_item( TableName=table, Key=key, UpdateExpression="set v = v + :inc", ExpressionAttributeValues={":inc": {"N": "1"}}, ReturnValues="UPDATED_OLD", ) assert ( "v" in response["Attributes"] ), "v has been updated, and should be returned here" response["Attributes"]["v"]["N"].should.equal("5") # second update response = ddb_mock.update_item( TableName=table, Key=key, UpdateExpression="set v = v + :inc", ExpressionAttributeValues={":inc": {"N": "1"}}, ReturnValues="UPDATED_OLD", ) assert ( "v" in response["Attributes"] ), "v has been updated, and should be returned here" response["Attributes"]["v"]["N"].should.equal("6") # third update response = ddb_mock.update_item( TableName=table, Key=key, UpdateExpression="set v = v + :inc", ExpressionAttributeValues={":inc": {"N": "1"}}, ReturnValues="UPDATED_NEW", ) assert ( "v" in response["Attributes"] ), "v has been updated, and should be returned here" response["Attributes"]["v"]["N"].should.equal("8") @mock_dynamodb2 def test_update_item_atomic_counter_from_zero(): table = "table_t" ddb_mock = boto3.client("dynamodb", region_name="eu-west-1") ddb_mock.create_table( TableName=table, KeySchema=[{"AttributeName": "t_id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "t_id", "AttributeType": "S"}], BillingMode="PAY_PER_REQUEST", ) key = {"t_id": {"S": "item1"}} ddb_mock.put_item( TableName=table, Item=key, ) ddb_mock.update_item( TableName=table, Key=key, UpdateExpression="add n_i :inc1, n_f :inc2", ExpressionAttributeValues={":inc1": {"N": "1.2"}, ":inc2": {"N": "-0.5"}}, ) updated_item = ddb_mock.get_item(TableName=table, Key=key)["Item"] assert updated_item["n_i"]["N"] == "1.2" assert updated_item["n_f"]["N"] == "-0.5" @mock_dynamodb2 def test_update_item_add_to_non_existent_set(): table = "table_t" ddb_mock = boto3.client("dynamodb", region_name="eu-west-1") ddb_mock.create_table( TableName=table, KeySchema=[{"AttributeName": "t_id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "t_id", "AttributeType": "S"}], BillingMode="PAY_PER_REQUEST", ) key = {"t_id": {"S": "item1"}} ddb_mock.put_item( TableName=table, Item=key, ) ddb_mock.update_item( TableName=table, Key=key, UpdateExpression="add s_i :s1", ExpressionAttributeValues={":s1": {"SS": ["hello"]}}, ) updated_item = ddb_mock.get_item(TableName=table, Key=key)["Item"] assert updated_item["s_i"]["SS"] == ["hello"] @mock_dynamodb2 def test_update_item_add_to_non_existent_number_set(): table = "table_t" ddb_mock = boto3.client("dynamodb", region_name="eu-west-1") ddb_mock.create_table( TableName=table, KeySchema=[{"AttributeName": "t_id", "KeyType": "HASH"}], AttributeDefinitions=[{"AttributeName": "t_id", "AttributeType": "S"}], BillingMode="PAY_PER_REQUEST", ) key = {"t_id": {"S": "item1"}} ddb_mock.put_item( TableName=table, Item=key, ) ddb_mock.update_item( TableName=table, Key=key, UpdateExpression="add s_i :s1", ExpressionAttributeValues={":s1": {"NS": ["3"]}}, ) updated_item = ddb_mock.get_item(TableName=table, Key=key)["Item"] assert updated_item["s_i"]["NS"] == ["3"] @mock_dynamodb2 def test_transact_write_items_fails_with_transaction_canceled_exception(): table_schema = { "KeySchema": [{"AttributeName": "id", "KeyType": "HASH"}], "AttributeDefinitions": [{"AttributeName": "id", "AttributeType": "S"},], } dynamodb = boto3.client("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) # Insert one item dynamodb.put_item(TableName="test-table", Item={"id": {"S": "foo"}}) # Update two items, the one that exists and another that doesn't with assert_raises(ClientError) as ex: dynamodb.transact_write_items( TransactItems=[ { "Update": { "Key": {"id": {"S": "foo"}}, "TableName": "test-table", "UpdateExpression": "SET #k = :v", "ConditionExpression": "attribute_exists(id)", "ExpressionAttributeNames": {"#k": "key"}, "ExpressionAttributeValues": {":v": {"S": "value"}}, } }, { "Update": { "Key": {"id": {"S": "doesnotexist"}}, "TableName": "test-table", "UpdateExpression": "SET #e = :v", "ConditionExpression": "attribute_exists(id)", "ExpressionAttributeNames": {"#e": "key"}, "ExpressionAttributeValues": {":v": {"S": "value"}}, } }, ] ) ex.exception.response["Error"]["Code"].should.equal("TransactionCanceledException") ex.exception.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400) ex.exception.response["Error"]["Message"].should.equal( "Transaction cancelled, please refer cancellation reasons for specific reasons [None, ConditionalCheckFailed]" ) @mock_dynamodb2 def test_gsi_projection_type_keys_only(): table_schema = { "KeySchema": [{"AttributeName": "partitionKey", "KeyType": "HASH"}], "GlobalSecondaryIndexes": [ { "IndexName": "GSI-K1", "KeySchema": [ {"AttributeName": "gsiK1PartitionKey", "KeyType": "HASH"}, {"AttributeName": "gsiK1SortKey", "KeyType": "RANGE"}, ], "Projection": {"ProjectionType": "KEYS_ONLY",}, } ], "AttributeDefinitions": [ {"AttributeName": "partitionKey", "AttributeType": "S"}, {"AttributeName": "gsiK1PartitionKey", "AttributeType": "S"}, {"AttributeName": "gsiK1SortKey", "AttributeType": "S"}, ], } item = { "partitionKey": "pk-1", "gsiK1PartitionKey": "gsi-pk", "gsiK1SortKey": "gsi-sk", "someAttribute": "lore ipsum", } dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) table = dynamodb.Table("test-table") table.put_item(Item=item) items = table.query( KeyConditionExpression=Key("gsiK1PartitionKey").eq("gsi-pk"), IndexName="GSI-K1", )["Items"] items.should.have.length_of(1) # Item should only include GSI Keys and Table Keys, as per the ProjectionType items[0].should.equal( { "gsiK1PartitionKey": "gsi-pk", "gsiK1SortKey": "gsi-sk", "partitionKey": "pk-1", } ) @mock_dynamodb2 def test_lsi_projection_type_keys_only(): table_schema = { "KeySchema": [ {"AttributeName": "partitionKey", "KeyType": "HASH"}, {"AttributeName": "sortKey", "KeyType": "RANGE"}, ], "LocalSecondaryIndexes": [ { "IndexName": "LSI", "KeySchema": [ {"AttributeName": "partitionKey", "KeyType": "HASH"}, {"AttributeName": "lsiK1SortKey", "KeyType": "RANGE"}, ], "Projection": {"ProjectionType": "KEYS_ONLY",}, } ], "AttributeDefinitions": [ {"AttributeName": "partitionKey", "AttributeType": "S"}, {"AttributeName": "sortKey", "AttributeType": "S"}, {"AttributeName": "lsiK1SortKey", "AttributeType": "S"}, ], } item = { "partitionKey": "pk-1", "sortKey": "sk-1", "lsiK1SortKey": "lsi-sk", "someAttribute": "lore ipsum", } dynamodb = boto3.resource("dynamodb", region_name="us-east-1") dynamodb.create_table( TableName="test-table", BillingMode="PAY_PER_REQUEST", **table_schema ) table = dynamodb.Table("test-table") table.put_item(Item=item) items = table.query( KeyConditionExpression=Key("partitionKey").eq("pk-1"), IndexName="LSI", )["Items"] items.should.have.length_of(1) # Item should only include GSI Keys and Table Keys, as per the ProjectionType items[0].should.equal( {"partitionKey": "pk-1", "sortKey": "sk-1", "lsiK1SortKey": "lsi-sk"} )