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

Support update_item with map and numeric types.

Browse Source
This commit is contained in:
Paul Craciunoiu 2015-12-21 16:45:08 -07:00
parent fef748b1c1
commit f1099dd006
2 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
moto/dynamodb2/models.py
View File

@ -129,6 +129,10 @@ class Item(object):
# TODO deal with other types # TODO deal with other types
if isinstance(new_value, list) or isinstance(new_value, set): if isinstance(new_value, list) or isinstance(new_value, set):
self.attrs[attribute_name] = DynamoType({"SS": new_value}) self.attrs[attribute_name] = DynamoType({"SS": new_value})
elif isinstance(new_value, dict):
self.attrs[attribute_name] = DynamoType({"M": new_value})
elif update_action['Value'].keys() == ['N']:
self.attrs[attribute_name] = DynamoType({"N": new_value})
else: else:
self.attrs[attribute_name] = DynamoType({"S": new_value}) self.attrs[attribute_name] = DynamoType({"S": new_value})

11
tests/test_dynamodb2/test_dynamodb_table_with_range_key.py
View File

@ -807,6 +807,14 @@ def test_update_item_range_key_set():
'Action': u'PUT', 'Action': u'PUT',
'Value': 'johndoe2' 'Value': 'johndoe2'
}, },
'created': {
'Action': u'PUT',
'Value': Decimal('4'),
},
'mapfield': {
'Action': u'PUT',
'Value': {'key': 'value'},
}
}, },
) )
@ -815,7 +823,8 @@ def test_update_item_range_key_set():
'username': "johndoe2", 'username': "johndoe2",
'forum_name': 'the-key', 'forum_name': 'the-key',
'subject': '123', 'subject': '123',
'created': Decimal('3'), 'created': Decimal('4'),
'mapfield': {'key': 'value'},
}) })