public async Task BasicTests()
{
foreach ((PageList pages, int takeCount, long expectedPageIndex) in TestCases())
{
(List <CosmosElement> elements, long pageIndex) = await TakeQueryPipelineStageTests.CreateAndDrainAsync(
pages : pages,
executionEnvironment : ExecutionEnvironment.Compute,
takeCount : takeCount,
continuationToken : null);
Assert.AreEqual(expectedPageIndex, pageIndex);
Assert.AreEqual(Math.Min(takeCount, pages.Select(x => x.Count).Sum()), elements.Count);
}
}
public async Task SanityTests()
{
long[] values = new long[] { 42, 1337 };
PageList pages = values
.Select(value => new List <CosmosElement>()
{
CosmosElement.Parse($"{{\"item\": {value}}}")
})
.ToList();
foreach (int takeCount in new int[] { 0, 1, values.Length, 2 * values.Length })
{
(List <CosmosElement> elements, _) = await TakeQueryPipelineStageTests.CreateAndDrainAsync(
pages : pages,
executionEnvironment : ExecutionEnvironment.Compute,
takeCount : takeCount,
continuationToken : null);
Assert.AreEqual(Math.Min(takeCount, values.Length), elements.Count);
}
}
