public async Task TaskResultIsSetOnCompleteAsync()
{
ItemBatchOperation operation = new ItemBatchOperation(OperationType.Create, 0);
ItemBatchOperationContext batchAsyncOperationContext = new ItemBatchOperationContext(string.Empty);
operation.AttachContext(batchAsyncOperationContext);
BatchOperationResult expected = new BatchOperationResult(HttpStatusCode.OK);
batchAsyncOperationContext.Complete(null, expected);
Assert.AreEqual(expected, await batchAsyncOperationContext.OperationTask);
Assert.AreEqual(TaskStatus.RanToCompletion, batchAsyncOperationContext.OperationTask.Status);
}
public async Task TaskResultIsSetOnCompleteAsync()
{
ItemBatchOperation operation = new ItemBatchOperation(OperationType.Create, 0, Cosmos.PartitionKey.Null);
ItemBatchOperationContext batchAsyncOperationContext = new ItemBatchOperationContext(string.Empty, NoOpTrace.Singleton);
operation.AttachContext(batchAsyncOperationContext);
TransactionalBatchOperationResult expected = new TransactionalBatchOperationResult(HttpStatusCode.OK);
batchAsyncOperationContext.Complete(null, expected);
Assert.AreEqual(expected, await batchAsyncOperationContext.OperationTask);
Assert.AreEqual(TaskStatus.RanToCompletion, batchAsyncOperationContext.OperationTask.Status);
}
public async Task TraceIsJoinedOnCompletionWithRetry()
{
IDocumentClientRetryPolicy retryPolicy = new BulkExecutionRetryPolicy(
Mock.Of <ContainerInternal>(),
OperationType.Read,
new ResourceThrottleRetryPolicy(1));
Trace rootTrace = Trace.GetRootTrace(name: "RootTrace");
ItemBatchOperation operation = new ItemBatchOperation(OperationType.Create, 0, Cosmos.PartitionKey.Null);
// Start with the base trace
ItemBatchOperationContext batchAsyncOperationContext = new ItemBatchOperationContext(Guid.NewGuid().ToString(), rootTrace, retryPolicy);
operation.AttachContext(batchAsyncOperationContext);
// Simulate a retry scenario that should append to the context traces
Trace retryTrace = Trace.GetRootTrace(name: "TransportTrace");
TransactionalBatchOperationResult retryResult = new TransactionalBatchOperationResult(HttpStatusCode.TooManyRequests)
{
Trace = retryTrace
};
ShouldRetryResult shouldRetryResult = await batchAsyncOperationContext.ShouldRetryAsync(retryResult, default);
Assert.IsTrue(shouldRetryResult.ShouldRetry);
// Simulate the completion that should append to the context traces
Trace transportTrace = Trace.GetRootTrace(name: "TransportTrace");
TransactionalBatchOperationResult result = new TransactionalBatchOperationResult(HttpStatusCode.OK)
{
Trace = transportTrace
};
batchAsyncOperationContext.Complete(null, result);
Assert.AreEqual(result, await batchAsyncOperationContext.OperationTask);
Assert.AreEqual(2, result.Trace.Children.Count, "The final trace should have the initial trace, plus the retries, plus the final trace");
Assert.AreEqual(rootTrace, result.Trace, "The first trace child should be the initial root");
Assert.AreEqual(retryTrace, result.Trace.Children[0], "The second trace child should be the one from the retry");
Assert.AreEqual(transportTrace, result.Trace.Children[1], "The third trace child should be the one from the final result");
}
