public void CreateBatchesFromShuffledSequenceWith2PartitionKeys()
{
// Arrange
var partitioner = new TableBatchPartitioner();
const int count = 300;
IEnumerable <ITableEntity> entities = Enumerable
.Range(0, count)
.Select(p => (ITableEntity) new DynamicTableEntity(string.Format("PK{0}", p % 2), p.ToString(CultureInfo.InvariantCulture)));
// Act
List <TableBatchOperation> batches = partitioner.GetBatches(entities, TableOperation.Insert).ToList();
// Assert
Assert.NotNull(batches);
Assert.Equal(count, batches.Sum(p => p.Count));
var fieldInfo = typeof(TableBatchOperation).GetField("batchPartitionKey", BindingFlags.NonPublic | BindingFlags.Instance);
var aggregateResult = batches.Select(p => fieldInfo != null ? new
{
partitonKey = (string)fieldInfo.GetValue(p),
count = p.Count
} : null)
.OrderBy(p => p.partitonKey)
.GroupBy(p => p.partitonKey)
.ToList();
Assert.Equal("PK0", aggregateResult[0].Key);
Assert.Equal("PK1", aggregateResult[1].Key);
Assert.Equal(100, aggregateResult[0].ElementAt(0).count);
Assert.Equal(50, aggregateResult[0].ElementAt(1).count);
Assert.Equal(100, aggregateResult[1].ElementAt(0).count);
Assert.Equal(50, aggregateResult[1].ElementAt(1).count);
}
public void CreateBatchesWithNullEntitiesParameter()
{
// Arrange
var partitioner = new TableBatchPartitioner();
List <TableBatchOperation> batches = null;
// Act
Assert.Throws <ArgumentNullException>(() =>
{
batches = partitioner.GetBatches(null, TableOperation.Insert).ToList();
});
// Assert
Assert.Null(batches);
}
public void CreateBatchesFromSequencWithDifferentPartitionKeys()
{
// Arrange
var partitioner = new TableBatchPartitioner();
const int count = 200;
IEnumerable <ITableEntity> entities = Enumerable
.Range(0, count)
.Select(p => (ITableEntity) new DynamicTableEntity(string.Format("PK{0}", p), string.Empty));
// Act
List <TableBatchOperation> batches = partitioner.GetBatches(entities, TableOperation.Insert).ToList();
// Assert
Assert.NotNull(batches);
Assert.Equal(200, batches.Count);
}