public async Task Add_NewState_Expires()
{
// Arrange
var tcs = new TaskCompletionSource <IMemoryOwner <byte> >();
var operation = new Mock <IOperation>();
operation
.Setup(m => m.HandleOperationCompleted(It.IsAny <IMemoryOwner <byte> >()))
.Callback((IMemoryOwner <byte> response) => tcs.TrySetResult(response));
var state = MakeState(5, operation.Object);
using var set = new InFlightOperationSet();
// Act
set.Add(state, 10);
// Wait up to 15 seconds for the task to complete
await Task.WhenAny(tcs.Task, Task.Delay(15000));
// Assert
Assert.True(tcs.Task.IsCompleted);
using var data = tcs.Task.Result;
var status = (ResponseStatus)ByteConverter.ToInt16(data.Memory.Span.Slice(HeaderOffsets.Status));
Assert.Equal(ResponseStatus.OperationTimeout, status);
}
public async Task WaitForAllOperationsAsync_WithStates_WaitsOnAll()
{
// Arrange
var state1 = MakeState(5);
var state2 = MakeState(6);
using var set = new InFlightOperationSet();
set.Add(state1, 75000);
set.Add(state2, 75000);
// Act
var task = set.WaitForAllOperationsAsync(TimeSpan.FromSeconds(10));
// Assert
await Task.Delay(10).ConfigureAwait(false);
Assert.False(task.IsCompleted);
state1.Complete(null);
await Task.Delay(10).ConfigureAwait(false);
Assert.False(task.IsCompleted);
state2.Complete(null);
await Task.Delay(10).ConfigureAwait(false);
Assert.True(task.IsCompleted);
}
public async Task WaitForAllOperationsAsync_WithTimeout_TimeOut()
{
// Arrange
var state1 = MakeState(5);
var state2 = MakeState(6);
using var set = new InFlightOperationSet();
set.Add(state1, 75000);
set.Add(state2, 75000);
// Act
await Assert.ThrowsAsync <TimeoutException>(() => set.WaitForAllOperationsAsync(TimeSpan.FromMilliseconds(100)).AsTask());
}
public void WaitForAllOperationsAsync_NoStates_NoWait()
{
// Arrange
using var set = new InFlightOperationSet();
// Act
var task = set.WaitForAllOperationsAsync(TimeSpan.FromSeconds(10));
// Assert
Assert.True(task.IsCompleted);
}
public void Add_NewState_AddsToCount()
{
// Arrange
var state = MakeState(5);
using var set = new InFlightOperationSet();
// Act
set.Add(state, 75000);
// Assert
Assert.Equal(1, set.Count);
}
public void Add_NewState_AddsToCount()
{
// Arrange
var state = MakeState(5);
using var set = new InFlightOperationSet(TimeSpan.FromSeconds(75));
// Act
set.Add(state);
// Assert
Assert.Equal(1, set.Count);
}
public void TryRemove_InSet_DoesNotGet()
{
// Arrange
var state = MakeState(5);
using var set = new InFlightOperationSet();
set.Add(state, 75000);
// Act
var result = set.TryRemove(6, out var outState);
// Assert
Assert.False(result);
Assert.Null(outState);
Assert.Equal(1, set.Count);
}
public void TryRemove_InSet_GetsAndRemoves()
{
// Arrange
var state = MakeState(5);
using var set = new InFlightOperationSet();
set.Add(state, 75000);
// Act
var result = set.TryRemove(5, out var outState);
// Assert
Assert.True(result);
Assert.Equal(state, outState);
Assert.Equal(0, set.Count);
}
public async Task Add_NewState_Expires()
{
// Arrange
var operation = new FakeOperation();
var state = MakeState(5, operation);
using var set = new InFlightOperationSet(TimeSpan.FromMilliseconds(10), TimeSpan.FromSeconds(1));
// Act
set.Add(state);
// Wait up to 15 seconds for the task to complete
await Task.WhenAny(state.CompletionTask, Task.Delay(15000));
// Assert
Assert.True(state.CompletionTask.IsCompleted);
}
public async Task Add_NewState_Expires()
{
// Arrange
var tcs = new TaskCompletionSource <ResponseStatus>();
var state = MakeState(5, (_, status) => { tcs.TrySetResult(status); });
using var set = new InFlightOperationSet();
// Act
set.Add(state, 10);
// Wait up to 15 seconds for the task to complete
await Task.WhenAny(tcs.Task, Task.Delay(15000));
// Assert
Assert.True(tcs.Task.IsCompleted);
Assert.Equal(ResponseStatus.OperationTimeout, tcs.Task.Result);
}
