From 6cf74742f84582b756a2d0e0a05b94027c8a118f Mon Sep 17 00:00:00 2001
From: Chris Keogh <chris.keogh@xero.com>
Date: Mon, 25 Sep 2017 11:44:10 +1300
Subject: [PATCH] add test for get_item return consumed capacity

---
 tests/test_dynamodb2/test_dynamodb.py | 51 ++++++++++++++++++++++++++-
 1 file changed, 50 insertions(+), 1 deletion(-)

diff --git a/tests/test_dynamodb2/test_dynamodb.py b/tests/test_dynamodb2/test_dynamodb.py
index 08cd9b56c..35c14f396 100644
--- a/tests/test_dynamodb2/test_dynamodb.py
+++ b/tests/test_dynamodb2/test_dynamodb.py
@@ -405,11 +405,11 @@ def test_basic_projection_expressions_with_attr_expression_names():
     results = table.query(
         KeyConditionExpression=Key('forum_name').eq(
             'the-key'),
+        ProjectionExpression='#rl, #rt, subject',
         ExpressionAttributeNames={
             '#rl': 'body',
             '#rt': 'attachment'
             },
-        ProjectionExpression='#rl, #rt, subject'
     )
 
     assert 'body' in results['Items'][0]
@@ -513,3 +513,52 @@ def test_update_item_returns_consumed_capacity():
     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']