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

update expression attribute test

Browse Source
This commit is contained in:
Chris Keogh 2017-09-22 15:50:08 +12:00
parent 87d7cacda6
commit d0f38407a1
2 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
moto/dynamodb2/models.py
View File

@ -412,8 +412,8 @@ class Table(BaseModel):
return None return None
def query(self, hash_key, range_comparison, range_objs, limit, def query(self, hash_key, range_comparison, range_objs, limit,
exclusive_start_key, scan_index_forward, projection_expression, index_name=None, **filter_kwargs): exclusive_start_key, scan_index_forward, projection_expression,
index_name=None, **filter_kwargs):
results = [] results = []
if index_name: if index_name:
all_indexes = (self.global_indexes or []) + (self.indexes or []) all_indexes = (self.global_indexes or []) + (self.indexes or [])

15
tests/test_dynamodb2/test_dynamodb.py
View File

@ -390,24 +390,31 @@ def test_basic_projection_expressions_with_attr_expression_names():
table.put_item(Item={ table.put_item(Item={
'forum_name': 'the-key', 'forum_name': 'the-key',
'subject': '123', 'subject': '123',
'body': 'some test message' 'body': 'some test message',
'attachment': 'something'
}) })
table.put_item(Item={ table.put_item(Item={
'forum_name': 'not-the-key', 'forum_name': 'not-the-key',
'subject': '123', 'subject': '123',
'body': 'some other test message' 'body': 'some other test message',
'attachment': 'something'
}) })
# Test a query returning all items # Test a query returning all items
results = table.query( results = table.query(
KeyConditionExpression=Key('forum_name').eq( KeyConditionExpression=Key('forum_name').eq(
'the-key'), 'the-key'),
ProjectionExpression='#rl, subject', ProjectionExpression='#rl, #rt, subject',
ExpressionAttributeNames={'#rl':'body'}, ExpressionAttributeNames={
'#rl': 'body',
'#rt': 'attachment'
},
) )
assert 'body' in results['Items'][0] assert 'body' in results['Items'][0]
assert results['Items'][0]['body'] == 'some test message' assert results['Items'][0]['body'] == 'some test message'
assert 'subject' in results['Items'][0] assert 'subject' in results['Items'][0]
assert results['Items'][0]['subject'] == '123' assert results['Items'][0]['subject'] == '123'
assert 'attachment' in results['Items'][0]
assert results['Items'][0]['attachment'] == 'something'