From d0f38407a1af857f2b23c986a0daa05fe396a729 Mon Sep 17 00:00:00 2001
From: Chris Keogh <chris.keogh@xero.com>
Date: Fri, 22 Sep 2017 15:50:08 +1200
Subject: [PATCH] update expression attribute test

---
 moto/dynamodb2/models.py              |  4 ++--
 tests/test_dynamodb2/test_dynamodb.py | 15 +++++++++++----
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/moto/dynamodb2/models.py b/moto/dynamodb2/models.py
index f21f9ebb..fde26972 100644
--- a/moto/dynamodb2/models.py
+++ b/moto/dynamodb2/models.py
@@ -412,8 +412,8 @@ class Table(BaseModel):
             return None
 
     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 = []
         if index_name:
             all_indexes = (self.global_indexes or []) + (self.indexes or [])
diff --git a/tests/test_dynamodb2/test_dynamodb.py b/tests/test_dynamodb2/test_dynamodb.py
index 4eb73730..2d58740f 100644
--- a/tests/test_dynamodb2/test_dynamodb.py
+++ b/tests/test_dynamodb2/test_dynamodb.py
@@ -390,24 +390,31 @@ def test_basic_projection_expressions_with_attr_expression_names():
     table.put_item(Item={
         'forum_name': 'the-key',
         'subject': '123',
-        'body': 'some test message'
+        'body': 'some test message',
+        'attachment': 'something'
     })
 
     table.put_item(Item={
         'forum_name': 'not-the-key',
         'subject': '123',
-        'body': 'some other test message'
+        '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, subject',
-        ExpressionAttributeNames={'#rl':'body'},
+        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'