public async Task honor_max_queue_size()
{
var cts = new CancellationTokenSource(10000);
var maxQueueSize = 2;
var mockQueueAwaiter = new Mock <IQueueAwaiter>();
var queue = new SortingQueue(mockQueueAwaiter.Object, maxQueueSize);
var subscriberEvent1 = new SubscriberEvent(null, null, 0, null, null);
var subscriberEvent2 = new SubscriberEvent(null, null, 0, null, null);
var subscriberEvent3 = new SubscriberEvent(null, null, 0, null, null);
var enqueueuSignalSetCount = 0;
var awaitingDequeueSignal = new ManualResetEventSlim(true);
var mockDequeueSignal = new ManualResetEventSlim(false);
mockQueueAwaiter.Setup(x => x.SetEnqueueSignal()).Callback(() => enqueueuSignalSetCount++);
mockQueueAwaiter.Setup(x => x.AwaitDequeueSignalAsync()).Callback(() => awaitingDequeueSignal.Set()).Returns(mockDequeueSignal.WaitHandle.AsTask());
await queue.EnqueueWithWaitAsync(subscriberEvent1, cts.Token);
await queue.EnqueueWithWaitAsync(subscriberEvent2, cts.Token);
var enqueueTask = queue.EnqueueWithWaitAsync(subscriberEvent3, cts.Token);
await Task.WhenAny(new[] { awaitingDequeueSignal.WaitHandle.AsTask(), cts.Token.WaitHandle.AsTask() });
if (cts.Token.IsCancellationRequested)
{
throw new TimeoutException();
}
Assert.Equal(2, enqueueuSignalSetCount);
Assert.Equal(maxQueueSize, queue.QueueCount);
SubscriberEvent se1;
queue.TryDequeue(out se1);
Assert.Equal(maxQueueSize - 1, queue.QueueCount);
mockDequeueSignal.Set();
await Task.WhenAny(new[] { enqueueTask, cts.Token.WaitHandle.AsTask() });
if (cts.Token.IsCancellationRequested)
{
throw new TimeoutException();
}
Assert.Equal(maxQueueSize, queue.QueueCount);
}
public async Task enqueue_and_dequeue_single_item()
{
var cts = new CancellationTokenSource(10000);
var maxQueueSize = 1;
var mockQueueAwaiter = new Mock <IQueueAwaiter>();
var queue = new SortingQueue(mockQueueAwaiter.Object, maxQueueSize);
var subscriberEvent = new SubscriberEvent(null, null, 0, null, null);
await queue.EnqueueWithWaitAsync(subscriberEvent, cts.Token);
SubscriberEvent dequeuedSubscriberEvent;
var tryResult = queue.TryDequeue(out dequeuedSubscriberEvent);
Assert.True(tryResult);
Assert.Equal(subscriberEvent, dequeuedSubscriberEvent);
}
public async Task set_queue_signals()
{
var cts = new CancellationTokenSource(10000);
var maxQueueSize = 1;
var mockQueueAwaiter = new Mock <IQueueAwaiter>();
var queue = new SortingQueue(mockQueueAwaiter.Object, maxQueueSize);
var subscriberEvent = new SubscriberEvent(null, null, 0, null, null);
SubscriberEvent se1;
queue.TryDequeue(out se1);
mockQueueAwaiter.VerifyNoOtherCalls();
await queue.EnqueueWithWaitAsync(subscriberEvent, cts.Token);
mockQueueAwaiter.Verify(x => x.SetEnqueueSignal());
mockQueueAwaiter.VerifyNoOtherCalls();
SubscriberEvent se2;
queue.TryDequeue(out se2);
mockQueueAwaiter.Verify(x => x.SetDequeueSignal());
}