public async Task GetChangeFeedTokensAsync_DrainFromJustOnePartition()
{
int pkRangesCount = (await this.LargerContainer.ClientContext.DocumentClient.ReadPartitionKeyRangeFeedAsync(this.LargerContainer.LinkUri)).Count;
ContainerInternal itemsCore = this.LargerContainer;
IEnumerable <string> tokens = await itemsCore.GetChangeFeedTokensAsync();
Assert.IsTrue(pkRangesCount > 1, "Should have created a multi partition container.");
Assert.AreEqual(pkRangesCount, tokens.Count());
int totalDocuments = 200;
await this.CreateRandomItems(this.LargerContainer, totalDocuments, randomPartitionKey : true);
string token = tokens.First();
int count = 0;
FeedIterator iteratorForToken =
itemsCore.GetStandByFeedIterator(
ChangeFeedStartFrom.ContinuationToken(token));
while (iteratorForToken.HasMoreResults)
{
using (ResponseMessage responseMessage = await iteratorForToken.ReadNextAsync(this.cancellationToken))
{
if (!responseMessage.IsSuccessStatusCode)
{
break;
}
Collection <ToDoActivity> response = TestCommon.SerializerCore.FromStream <CosmosFeedResponseUtil <ToDoActivity> >(responseMessage.Content).Data;
count += response.Count;
}
}
Assert.IsTrue(count > 0 && count < totalDocuments);
}
public async Task GetChangeFeedTokensAsync_MatchesPkRanges()
{
int pkRangesCount = (await this.LargerContainer.ClientContext.DocumentClient.ReadPartitionKeyRangeFeedAsync(this.LargerContainer.LinkUri)).Count;
ContainerInternal itemsCore = this.LargerContainer;
IEnumerable <string> tokens = await itemsCore.GetChangeFeedTokensAsync();
Assert.AreEqual(pkRangesCount, tokens.Count());
}
public async Task GetChangeFeedTokensAsync_AllowsParallelProcessing()
{
int pkRangesCount = (await this.LargerContainer.ClientContext.DocumentClient.ReadPartitionKeyRangeFeedAsync(this.LargerContainer.LinkUri)).Count;
ContainerInternal itemsCore = this.LargerContainer;
IEnumerable <string> tokens = await itemsCore.GetChangeFeedTokensAsync();
Assert.IsTrue(pkRangesCount > 1, "Should have created a multi partition container.");
Assert.AreEqual(pkRangesCount, tokens.Count());
int totalDocuments = 200;
await this.CreateRandomItems(this.LargerContainer, totalDocuments, randomPartitionKey : true);
List <Task <int> > tasks = tokens.Select(token => Task.Run(async() =>
{
int count = 0;
FeedIterator iteratorForToken =
itemsCore.GetStandByFeedIterator(
requestOptions: new ChangeFeedRequestOptions()
{
From = ChangeFeedRequestOptions.StartFrom.CreateFromContinuation(token),
});
while (true)
{
using (ResponseMessage responseMessage =
await iteratorForToken.ReadNextAsync(this.cancellationToken))
{
if (!responseMessage.IsSuccessStatusCode)
{
break;
}
Collection <ToDoActivity> response = TestCommon.SerializerCore.FromStream <CosmosFeedResponseUtil <ToDoActivity> >(responseMessage.Content).Data;
count += response.Count;
}
}
return(count);
})).ToList();
await Task.WhenAll(tasks);
int documentsRead = 0;
foreach (Task <int> task in tasks)
{
documentsRead += task.Result;
}
Assert.AreEqual(totalDocuments, documentsRead);
}