public async Task WhenAllTest()
{
var taskSources = new ValueTaskCompletionSource <int> [3];
for (var i = 0; i < taskSources.Count(); i++)
{
taskSources[i] = ValueTaskCompletionSource <int> .Create();
}
var task = taskSources.Select(p => p.Task).WhenAll(preserveOrder: true);
Assert.IsFalse(task.IsCompleted);
// Complete the tasks in non-sort order and wait some time in between
taskSources[1].SetResult(1);
await Task.Delay(10);
taskSources[2].SetResult(2);
await Task.Delay(10);
taskSources[0].SetResult(0);
await Task.Delay(20); // Wait some time to allow the continuations to execute
Assert.IsTrue(task.IsCompletedSuccessfully);
var result = task.GetAwaiter().GetResult(); // We are allowed only once to get the result.
Assert.IsNotNull(result);
Assert.IsTrue(new[] { 0, 1, 2 }.SequenceEqual(result));
}
public async Task PartiallyCompletedCanceledNonGeneric2WhenAllTest()
{
var taskSources = new ValueTaskCompletionSource[3];
for (var i = 0; i < taskSources.Count(); i++)
{
taskSources[i] = ValueTaskCompletionSource.Create();
}
taskSources[1].SetCanceled();
var task = taskSources.Select(p => p.Task).WhenAll();
Assert.IsFalse(task.IsCompleted);
// Complete the tasks in non-sort order and wait some time in between
taskSources[2].SetResult();
await Task.Delay(10);
taskSources[0].SetResult();
Assert.IsTrue(task.IsCompleted);
Assert.IsTrue(task.IsCanceled);
await Assert.ThrowsExceptionAsync <TaskCanceledException>(async() =>
{
await task;
});
}
public async Task PartiallyCompletedExceptionNonPreserveOrder2WhenAllTest()
{
var taskSources = new ValueTaskCompletionSource <int> [3];
for (var i = 0; i < taskSources.Count(); i++)
{
taskSources[i] = ValueTaskCompletionSource <int> .Create();
}
taskSources[1].SetException(new CustomException());
var task = taskSources.Select(p => p.Task).WhenAll(preserveOrder: false);
Assert.IsFalse(task.IsCompleted);
// Complete the tasks in non-sort order and wait some time in between
taskSources[2].SetResult(2);
await Task.Delay(10);
taskSources[0].SetResult(0);
Assert.IsTrue(task.IsCompleted);
Assert.IsTrue(task.IsFaulted);
await Assert.ThrowsExceptionAsync <CustomException>(async() =>
{
await task;
});
}
public async Task WhenAllNonGenericTest()
{
var taskSources = new ValueTaskCompletionSource[3];
for (var i = 0; i < taskSources.Count(); i++)
{
taskSources[i] = ValueTaskCompletionSource.Create();
}
var task = taskSources.Select(p => p.Task).WhenAll();
Assert.IsFalse(task.IsCompleted);
// Complete the tasks in non-sort order and wait some time in between
taskSources[1].SetResult();
await Task.Delay(10);
taskSources[2].SetResult();
await Task.Delay(10);
taskSources[0].SetResult();
Assert.IsTrue(task.IsCompletedSuccessfully);
task.GetAwaiter().GetResult();
}
public async Task PartiallyCompletedCanceled2WhenAllTest()
{
var taskSources = new ValueTaskCompletionSource <int> [3];
for (var i = 0; i < taskSources.Count(); i++)
{
taskSources[i] = ValueTaskCompletionSource <int> .Create();
}
taskSources[1].SetCanceled();
var task = taskSources.Select(p => p.Task).WhenAll(preserveOrder: true);
Assert.IsFalse(task.IsCompleted);
// Complete the tasks in non-sort order and wait some time in between
taskSources[2].SetResult(2);
await Task.Delay(10);
taskSources[0].SetResult(0);
await Task.Delay(20); // Wait some time to allow the continuations to execute
Assert.IsTrue(task.IsCompleted);
Assert.IsTrue(task.IsCanceled);
await Assert.ThrowsExceptionAsync <TaskCanceledException>(async() =>
{
await task;
});
}