def test_create_table(self): """ TableConnection.create_table """ conn = TableConnection(self.test_table_name) kwargs = {"read_capacity_units": 1, "write_capacity_units": 1} self.assertRaises(ValueError, conn.create_table, **kwargs) kwargs["attribute_definitions"] = [ {"attribute_name": "key1", "attribute_type": "S"}, {"attribute_name": "key2", "attribute_type": "S"}, ] self.assertRaises(ValueError, conn.create_table, **kwargs) kwargs["key_schema"] = [ {"attribute_name": "key1", "key_type": "hash"}, {"attribute_name": "key2", "key_type": "range"}, ] params = { "TableName": "ci-table", "ProvisionedThroughput": {"WriteCapacityUnits": 1, "ReadCapacityUnits": 1}, "AttributeDefinitions": [ {"AttributeType": "S", "AttributeName": "key1"}, {"AttributeType": "S", "AttributeName": "key2"}, ], "KeySchema": [{"KeyType": "HASH", "AttributeName": "key1"}, {"KeyType": "RANGE", "AttributeName": "key2"}], } with patch(PATCH_METHOD) as req: req.return_value = {} conn.create_table(**kwargs) kwargs = req.call_args[0][1] self.assertEqual(kwargs, params)
def test_create_table_with_tags(self): conn = TableConnection(self.test_table_name) kwargs = { 'read_capacity_units': 1, 'write_capacity_units': 1, 'attribute_definitions': [{ 'attribute_name': 'key1', 'attribute_type': 'S' }, { 'attribute_name': 'key2', 'attribute_type': 'S' }], 'key_schema': [{ 'attribute_name': 'key1', 'key_type': 'hash' }, { 'attribute_name': 'key2', 'key_type': 'range' }], 'tags': { 'tag-key1': 'tag-value1', 'tag-key2': 'tag-value2', } } params = { 'TableName': 'ci-table', 'ProvisionedThroughput': { 'WriteCapacityUnits': 1, 'ReadCapacityUnits': 1 }, 'AttributeDefinitions': [{ 'AttributeType': 'S', 'AttributeName': 'key1' }, { 'AttributeType': 'S', 'AttributeName': 'key2' }], 'KeySchema': [{ 'KeyType': 'HASH', 'AttributeName': 'key1' }, { 'KeyType': 'RANGE', 'AttributeName': 'key2' }], 'Tags': [{ 'Key': 'tag-key1', 'Value': 'tag-value1' }, { 'Key': 'tag-key2', 'Value': 'tag-value2' }] } with patch(PATCH_METHOD) as req: req.return_value = {} conn.create_table(**kwargs) kwargs = req.call_args[0][1] self.assertEqual(kwargs, params)
def test_create_table(self): """ TableConnection.create_table """ conn = TableConnection(self.test_table_name) kwargs = { 'read_capacity_units': 1, 'write_capacity_units': 1, } self.assertRaises(ValueError, conn.create_table, **kwargs) kwargs['attribute_definitions'] = [{ 'attribute_name': 'key1', 'attribute_type': 'S' }, { 'attribute_name': 'key2', 'attribute_type': 'S' }] self.assertRaises(ValueError, conn.create_table, **kwargs) kwargs['key_schema'] = [{ 'attribute_name': 'key1', 'key_type': 'hash' }, { 'attribute_name': 'key2', 'key_type': 'range' }] params = { 'TableName': 'ci-table', 'BillingMode': PROVISIONED_BILLING_MODE, 'ProvisionedThroughput': { 'WriteCapacityUnits': 1, 'ReadCapacityUnits': 1 }, 'AttributeDefinitions': [{ 'AttributeType': 'S', 'AttributeName': 'key1' }, { 'AttributeType': 'S', 'AttributeName': 'key2' }], 'KeySchema': [{ 'KeyType': 'HASH', 'AttributeName': 'key1' }, { 'KeyType': 'RANGE', 'AttributeName': 'key2' }] } with patch(PATCH_METHOD) as req: req.return_value = {} conn.create_table(**kwargs) kwargs = req.call_args[0][1] self.assertEqual(kwargs, params)
def test_table_integration(ddb_url): table_name = 'pynamodb-ci-table' # For use with a fake dynamodb connection # See: http://aws.amazon.com/dynamodb/developer-resources/ conn = TableConnection(table_name, host=ddb_url) print(conn) print("conn.describe_table...") table = None try: table = conn.describe_table() except TableDoesNotExist: params = { 'read_capacity_units': 1, 'write_capacity_units': 1, 'attribute_definitions': [{ 'attribute_type': STRING, 'attribute_name': 'Forum' }, { 'attribute_type': STRING, 'attribute_name': 'Thread' }, { 'attribute_type': STRING, 'attribute_name': 'AltKey' }, { 'attribute_type': NUMBER, 'attribute_name': 'number' }], 'key_schema': [{ 'key_type': HASH, 'attribute_name': 'Forum' }, { 'key_type': RANGE, 'attribute_name': 'Thread' }], 'global_secondary_indexes': [{ 'index_name': 'alt-index', 'key_schema': [{ 'KeyType': 'HASH', 'AttributeName': 'AltKey' }], 'projection': { 'ProjectionType': 'KEYS_ONLY' }, 'provisioned_throughput': { 'ReadCapacityUnits': 1, 'WriteCapacityUnits': 1, } }], 'local_secondary_indexes': [{ 'index_name': 'view-index', 'key_schema': [{ 'KeyType': 'HASH', 'AttributeName': 'Forum' }, { 'KeyType': 'RANGE', 'AttributeName': 'AltKey' }], 'projection': { 'ProjectionType': 'KEYS_ONLY' } }] } print("conn.create_table...") conn.create_table(**params) while table is None: time.sleep(2) table = conn.describe_table() while table['TableStatus'] == 'CREATING': time.sleep(5) print(table['TableStatus']) table = conn.describe_table() print("conn.update_table...") conn.update_table(read_capacity_units=table.get( PROVISIONED_THROUGHPUT).get(READ_CAPACITY_UNITS) + 1, write_capacity_units=2) table = conn.describe_table() while table['TableStatus'] != 'ACTIVE': time.sleep(2) table = conn.describe_table() print("conn.put_item") conn.put_item( 'item1-hash', range_key='item1-range', attributes={'foo': { 'S': 'bar' }}, condition=NotExists(Path('Forum')), ) conn.get_item('item1-hash', range_key='item1-range') conn.delete_item('item1-hash', range_key='item1-range') items = [] for i in range(10): items.append({"Forum": "FooForum", "Thread": "thread-{}".format(i)}) print("conn.batch_write_items...") conn.batch_write_item(put_items=items) print("conn.batch_get_items...") data = conn.batch_get_item(items) print("conn.query...") conn.query( "FooForum", range_key_condition=(BeginsWith(Path('Thread'), Value('thread'))), ) print("conn.scan...") conn.scan() print("conn.delete_table...") conn.delete_table()
'KeyType': 'HASH', 'AttributeName': 'Forum' }, { 'KeyType': 'RANGE', 'AttributeName': 'AltKey' } ], 'projection': { 'ProjectionType': 'KEYS_ONLY' } } ] } print("conn.create_table...") conn.create_table(**params) while table is None: time.sleep(2) table = conn.describe_table() while table['TableStatus'] == 'CREATING': time.sleep(5) print(table['TableStatus']) table = conn.describe_table() print("conn.update_table...") conn.update_table( read_capacity_units=table.get(PROVISIONED_THROUGHPUT).get(READ_CAPACITY_UNITS) + 1, write_capacity_units=2 )
def test_create_table(self): """ TableConnection.create_table """ conn = TableConnection(self.test_table_name) kwargs = { 'read_capacity_units': 1, 'write_capacity_units': 1, } self.assertRaises(ValueError, conn.create_table, **kwargs) kwargs['attribute_definitions'] = [ { 'attribute_name': 'key1', 'attribute_type': 'S' }, { 'attribute_name': 'key2', 'attribute_type': 'S' } ] self.assertRaises(ValueError, conn.create_table, **kwargs) kwargs['key_schema'] = [ { 'attribute_name': 'key1', 'key_type': 'hash' }, { 'attribute_name': 'key2', 'key_type': 'range' } ] params = { 'TableName': 'ci-table', 'ProvisionedThroughput': { 'WriteCapacityUnits': 1, 'ReadCapacityUnits': 1 }, 'AttributeDefinitions': [ { 'AttributeType': 'S', 'AttributeName': 'key1' }, { 'AttributeType': 'S', 'AttributeName': 'key2' } ], 'KeySchema': [ { 'KeyType': 'HASH', 'AttributeName': 'key1' }, { 'KeyType': 'RANGE', 'AttributeName': 'key2' } ] } with patch(PATCH_METHOD) as req: req.return_value = {} conn.create_table( **kwargs ) kwargs = req.call_args[0][1] self.assertEqual(kwargs, params)
'index_name': 'view-index', 'key_schema': [{ 'KeyType': 'HASH', 'AttributeName': 'Forum' }, { 'KeyType': 'RANGE', 'AttributeName': 'AltKey' }], 'projection': { 'ProjectionType': 'KEYS_ONLY' } }] } print("conn.create_table...") conn.create_table(**params) while table is None: time.sleep(2) table = conn.describe_table() while table['TableStatus'] == 'CREATING': time.sleep(5) print(table['TableStatus']) table = conn.describe_table() print("conn.update_table...") conn.update_table(read_capacity_units=table.get(PROVISIONED_THROUGHPUT).get( READ_CAPACITY_UNITS) + 1, write_capacity_units=2) table = conn.describe_table()