private static async Task <List <CosmosElement> > CreateAndDrainAsync(
IReadOnlyList <IReadOnlyList <CosmosElement> > pages,
ExecutionEnvironment executionEnvironment,
int offsetCount,
CosmosElement continuationToken)
{
IQueryPipelineStage source = new MockQueryPipelineStage(pages);
TryCatch <IQueryPipelineStage> tryCreateSkipQueryPipelineStage = SkipQueryPipelineStage.MonadicCreate(
executionEnvironment: executionEnvironment,
offsetCount: offsetCount,
continuationToken: continuationToken,
cancellationToken: default,
private static async Task <(List <CosmosElement>, long)> CreateAndDrainAsync(
IReadOnlyList <IReadOnlyList <CosmosElement> > pages,
ExecutionEnvironment executionEnvironment,
int takeCount,
CosmosElement continuationToken)
{
MockQueryPipelineStage source = new MockQueryPipelineStage(pages);
TryCatch <IQueryPipelineStage> tryCreateSkipQueryPipelineStage = TakeQueryPipelineStage.MonadicCreateLimitStage(
executionEnvironment: executionEnvironment,
limitCount: takeCount,
requestContinuationToken: continuationToken,
cancellationToken: default,
private static async Task <List <CosmosElement> > CreateAndDrainAsync(
IReadOnlyList <IReadOnlyList <CosmosElement> > pages,
ExecutionEnvironment executionEnvironment,
CosmosElement continuationToken,
DistinctQueryType distinctQueryType)
{
IQueryPipelineStage source = new MockQueryPipelineStage(pages);
TryCatch <IQueryPipelineStage> tryCreateDistinctQueryPipelineStage = DistinctQueryPipelineStage.MonadicCreate(
executionEnvironment: executionEnvironment,
requestContinuation: continuationToken,
distinctQueryType: distinctQueryType,
cancellationToken: default,
public static MockQueryPipelineStage Create(
IReadOnlyList <IReadOnlyList <CosmosElement> > pages,
CosmosElement continuationToken)
{
MockQueryPipelineStage stage = new MockQueryPipelineStage(pages);
if (continuationToken != null)
{
CosmosNumber index = continuationToken as CosmosNumber;
stage.PageIndex = Number64.ToLong(index.Value);
}
return(stage);
}
private static async Task <List <CosmosElement> > CreateAndDrainAsync(
IReadOnlyList <IReadOnlyList <CosmosElement> > pages,
ExecutionEnvironment executionEnvironment,
CosmosElement continuationToken,
IReadOnlyDictionary <string, AggregateOperator?> groupByAliasToAggregateType,
IReadOnlyList <string> orderedAliases,
bool hasSelectValue)
{
IQueryPipelineStage source = new MockQueryPipelineStage(pages);
TryCatch <IQueryPipelineStage> tryCreateGroupByStage = GroupByQueryPipelineStage.MonadicCreate(
executionEnvironment: executionEnvironment,
continuationToken: continuationToken,
cancellationToken: default,
