AWSiOSSDKTests/AWSDynamoDBTests.m

/*
 * Copyright 2010-2014 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file is distributed
 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

#if AWS_TEST_DYNAMODB

#import <XCTest/XCTest.h>
#import "DynamoDB.h"
#import "AWSTestUtility.h"

NSString *const AWSDynamoDBTestTable1 = @"AWSSDKForiOSv2Test1";
NSString *const AWSDynamoDBTestTable2 = @"AWSSDKForiOSv2Test2";

@interface AWSDynamoDBTests : XCTestCase

@end

@implementation AWSDynamoDBTests

static NSString *table1Name = nil;
static NSString *table2Name = nil;

#pragma mark - Setups

+ (void)setUp {
    [super setUp];
    [AWSTestUtility setupCognitoCredentialsProvider];

    NSTimeInterval timeIntervalSinceReferenceDate = [NSDate timeIntervalSinceReferenceDate];
    table1Name = [NSString stringWithFormat:@"%@-%f", AWSDynamoDBTestTable1, timeIntervalSinceReferenceDate];
    table2Name = [NSString stringWithFormat:@"%@-%f", AWSDynamoDBTestTable2, timeIntervalSinceReferenceDate];

    [self createTable:table1Name];
}

- (void)setUp {
    [super setUp];
}

- (void)tearDown {
    [super tearDown];
}

+ (void)tearDown {
    [super tearDown];

    [self deleteTable:table1Name];
}

#pragma mark - Utilities

+ (BFTask *)waitForTableToBeActive:(NSString *)tableName {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    BFTask *task = [BFTask taskWithResult:nil];
    for(int32_t i = 0; i < 16; i++) {
        task = [task continueWithSuccessBlock:^id(BFTask *task) {
            if (task.result) {
                AWSDynamoDBDescribeTableOutput *describeTableOutput = task.result;
                AWSDynamoDBTableStatus tableStatus = describeTableOutput.table.tableStatus;
                if (tableStatus == AWSDynamoDBTableStatusActive) {
                    return task;
                }
            }

            sleep(15);

            AWSDynamoDBDescribeTableInput *describeTableInput = [AWSDynamoDBDescribeTableInput new];
            describeTableInput.tableName = tableName;
            return [dynamoDB describeTable:describeTableInput];
        }];
    }

    return task;
}

+ (BFTask *)waitForTableToBeDeleted:(NSString *)tableName {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    BFTask *task = [BFTask taskWithResult:nil];
    for(int32_t i = 0; i < 16; i ++) {
        task = [task continueWithSuccessBlock:^id(BFTask *task) {
            sleep(15);

            AWSDynamoDBDescribeTableInput *describeTableInput = [AWSDynamoDBDescribeTableInput new];
            describeTableInput.tableName = tableName;
            return [dynamoDB describeTable:describeTableInput];
        }];
    }

    return task;
}

+ (BOOL)createTable:(NSString *)tableName {
    __block BOOL succeeded = NO;
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    AWSDynamoDBDescribeTableInput *describeTableInput = [AWSDynamoDBDescribeTableInput new];
    describeTableInput.tableName = tableName;

    [[[[[dynamoDB describeTable:describeTableInput
         ] continueWithBlock:^id(BFTask *task) {
        if ([task.error.domain isEqualToString:AWSDynamoDBErrorDomain]
            && task.error.code == AWSDynamoDBErrorResourceNotFound) {
            AWSDynamoDBAttributeDefinition *attributeDefinition = [AWSDynamoDBAttributeDefinition new];
            attributeDefinition.attributeName = @"hashKey";
            attributeDefinition.attributeType = AWSDynamoDBScalarAttributeTypeS;

            AWSDynamoDBKeySchemaElement *keySchemaElement = [AWSDynamoDBKeySchemaElement new];
            keySchemaElement.attributeName = @"hashKey";
            keySchemaElement.keyType = AWSDynamoDBKeyTypeHash;

            AWSDynamoDBProvisionedThroughput *provisionedThroughput = [AWSDynamoDBProvisionedThroughput new];
            provisionedThroughput.readCapacityUnits = @1;
            provisionedThroughput.writeCapacityUnits = @1;

            AWSDynamoDBCreateTableInput *createTableInput = [AWSDynamoDBCreateTableInput new];
            createTableInput.tableName = tableName;
            createTableInput.attributeDefinitions = @[attributeDefinition];
            createTableInput.keySchema = @[keySchemaElement];
            createTableInput.provisionedThroughput = provisionedThroughput;

            return [dynamoDB createTable:createTableInput];
        }

        return nil;
    }] continueWithSuccessBlock:^id(BFTask *task) {
        return [self waitForTableToBeActive:tableName];
    }] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            succeeded = NO;
        } else {
            succeeded = YES;
        }

        return nil;
    }] waitUntilFinished];

    return succeeded;
}

+ (BOOL)deleteTable:(NSString *)tableName {
    __block BOOL succeeded = NO;
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    AWSDynamoDBDeleteTableInput *deleteTableInput = [AWSDynamoDBDeleteTableInput new];
    deleteTableInput.tableName = tableName;

    [[[[dynamoDB deleteTable:deleteTableInput
        ] continueWithSuccessBlock:^id(BFTask *task) {
        return [self waitForTableToBeDeleted:tableName];
    }] continueWithBlock:^id(BFTask *task) {
        if (task.error
            && (![task.error.domain isEqualToString:AWSDynamoDBErrorDomain]
                || task.error.code != AWSDynamoDBErrorResourceNotFound)) {
                succeeded = NO;
            } else {
                succeeded = YES;
            }

        return nil;
    }] waitUntilFinished];

    return succeeded;
}

#pragma mark - Tests

- (void)testDescribeTable {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    AWSDynamoDBDescribeTableInput *describeTableInput = [AWSDynamoDBDescribeTableInput new];
    describeTableInput.tableName = table1Name;

    [[[dynamoDB describeTable:describeTableInput
       ] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        } else {
            AWSDynamoDBDescribeTableOutput *describeTableOutput = task.result;
            AWSDynamoDBTableStatus tableStatus = describeTableOutput.table.tableStatus;
            XCTAssertEqual(tableStatus, AWSDynamoDBTableStatusActive);
        }

        return nil;
    }] waitUntilFinished];
}

- (void)testCancelDescribeTable {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    AWSDynamoDBDescribeTableInput *describeTableInput = [AWSDynamoDBDescribeTableInput new];
    describeTableInput.tableName = table1Name;

    BFTask *task = [[dynamoDB describeTable:describeTableInput] continueWithBlock:^id(BFTask *task) {
        if (!([task.error.domain isEqualToString:NSURLErrorDomain]
              && task.error.code == NSURLErrorCancelled)) {
            XCTFail(@"Failed to cancel a request. task: [%@]", task);
        }

        return nil;
    }];

    [describeTableInput cancel];

    [task waitUntilFinished];
}

- (void)testListTables {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    AWSDynamoDBListTablesInput *listTablesInput = [AWSDynamoDBListTablesInput new];

    [[[dynamoDB listTables:listTablesInput
       ] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        } else {
            AWSDynamoDBListTablesOutput *listTableOutput = task.result;
            NSArray *tableNames = listTableOutput.tableNames;
            XCTAssertTrue([tableNames containsObject:table1Name]);
        }

        return nil;
    }] waitUntilFinished];
}

- (void)testPutItem {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    AWSDynamoDBPutItemInput *putItemInput = [AWSDynamoDBPutItemInput new];
    putItemInput.tableName = table1Name;
    AWSDynamoDBAttributeValue *hashValue = [AWSDynamoDBAttributeValue new];
    hashValue.S = @"testPutItem";
    AWSDynamoDBAttributeValue *otherValue = [AWSDynamoDBAttributeValue new];
    otherValue.S = @"otherValue";
    putItemInput.item = @{
                          @"hashKey" : hashValue,
                          @"otherKey" : otherValue
                          };

    putItemInput.returnValues = AWSDynamoDBReturnValueAllOld;

    [[[dynamoDB putItem:putItemInput] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        } else {
            AWSDynamoDBPutItemOutput *putItemOutput = task.result;
            AZLogDebug(@"Result of putItemOutput is:%@",[putItemOutput description]);
            //XCTAssertNotNil(putItemOutput, @"putItemOutput should NOT be nil!");
        }

        return nil;
    }] waitUntilFinished];
}

- (void) testGetItem {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    NSString *myKey = @"testGetItem";
    NSString *myValue = @"otherValue";

    AWSDynamoDBPutItemInput *putItemInput = [AWSDynamoDBPutItemInput new];
    putItemInput.tableName = table1Name;
    AWSDynamoDBAttributeValue *hashValue = [AWSDynamoDBAttributeValue new];
    hashValue.S = myKey;
    AWSDynamoDBAttributeValue *otherValue = [AWSDynamoDBAttributeValue new];
    otherValue.S = myValue;
    putItemInput.item = @{
                          @"hashKey" : hashValue,
                          @"otherKey" : otherValue
                          };
    putItemInput.returnValues = AWSDynamoDBReturnValueAllOld;

    [[[[dynamoDB putItem:putItemInput
        ] continueWithSuccessBlock:^id(BFTask *task) {

        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        }

        AWSDynamoDBGetItemInput *getItemInput = [AWSDynamoDBGetItemInput new];
        getItemInput.tableName = table1Name;

        AWSDynamoDBAttributeValue *hashValue = [AWSDynamoDBAttributeValue new];
        hashValue.S = myKey;

        getItemInput.key = @{@"hashKey": hashValue};

        return [dynamoDB getItem:getItemInput];
    }] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        } else {
            AWSDynamoDBGetItemOutput *getItemOutput = task.result;
            NSDictionary *items = getItemOutput.item;
            XCTAssertEqualObjects(hashValue, [items objectForKey:@"hashKey"] );
            XCTAssertEqualObjects(otherValue,[items objectForKey:@"otherKey"] );
        }

        return nil;
    }] waitUntilFinished];
}

- (void)testDeleteItem {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    AWSDynamoDBDeleteItemInput *deleteItemInput = [AWSDynamoDBDeleteItemInput new];
    deleteItemInput.tableName = table1Name;

    NSString *myKey = @"testGetItem";
    AWSDynamoDBAttributeValue *hashValue = [AWSDynamoDBAttributeValue new];
    hashValue.S = myKey;

    deleteItemInput.key = @{@"hashKey": hashValue};
    deleteItemInput.returnConsumedCapacity = AWSDynamoDBReturnConsumedCapacityTotal;
    deleteItemInput.returnItemCollectionMetrics = AWSDynamoDBReturnItemCollectionMetricsSize;

    [[[dynamoDB deleteItem:deleteItemInput
       ] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        } else {
            AWSDynamoDBDeleteItemOutput *deleteItemOutput = task.result;
            XCTAssertNotNil(deleteItemOutput.consumedCapacity, @"deleteItemOutput.consumedCapacity should not be NIL!");
        }

        return nil;
    }] waitUntilFinished];
}

- (void)testQuery {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    NSString *myKey = @"testPutItemForQuery";
    NSString *myValue = @"otherValue";

    AWSDynamoDBPutItemInput *putItemInput = [AWSDynamoDBPutItemInput new];
    putItemInput.tableName = table1Name;
    AWSDynamoDBAttributeValue *hashValue = [AWSDynamoDBAttributeValue new];
    hashValue.S = myKey;
    AWSDynamoDBAttributeValue *otherValue = [AWSDynamoDBAttributeValue new];
    otherValue.S = myValue;
    putItemInput.item = @{
                          @"hashKey" : hashValue,
                          @"otherKey" : otherValue
                          };

    [[[[dynamoDB putItem:putItemInput
        ] continueWithSuccessBlock:^id(BFTask *task) {
        XCTAssertNil(task.error);
        XCTAssertTrue([task.result isKindOfClass:[AWSDynamoDBPutItemOutput class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBPutItemOutput class]));

        AWSDynamoDBCondition *conditon = [AWSDynamoDBCondition new];
        conditon.attributeValueList = @[hashValue];
        conditon.comparisonOperator = AWSDynamoDBComparisonOperatorEQ;

        AWSDynamoDBQueryInput *queryInput = [AWSDynamoDBQueryInput new];
        queryInput.tableName = table1Name;
        queryInput.consistentRead = @YES;
        queryInput.keyConditions = @{@"hashKey": conditon};

        return [dynamoDB query:queryInput];
    }] continueWithBlock:^id(BFTask *task) {
        XCTAssertNil(task.error);
        XCTAssertTrue([task.result isKindOfClass:[AWSDynamoDBQueryOutput class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBQueryOutput class]));

        AWSDynamoDBQueryOutput *queryOutput = task.result;
        XCTAssertEqualObjects(queryOutput.count, @1, @"number of result from query is not equal to One");

        NSDictionary *itemDictionary = [queryOutput.items firstObject];

        XCTAssert([[itemDictionary objectForKey:@"hashKey"] isKindOfClass:[AWSDynamoDBAttributeValue class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBAttributeValue class]));
        AWSDynamoDBAttributeValue *resultAttributeValueHashKey = [itemDictionary objectForKey:@"hashKey"];
        XCTAssertEqualObjects(resultAttributeValueHashKey, hashValue);

        XCTAssert([[itemDictionary objectForKey:@"otherKey"] isKindOfClass:[AWSDynamoDBAttributeValue class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBAttributeValue class]));
        AWSDynamoDBAttributeValue *resultAttributeValueOtherKey = [itemDictionary objectForKey:@"otherKey"];
        XCTAssertEqualObjects(resultAttributeValueOtherKey, otherValue);

        return nil;
    }] waitUntilFinished];
}

- (void)testScan {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    AWSDynamoDBScanInput *scanInput = [AWSDynamoDBScanInput new];
    scanInput.tableName = table1Name;

    NSString *key = @"testScan";

    //Put item
    AWSDynamoDBPutItemInput *putItemInput = [AWSDynamoDBPutItemInput new];
    putItemInput.tableName = table1Name;
    AWSDynamoDBAttributeValue *hashValue = [AWSDynamoDBAttributeValue new];
    hashValue.S = key;
    AWSDynamoDBAttributeValue *otherValue = [AWSDynamoDBAttributeValue new];
    otherValue.S = @"otherValue";
    putItemInput.item = @{
                          @"hashKey" : hashValue,
                          @"otherKey" : otherValue
                          };

    [[[[dynamoDB putItem:putItemInput
        ] continueWithSuccessBlock:^id(BFTask *task) {
        XCTAssertTrue([task.result isKindOfClass:[AWSDynamoDBPutItemOutput class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBPutItemOutput class]));

        return [dynamoDB scan:scanInput];
    }] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"Scan operation failed. error: [%@]", task.error);
        } else {
            AWSDynamoDBScanOutput *scanOutput = task.result;


            if ([scanOutput.count intValue] > 0) {
                NSArray *items = scanOutput.items;
                BOOL containsScanKey = NO;
                for(NSDictionary *entry in items){
                    if ([[entry objectForKey:@"hashKey"] isEqual:hashValue]){
                        containsScanKey = YES;
                    }
                }

                XCTAssertTrue(containsScanKey, @"Scan did not contain expected key");

            } else {
                XCTFail(@"Reponse object for scan returned with a count < 1.");
            }
        }

        return nil;
    }] waitUntilFinished];
}

- (void)testUpdateItem {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    NSString *key = @"testUpdateItem";
    AWSDynamoDBAttributeValue *updatedValue = [AWSDynamoDBAttributeValue new];
    updatedValue.S = @"updatedValue";

    //Put item
    AWSDynamoDBPutItemInput *putItemInput = [AWSDynamoDBPutItemInput new];
    putItemInput.tableName = table1Name;
    AWSDynamoDBAttributeValue *hashValue = [AWSDynamoDBAttributeValue new];
    hashValue.S = key;
    AWSDynamoDBAttributeValue *otherValue = [AWSDynamoDBAttributeValue new];
    otherValue.S = @"otherValue";
    putItemInput.item = @{
                          @"hashKey" : hashValue,
                          @"otherKey" : otherValue
                          };

    [[[[dynamoDB putItem:putItemInput
        ] continueWithSuccessBlock:^id(BFTask *task) {

        XCTAssertTrue([task.result isKindOfClass:[AWSDynamoDBPutItemOutput class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBPutItemOutput class]));

        AWSDynamoDBUpdateItemInput *updateInput = [AWSDynamoDBUpdateItemInput new];
        updateInput.tableName = table1Name;
        updateInput.key = @{@"hashKey": hashValue};

        AWSDynamoDBAttributeValueUpdate *valueUpdate = [AWSDynamoDBAttributeValueUpdate new];
        valueUpdate.value = updatedValue;
        valueUpdate.action = AWSDynamoDBAttributeActionPut;

        updateInput.attributeUpdates = @{@"otherKey": valueUpdate};
        updateInput.returnValues = AWSDynamoDBReturnValueUpdatedNew;

        return [dynamoDB updateItem:updateInput];
    }] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        } else {
            AWSDynamoDBUpdateItemOutput *updateItemOutput = task.result;
            AWSDynamoDBAttributeValue *outputAttributeValue  = [updateItemOutput.attributes objectForKey:@"otherKey"];

            XCTAssertEqualObjects(outputAttributeValue, updatedValue);
        }

        return nil;
    }] waitUntilFinished];
}

- (void)testUpdateTable {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    if (![[self class] createTable:table2Name]) {
        XCTFail(@"failed to createTable: %@",table2Name);
    }

    NSNumber *newCapacity = @2;
    AWSDynamoDBProvisionedThroughput *provisionedThroughput = [AWSDynamoDBProvisionedThroughput new];
    provisionedThroughput.readCapacityUnits = newCapacity;
    provisionedThroughput.writeCapacityUnits = newCapacity;

    AWSDynamoDBUpdateTableInput *updateTableInput = [AWSDynamoDBUpdateTableInput new];
    updateTableInput.tableName = table2Name;
    updateTableInput.provisionedThroughput = provisionedThroughput;


    [[[[[dynamoDB updateTable:updateTableInput
         ] continueWithSuccessBlock:^id(BFTask *task) {
        XCTAssertNil(task.error);
        return [[self class] waitForTableToBeActive:table2Name];
    }] continueWithSuccessBlock:^id(BFTask *task) {
        XCTAssertNil(task.error);
        AWSDynamoDBDescribeTableInput *describeTableInput = [AWSDynamoDBDescribeTableInput new];
        describeTableInput.tableName = table2Name;

        return [dynamoDB describeTable:describeTableInput];
    }] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        } else {
            AWSDynamoDBDescribeTableOutput *describeTableOutput = task.result;
            AWSDynamoDBTableDescription *tableDescription = describeTableOutput.table;
            XCTAssertEqualObjects(tableDescription.provisionedThroughput.readCapacityUnits, provisionedThroughput.readCapacityUnits);
            XCTAssertEqualObjects(tableDescription.provisionedThroughput.writeCapacityUnits, provisionedThroughput.writeCapacityUnits);
        }
        return nil;
    }] waitUntilFinished];

    XCTAssertTrue([[self class] deleteTable:table2Name], @"Failed to delete table created for testing update table");
}

- (void)testBatchGetItem {
    NSString *key1 = @"putItemForBatch1";
    NSString *key2 = @"putItemForBatch2";
    AWSDynamoDBAttributeValue *attributeValue1 = [AWSDynamoDBAttributeValue new];
    attributeValue1.S = key1;
    AWSDynamoDBAttributeValue *attributeValue2 = [AWSDynamoDBAttributeValue new];
    attributeValue2.S = key2;

    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    //Put item
    AWSDynamoDBPutItemInput *putItemInput = [AWSDynamoDBPutItemInput new];
    putItemInput.tableName = table1Name;
    AWSDynamoDBAttributeValue *hashValue = [AWSDynamoDBAttributeValue new];
    hashValue.S = key1;
    AWSDynamoDBAttributeValue *otherValue = [AWSDynamoDBAttributeValue new];
    otherValue.S = @"otherValue";
    putItemInput.item = @{
                          @"hashKey" : hashValue,
                          @"otherKey" : otherValue
                          };

    [[[[[dynamoDB putItem:putItemInput
         ] continueWithSuccessBlock:^id(BFTask *task) {
        XCTAssertNil(task.error);
        XCTAssertTrue([task.result isKindOfClass:[AWSDynamoDBPutItemOutput class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBPutItemOutput class]));

        //Put item
        AWSDynamoDBPutItemInput *putItemInput2 = [AWSDynamoDBPutItemInput new];
        putItemInput2.tableName = table1Name;
        AWSDynamoDBAttributeValue *hashValue2 = [AWSDynamoDBAttributeValue new];
        hashValue2.S = key2;
        AWSDynamoDBAttributeValue *otherValue2 = [AWSDynamoDBAttributeValue new];
        otherValue2.S = @"otherValue";
        putItemInput2.item = @{
                               @"hashKey" : hashValue2,
                               @"otherKey" : otherValue2
                               };
        return [dynamoDB putItem:putItemInput2];
    }] continueWithBlock:^id(BFTask *task) {
        XCTAssertNil(task.error);
        XCTAssertTrue([task.result isKindOfClass:[AWSDynamoDBPutItemOutput class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBPutItemOutput class]));

        AWSDynamoDBKeysAndAttributes *keysAndAttributes = [AWSDynamoDBKeysAndAttributes new];
        keysAndAttributes.keys = @[@{@"hashKey":attributeValue1},@{@"hashKey":attributeValue2}];
        keysAndAttributes.consistentRead = @YES;

        AWSDynamoDBBatchGetItemInput *batchGetItemInput = [AWSDynamoDBBatchGetItemInput new];
        batchGetItemInput.requestItems = @{table1Name: keysAndAttributes};


        return [dynamoDB batchGetItem:batchGetItemInput];
    }] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        } else {
            XCTAssertTrue([task.result isKindOfClass:[AWSDynamoDBBatchGetItemOutput class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBBatchGetItemOutput class]));
            AWSDynamoDBBatchGetItemOutput *batchGetItemOutput = task.result;
            NSArray *attributeValueArray = [batchGetItemOutput.responses objectForKey:table1Name];

            BOOL key1Bool = NO;
            BOOL key2Bool = NO;
            for(NSDictionary *currentItem in attributeValueArray) {
                AWSDynamoDBAttributeValue *outputAttributeValue = [currentItem objectForKey:@"hashKey"];
                if ([outputAttributeValue isEqual:attributeValue1]) {
                    key1Bool = YES;
                }
                if ([outputAttributeValue isEqual:attributeValue2]) {
                    key2Bool = YES;
                }
            }

            XCTAssertTrue(key1Bool, @"Key1 was not found");
            XCTAssertTrue(key2Bool, @"Key 2 was not found");
        }

        return nil;
    }] waitUntilFinished];
}

- (void)testBatchWriteItem {
    NSString *key1 = @"putItemForBatchWrite1";
    NSString *key2 = @"putItemForBatchWrite2";
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    //Write Request 1
    AWSDynamoDBAttributeValue *hashValue1 = [AWSDynamoDBAttributeValue new];
    hashValue1.S = key1;
    AWSDynamoDBAttributeValue *otherValue1 = [AWSDynamoDBAttributeValue new];
    otherValue1.S = @"otherValue";

    AWSDynamoDBWriteRequest *writeRequest = [AWSDynamoDBWriteRequest new];
    writeRequest.putRequest = [AWSDynamoDBPutRequest new];
    writeRequest.putRequest.item = @{
                                     @"hashKey" : hashValue1,
                                     @"otherKey" : otherValue1
                                     };

    //Write Request 2
    AWSDynamoDBAttributeValue *hashValue2 = [AWSDynamoDBAttributeValue new];
    hashValue2.S = key2;
    AWSDynamoDBAttributeValue *otherValue2 = [AWSDynamoDBAttributeValue new];
    otherValue2.S = @"otherValue";

    AWSDynamoDBWriteRequest *writeRequest2 = [AWSDynamoDBWriteRequest new];
    writeRequest2.putRequest = [AWSDynamoDBPutRequest new];
    writeRequest2.putRequest.item = @{
                                      @"hashKey" : hashValue2,
                                      @"otherKey" : otherValue2
                                      };

    AWSDynamoDBBatchWriteItemInput *batchWriteItemInput = [AWSDynamoDBBatchWriteItemInput new];
    batchWriteItemInput.requestItems = @{table1Name: @[writeRequest,writeRequest2]};

    [[[dynamoDB batchWriteItem:batchWriteItemInput
       ] continueWithBlock:^id(BFTask *task) {
        if (task.error) {
            XCTFail(@"The request failed. error: [%@]", task.error);
        } else {
            XCTAssertTrue([task.result isKindOfClass:[AWSDynamoDBBatchWriteItemOutput class]], @"The response object is not a class of [%@]", NSStringFromClass([AWSDynamoDBBatchWriteItemOutput class]));
            AWSDynamoDBBatchWriteItemOutput *batchWriteItemOutput = task.result;
            NSDictionary *unprocessedItems = batchWriteItemOutput.unprocessedItems;
            XCTAssertEqual([unprocessedItems count], (NSUInteger)0);
        }

        return nil;
    }] waitUntilFinished];
}

- (void)testCreateAndDeleteTable {
    AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

    AWSDynamoDBAttributeDefinition *attributeDefinition = [AWSDynamoDBAttributeDefinition new];
    attributeDefinition.attributeName = @"hashKey";
    attributeDefinition.attributeType = AWSDynamoDBScalarAttributeTypeS;

    AWSDynamoDBKeySchemaElement *keySchemaElement = [AWSDynamoDBKeySchemaElement new];
    keySchemaElement.attributeName = @"hashKey";
    keySchemaElement.keyType = AWSDynamoDBKeyTypeHash;

    AWSDynamoDBProvisionedThroughput *provisionedThroughput = [AWSDynamoDBProvisionedThroughput new];
    provisionedThroughput.readCapacityUnits = @1;
    provisionedThroughput.writeCapacityUnits = @1;

    AWSDynamoDBCreateTableInput *createTableInput = [AWSDynamoDBCreateTableInput new];
    createTableInput.tableName = table2Name;
    createTableInput.attributeDefinitions = @[attributeDefinition];
    createTableInput.keySchema = @[keySchemaElement];
    createTableInput.provisionedThroughput = provisionedThroughput;


    [[[[[[dynamoDB createTable:createTableInput
          ] continueWithSuccessBlock:^id(BFTask *task) {
        XCTAssertNil(task.error);
        AWSDynamoDBCreateTableOutput *createTableOutput = task.result;
        AWSDynamoDBTableDescription *tableDescription = createTableOutput.tableDescription;
        XCTAssertEqualObjects(tableDescription.tableName,table2Name);
        return [[self class] waitForTableToBeActive:table2Name];
    }] continueWithSuccessBlock:^id(BFTask *task) {
        XCTAssertNil(task.error);
        AWSDynamoDBDeleteTableInput *deleteTableInput = [AWSDynamoDBDeleteTableInput new];
        deleteTableInput.tableName = table2Name;
        return [dynamoDB deleteTable: deleteTableInput];
    }] continueWithSuccessBlock:^id(BFTask *task) {
        XCTAssertNil(task.error);
        AWSDynamoDBDeleteTableOutput *deleteTableOutput = task.result;
        AWSDynamoDBTableDescription *tableDescription = deleteTableOutput.tableDescription;
        XCTAssertEqualObjects(tableDescription.tableName, table2Name);
        XCTAssertEqual(tableDescription.tableStatus, AWSDynamoDBTableStatusDeleting);

        return [[self class] waitForTableToBeDeleted:table2Name];
    }] continueWithBlock:^id(BFTask *task) {
        if (task.error
            && (![task.error.domain isEqualToString:AWSDynamoDBErrorDomain]
                || task.error.code != AWSDynamoDBErrorResourceNotFound)) {
                XCTFail(@"The request failed. error: [%@]", task.error);
            }

        return nil;
    }] waitUntilFinished];
}

/*
 - (void)testThrottling {
 XCTAssertTrue([[self class] createTable:table2Name], @"Failed to create a table.");

 AWSDynamoDB *dynamoDB = [AWSDynamoDB defaultDynamoDB];

 NSMutableString *mutableString = [NSMutableString new];
 for(int32_t i = 0; i < 5000; i++) {
 [mutableString appendString:@"0123456789"];
 }

 BFTask *task = [BFTask taskWithResult:nil];

 for (int32_t i = 0; i < 200; i++) {
 task = [task continueWithSuccessBlock:^id(BFTask *task) {
 //Put item
 AWSDynamoDBPutItemInput *putItemInput = [AWSDynamoDBPutItemInput new];
 putItemInput.tableName = table2Name;
 AWSDynamoDBAttributeValue *hashValue = [AWSDynamoDBAttributeValue new];
 hashValue.S = [NSString stringWithFormat:@"testPutItem%d", i];
 AWSDynamoDBAttributeValue *otherValue = [AWSDynamoDBAttributeValue new];
 otherValue.S = [NSString stringWithFormat:@"%@%d", mutableString, i];
 putItemInput.item = @{
 @"hashKey" : hashValue,
 @"otherKey" : otherValue
 };
 return [dynamoDB putItem:putItemInput];
 }];
 }

 [[task continueWithBlock:^id(BFTask *task) {
 if (!task.error
 && !([task.error.domain isEqualToString:AWSDynamoDBErrorDomain]
 || task.error.code == AWSDynamoDBErrorProvisionedThroughputExceeded)) {
 XCTFail(@"Throttling didn't happen. result: [%@] error: [%@]", task.result, task.error);
 }
 
 return nil;
 }] waitUntilFinished];
 
 XCTAssertTrue([[self class] deleteTable:table2Name], @"Failed to delete a table.");
 }
 */

@end

#endif