public async Task Enqueue_ShouldQueue_PendingTask()
{
// Arrange
const int expected1 = 291;
const int expected2 = 872;
using (var breakpoint = new Breakpoint())
{
var queue = new CoalescingTaskQueue <int, int>((newArg, arg, running) => CoalesceDecision.None);
async Task <int> TestAction(int result)
{
// ReSharper disable once AccessToDisposedClosure
await breakpoint.Hit().WaitForContinue();
return(result);
}
// Act
var task1 = queue.Enqueue(() => TestAction(expected1), expected1);
var task2 = queue.Enqueue(() => TestAction(expected2), expected2);
await breakpoint.WaitForHitAndContinue().TimeoutAfter(TestTimeout);
await breakpoint.WaitForHitAndContinue().TimeoutAfter(TestTimeout);
// Assert
task1.Result.Should().Be(expected1);
task2.Result.Should().Be(expected2);
}
}
public async Task Enqueue_ShouldNotJoin_CancelledRunningTask()
{
// Arrange
const int expected = 6874;
using (var breakpoint = new Breakpoint())
{
var queue = new CoalescingTaskQueue <int, int>((newArg, arg, running) =>
CoalesceDecision.Join);
async Task <int> TestAction(int result, CancellationToken ct)
{
// ReSharper disable once AccessToDisposedClosure
await breakpoint.Hit().WaitForContinue();
ct.ThrowIfCancellationRequested();
return(result);
}
// Act
_ = queue.Enqueue(ct => TestAction(1981, ct), 1981);
var hit1 = await breakpoint.WaitForHit().TimeoutAfter(TestTimeout);
queue.Cancel();
var task2 = queue.Enqueue(ct => TestAction(expected, ct), expected);
hit1.Continue();
await breakpoint.WaitForHitAndContinue().TimeoutAfter(TestTimeout);
// Assert
task2.Result.Should().Be(expected);
}
}
public async Task Cancel_ShouldCancel_PendingAndRunningTasks()
{
// Arrange
using (var breakpoint = new Breakpoint())
{
var queue = new CoalescingTaskQueue <int, int>((newArg, arg, running) =>
CoalesceDecision.None);
async Task <int> TestAction(int result, CancellationToken ct)
{
// ReSharper disable once AccessToDisposedClosure
await breakpoint.Hit().WaitForContinue();
ct.ThrowIfCancellationRequested();
return(result);
}
// Act
var task1 = queue.Enqueue(ct => TestAction(6517, ct), 6517);
var hit1 = await breakpoint.WaitForHit().TimeoutAfter(TestTimeout);
var task2 = queue.Enqueue(ct => TestAction(101, ct), 101);
queue.Cancel();
hit1.Continue();
await Task.WhenAny(task1);
// Assert
task1.IsCanceled.Should().BeTrue();
task2.IsCanceled.Should().BeTrue();
}
}
public async Task Enqueue_ShouldCancel_PendingAndRunningTasks()
{
// Arrange
const int expected3 = 3617;
using (var breakpoint = new Breakpoint())
{
var queue = new CoalescingTaskQueue <int, int>((newArg, arg, running) =>
newArg == expected3 ? CoalesceDecision.Cancel : CoalesceDecision.None);
async Task <int> TestAction(int result, CancellationToken ct)
{
// ReSharper disable once AccessToDisposedClosure
await breakpoint.Hit().WaitForContinue();
ct.ThrowIfCancellationRequested();
return(result);
}
// Act
var task1 = queue.Enqueue(ct => TestAction(202, ct), 202);
var hit1 = await breakpoint.WaitForHit().TimeoutAfter(TestTimeout);
var task2 = queue.Enqueue(ct => TestAction(303, ct), 303);
var task3 = queue.Enqueue(ct => TestAction(expected3, ct), expected3);
hit1.Continue();
await breakpoint.WaitForHitAndContinue().TimeoutAfter(TestTimeout);
// Assert
task1.IsCanceled.Should().BeTrue();
task2.IsCanceled.Should().BeTrue();
task3.Result.Should().Be(expected3);
}
}
public async Task TimeoutAfter_ShouldThrow_TimeoutException_WhenActionTimedOut()
{
// Arrange
var breakpoint = new Breakpoint();
async Task Action(CancellationToken ct)
{
await breakpoint.Hit().WaitForContinue();
ct.ThrowIfCancellationRequested();
}
// Act
var task = TaskExtensions.TimeoutAfter(Action, TimeSpan.Zero, CancellationToken.None);
var hit = await breakpoint.WaitForHit().TimeoutAfter(TestTimeout);
await Task.Delay(TimeSpan.FromMilliseconds(100));
hit.Continue();
Func <Task> action = async() => await task;
// Assert
await action.Should().ThrowAsync <TimeoutException>();
}
public async Task ExecuteAsync_ShouldCancelAction_WhenCancellationToken_IsCancelled()
{
// Arrange
var policy = new ServiceRetryPolicy(0, TimeSpan.Zero);
using (var cancellationSource = new CancellationTokenSource())
using (var breakpoint = new Breakpoint())
{
var cancelled = false;
// Act
var task = policy.ExecuteAsync(
async ct =>
{
// ReSharper disable once AccessToDisposedClosure
await breakpoint.Hit().WaitForContinue();
cancelled = ct.IsCancellationRequested;
ct.ThrowIfCancellationRequested();
return(Result.Ok());
},
cancellationSource.Token);
var hit = await breakpoint.WaitForHit().TimeoutAfter(TestTimeout);
cancellationSource.Cancel();
hit.Continue();
var result = await task.TimeoutAfter(TestTimeout);
// Assert
result.Success.Should().BeFalse();
cancelled.Should().BeTrue();
}
}
public async Task Enqueue_ShouldJoin_PendingTask()
{
// Arrange
const int expected1 = 3905;
const int expected2 = 4487;
using (var breakpoint = new Breakpoint())
{
var queue = new CoalescingTaskQueue <int, int>((newArg, arg, running) =>
newArg > 0 ? CoalesceDecision.None : CoalesceDecision.Join);
async Task <int> TestAction(int result, CancellationToken ct)
{
// ReSharper disable once AccessToDisposedClosure
await breakpoint.Hit().WaitForContinue();
ct.ThrowIfCancellationRequested();
return(result);
}
// Act
var task1 = queue.Enqueue(ct => TestAction(expected1, ct), expected1);
var hit1 = await breakpoint.WaitForHit().TimeoutAfter(TestTimeout);
var task2 = queue.Enqueue(ct => TestAction(expected2, ct), expected2);
var task3 = queue.Enqueue(ct => TestAction(-15, ct), -15);
hit1.Continue();
await breakpoint.WaitForHitAndContinue().TimeoutAfter(TestTimeout);
// Assert
task1.Result.Should().Be(expected1);
task2.Result.Should().Be(expected2);
task3.Result.Should().Be(expected2);
}
}
public async Task TimeoutAfter_ShouldThrow_TaskCanceledException_WhenActionCancelled()
{
// Arrange
var breakpoint = new Breakpoint();
var cancellationSource = new CancellationTokenSource();
async Task Action(CancellationToken ct)
{
await breakpoint.Hit().WaitForContinue();
ct.ThrowIfCancellationRequested();
}
// Act
var task = TaskExtensions.TimeoutAfter(Action, TimeSpan.Zero, cancellationSource.Token);
var hit = await breakpoint.WaitForHit().TimeoutAfter(TestTimeout);
cancellationSource.Cancel();
hit.Continue();
Func <Task> action = async() => await task.TimeoutAfter(TestTimeout);
// Assert
await action.Should().ThrowAsync <OperationCanceledException>();
}
