public void CheckSingleRow(string tableName, PrimaryKey primaryKey,
AttributeColumns attribute,
CapacityUnit expectCapacityUnitConsumed = null,
HashSet <string> columnsToGet = null,
bool isEmpty = false,
ColumnCondition condition = null)
{
var request = new GetRowRequest(tableName, primaryKey, columnsToGet, condition);
var response = OTSClient.GetRow(request);
PrimaryKey primaryKeyToExpect;
AttributeColumns attributeToExpect;
if (isEmpty)
{
primaryKeyToExpect = new PrimaryKey();
attributeToExpect = new AttributeColumns();
}
else if (columnsToGet == null || columnsToGet.Count == 0)
{
primaryKeyToExpect = primaryKey;
attributeToExpect = attribute;
}
else
{
primaryKeyToExpect = new PrimaryKey();
attributeToExpect = new AttributeColumns();
foreach (var columnName in columnsToGet)
{
if (primaryKey.ContainsKey(columnName))
{
primaryKeyToExpect.Add(columnName, primaryKey[columnName]);
}
if (attribute.ContainsKey(columnName))
{
attributeToExpect.Add(columnName, attribute[columnName]);
}
}
}
AssertColumns(primaryKeyToExpect, response.PrimaryKey);
AssertColumns(attributeToExpect, response.Attribute);
if (expectCapacityUnitConsumed != null)
{
AssertCapacityUnit(expectCapacityUnitConsumed, response.ConsumedCapacityUnit);
}
}
