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

Updated tests

Browse Source
This commit is contained in:
Ian Auld 2016-01-14 14:25:04 -08:00
parent 582db74dee
commit 06af5365ed
1 changed files with 28 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
tests/test_dynamodb2/test_dynamodb_table_with_range_key.py
View File

@ -13,7 +13,7 @@ from tests.helpers import requires_boto_gte
try: try:
from boto.dynamodb2.fields import GlobalAllIndex, HashKey, RangeKey, AllIndex from boto.dynamodb2.fields import GlobalAllIndex, HashKey, RangeKey, AllIndex
from boto.dynamodb2.table import Item, Table from boto.dynamodb2.table import Item, Table
from boto.dynamodb2.types import STRING from boto.dynamodb2.types import STRING, NUMBER
from boto.dynamodb2.exceptions import ValidationException from boto.dynamodb2.exceptions import ValidationException
from boto.dynamodb2.exceptions import ConditionalCheckFailedException from boto.dynamodb2.exceptions import ConditionalCheckFailedException
except ImportError: except ImportError:
@ -44,10 +44,10 @@ def create_table_with_local_indexes():
}, },
indexes=[ indexes=[
AllIndex( AllIndex(
'recipient_timestamp_index', 'threads_index',
parts=[ parts=[
HashKey('forum_name', data_type=STRING), HashKey('forum_name', data_type=STRING),
RangeKey('timestamp'), RangeKey('threads', data_type=NUMBER),
] ]
) )
] ]
@ -81,6 +81,7 @@ def test_create_table():
{'KeyType': 'HASH', 'AttributeName': 'forum_name'}, {'KeyType': 'HASH', 'AttributeName': 'forum_name'},
{'KeyType': 'RANGE', 'AttributeName': 'subject'} {'KeyType': 'RANGE', 'AttributeName': 'subject'}
], ],
'LocalSecondaryIndexes': [],
'ItemCount': 0, 'CreationDateTime': 1326499200.0, 'ItemCount': 0, 'CreationDateTime': 1326499200.0,
'GlobalSecondaryIndexes': [], 'GlobalSecondaryIndexes': [],
} }
@ -91,14 +92,14 @@ def test_create_table():
@requires_boto_gte("2.9") @requires_boto_gte("2.9")
@mock_dynamodb2 @mock_dynamodb2
@freeze_time("2012-01-14") @freeze_time("2012-01-14")
def test_create_table(): def test_create_table_with_local_index():
table = create_table_with_local_indexes() table = create_table_with_local_indexes()
expected = { expected = {
'Table': { 'Table': {
'AttributeDefinitions': [ 'AttributeDefinitions': [
{'AttributeName': 'forum_name', 'AttributeType': 'S'}, {'AttributeName': 'forum_name', 'AttributeType': 'S'},
{'AttributeName': 'subject', 'AttributeType': 'S'}, {'AttributeName': 'subject', 'AttributeType': 'S'},
{'AttributeName': 'timestamp', 'AttributeType': 'S'} {'AttributeName': 'threads', 'AttributeType': 'N'}
], ],
'ProvisionedThroughput': { 'ProvisionedThroughput': {
'NumberOfDecreasesToday': 0, 'NumberOfDecreasesToday': 0,
@ -114,10 +115,10 @@ def test_create_table():
], ],
'LocalSecondaryIndexes': [ 'LocalSecondaryIndexes': [
{ {
'IndexName': 'recipient_timestamp_index', 'IndexName': 'threads_index',
'KeySchema': [ 'KeySchema': [
{'AttributeName': 'forum_name', 'KeyType': 'HASH'}, {'AttributeName': 'forum_name', 'KeyType': 'HASH'},
{'AttributeName': 'timestamp', 'KeyType': 'RANGE'} {'AttributeName': 'threads', 'KeyType': 'RANGE'}
], ],
'Projection': {'ProjectionType': 'ALL'} 'Projection': {'ProjectionType': 'ALL'}
} }
@ -637,6 +638,26 @@ def test_query_with_global_indexes():
list(results).should.have.length_of(0) list(results).should.have.length_of(0)
@mock_dynamodb2
def test_query_with_local_indexes():
table = create_table_with_local_indexes()
item_data = {
'forum_name': 'Cool Forum',
'subject': 'Check this out!',
'version': '1',
'threads': 1,
'status': 'inactive'
}
item = Item(table, item_data)
item.save(overwrite=True)
item['version'] = '2'
item.save(overwrite=True)
# Revisit this query once support for QueryFilter is added
results = table.query(forum_name__eq='Cool Forum', index='threads_index')
list(results).should.have.length_of(1)
@mock_dynamodb2 @mock_dynamodb2
def test_reverse_query(): def test_reverse_query():
conn = boto.dynamodb2.layer1.DynamoDBConnection() conn = boto.dynamodb2.layer1.DynamoDBConnection()