Updating dynamodb2 tests for newer boto.
This commit is contained in:
		
							parent
							
								
									415291d11a
								
							
						
					
					
						commit
						12d61ede41
					
				| @ -4,13 +4,10 @@ python: | ||||
|   - 2.7 | ||||
| env: | ||||
|     matrix: | ||||
|       - BOTO_VERSION==2.31.1 | ||||
|       - BOTO_VERSION=2.25.0 | ||||
|       - BOTO_VERSION=2.19.0 | ||||
|       - BOTO_VERSION=2.12.0 | ||||
|       - BOTO_VERSION=2.11.0 | ||||
|       - BOTO_VERSION=2.10.0 | ||||
|       - BOTO_VERSION=2.9.9 | ||||
|       - BOTO_VERSION=2.8 | ||||
|       - BOTO_VERSION=2.7 | ||||
| install: | ||||
|   - pip install boto==$BOTO_VERSION | ||||
|  | ||||
| @ -9,11 +9,10 @@ try: | ||||
|     from boto.dynamodb2.fields import RangeKey | ||||
|     from boto.dynamodb2.table import Table | ||||
|     from boto.dynamodb2.table import Item | ||||
|     from boto.dynamodb.exceptions import DynamoDBKeyNotFoundError | ||||
|     from boto.dynamodb2.exceptions import ValidationException | ||||
|     from boto.dynamodb2.exceptions import ConditionalCheckFailedException | ||||
| except ImportError: | ||||
|     print "This boto version is not supported" | ||||
|     pass | ||||
| 
 | ||||
| 
 | ||||
| def create_table(): | ||||
|     table = Table.create('messages', schema=[ | ||||
| @ -25,10 +24,10 @@ def create_table(): | ||||
|     }) | ||||
|     return table | ||||
| 
 | ||||
| 
 | ||||
| def iterate_results(res): | ||||
|     for i in res: | ||||
|         print i | ||||
| 
 | ||||
|         pass | ||||
| 
 | ||||
| 
 | ||||
| @requires_boto_gte("2.9") | ||||
| @ -208,13 +207,12 @@ def test_delete_item(): | ||||
|     response.should.equal(True) | ||||
| 
 | ||||
|     table.count().should.equal(0) | ||||
|     item.delete.when.called_with().should.throw(ConditionalCheckFailedException) | ||||
|     item.delete().should.equal(False) | ||||
| 
 | ||||
| 
 | ||||
| @requires_boto_gte("2.9") | ||||
| @mock_dynamodb2 | ||||
| def test_delete_item_with_undeclared_table(): | ||||
|     conn = boto.connect_dynamodb() | ||||
|     table = Table("undeclared-table") | ||||
|     item_data = { | ||||
|         'forum_name': 'LOLCat Forum', | ||||
| @ -430,14 +428,18 @@ def test_batch_read(): | ||||
|     item_data['forum_name'] = 'another-key' | ||||
|     item_data['subject'] = '789' | ||||
|     item.save() | ||||
|     results = table.batch_get(keys=[ | ||||
|     results = table.batch_get( | ||||
|         keys=[ | ||||
|             {'forum_name': 'the-key', 'subject': '123'}, | ||||
|                 {'forum_name': 'another-key', 'subject': '789'}]) | ||||
|             {'forum_name': 'another-key', 'subject': '789'}, | ||||
|         ] | ||||
|     ) | ||||
| 
 | ||||
|     # Iterate through so that batch_item gets called | ||||
|     count = len([x for x in results]) | ||||
|     count.should.equal(2) | ||||
| 
 | ||||
| 
 | ||||
| @requires_boto_gte("2.9") | ||||
| @mock_dynamodb2 | ||||
| def test_get_key_fields(): | ||||
|  | ||||
| @ -6,11 +6,11 @@ from moto import mock_dynamodb2 | ||||
| from tests.helpers import requires_boto_gte | ||||
| try: | ||||
|     from boto.dynamodb2.fields import HashKey | ||||
|     from boto.dynamodb2.fields import RangeKey | ||||
|     from boto.dynamodb2.table import Table | ||||
|     from boto.dynamodb2.table import Item | ||||
| except ImportError: | ||||
|     print "This boto version is not supported" | ||||
|     pass | ||||
| 
 | ||||
| 
 | ||||
| def create_table(): | ||||
|     table = Table.create('messages', schema=[ | ||||
| @ -22,12 +22,11 @@ def create_table(): | ||||
|     return table | ||||
| 
 | ||||
| 
 | ||||
| 
 | ||||
| @requires_boto_gte("2.9") | ||||
| @mock_dynamodb2 | ||||
| @freeze_time("2012-01-14") | ||||
| def test_create_table(): | ||||
|     table = create_table() | ||||
|     create_table() | ||||
|     expected = { | ||||
|         'Table': { | ||||
|             'AttributeDefinitions': [ | ||||
| @ -48,7 +47,8 @@ def test_create_table(): | ||||
|     conn = boto.dynamodb2.connect_to_region( | ||||
|         'us-west-2', | ||||
|         aws_access_key_id="ak", | ||||
|         aws_secret_access_key="sk") | ||||
|         aws_secret_access_key="sk" | ||||
|     ) | ||||
| 
 | ||||
|     conn.describe_table('messages').should.equal(expected) | ||||
| 
 | ||||
| @ -78,7 +78,6 @@ def test_update_table_throughput(): | ||||
|         'write': 6, | ||||
|     }) | ||||
| 
 | ||||
| 
 | ||||
|     table.throughput["read"].should.equal(5) | ||||
|     table.throughput["write"].should.equal(6) | ||||
| 
 | ||||
| @ -172,7 +171,7 @@ def test_delete_item(): | ||||
| 
 | ||||
|     table.count().should.equal(0) | ||||
| 
 | ||||
|     item.delete.when.called_with().should.throw(JSONResponseError) | ||||
|     item.delete().should.equal(False) | ||||
| 
 | ||||
| 
 | ||||
| @requires_boto_gte("2.9") | ||||
| @ -346,9 +345,12 @@ def test_batch_read(): | ||||
|     item_data['forum_name'] = 'another-key' | ||||
|     item.save(overwrite=True) | ||||
| 
 | ||||
|     results = table.batch_get(keys=[ | ||||
|     results = table.batch_get( | ||||
|         keys=[ | ||||
|             {'forum_name': 'the-key1'}, | ||||
|                 {'forum_name': 'another-key'}]) | ||||
|             {'forum_name': 'another-key'}, | ||||
|         ] | ||||
|     ) | ||||
| 
 | ||||
|     # Iterate through so that batch_item gets called | ||||
|     count = len([x for x in results]) | ||||
| @ -380,4 +382,3 @@ def test_get_special_item(): | ||||
|     table.put_item(data=data) | ||||
|     returned_item = table.get_item(**{'date-joined': 127549192}) | ||||
|     dict(returned_item).should.equal(data) | ||||
|      | ||||
|  | ||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user