Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve code coverage

Browse Source
This commit is contained in:
Brian Rower 2017-09-12 23:41:12 -07:00
parent c1c592609a
commit 0097ab4c67
1 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests/test_dynamodb2/test_dynamodb_table_with_range_key.py
View File

@ -1470,6 +1470,22 @@ def test_update_item_add_with_expression():
).should.have.raised(ClientError) ).should.have.raised(ClientError)
dict(table.get_item(Key=item_key)['Item']).should.equal(current_item) dict(table.get_item(Key=item_key)['Item']).should.equal(current_item)
# Attempt to update with a bad expression
table.update_item.when.called_with(
Key=item_key,
UpdateExpression='ADD str_set bad_value'
).should.have.raised(ClientError)
# Attempt to add a string value instead of a string set
table.update_item.when.called_with(
Key=item_key,
UpdateExpression='ADD str_set :v',
ExpressionAttributeValues={
':v': 'new_string'
}
).should.have.raised(ClientError)
@mock_dynamodb2 @mock_dynamodb2
def test_update_item_delete_with_expression(): def test_update_item_delete_with_expression():
table = _create_table_with_range_key() table = _create_table_with_range_key()
@ -1518,6 +1534,22 @@ def test_update_item_delete_with_expression():
).should.have.raised(ClientError) ).should.have.raised(ClientError)
dict(table.get_item(Key=item_key)['Item']).should.equal(current_item) dict(table.get_item(Key=item_key)['Item']).should.equal(current_item)
# Try to delete a string set from a number set
table.update_item.when.called_with(
Key=item_key,
UpdateExpression='DELETE num_set :v',
ExpressionAttributeValues={
':v': {'del_str'}
}
).should.have.raised(ClientError)
dict(table.get_item(Key=item_key)['Item']).should.equal(current_item)
# Attempt to update with a bad expression
table.update_item.when.called_with(
Key=item_key,
UpdateExpression='DELETE num_val badvalue'
).should.have.raised(ClientError)
@mock_dynamodb2 @mock_dynamodb2
def test_boto3_query_gsi_range_comparison(): def test_boto3_query_gsi_range_comparison():