public void TryReplenishWithAutoReplenish_ReturnsFalse() { var limiter = new FixedWindowRateLimiter(new FixedWindowRateLimiterOptions(2, QueueProcessingOrder.OldestFirst, 1, TimeSpan.FromSeconds(1), autoReplenishment: true)); Assert.Equal(2, limiter.GetAvailablePermits()); Assert.False(limiter.TryReplenish()); Assert.Equal(2, limiter.GetAvailablePermits()); }
public override async Task CanAcquireResourceAsync_QueuesAndGrabsNewest() { var limiter = new FixedWindowRateLimiter(new FixedWindowRateLimiterOptions(2, QueueProcessingOrder.NewestFirst, 3, TimeSpan.FromMinutes(0), autoReplenishment: false)); var lease = await limiter.WaitAsync(2); Assert.True(lease.IsAcquired); var wait1 = limiter.WaitAsync(2); var wait2 = limiter.WaitAsync(); Assert.False(wait1.IsCompleted); Assert.False(wait2.IsCompleted); lease.Dispose(); Assert.True(limiter.TryReplenish()); // second queued item completes first with NewestFirst lease = await wait2; Assert.True(lease.IsAcquired); Assert.False(wait1.IsCompleted); lease.Dispose(); Assert.Equal(1, limiter.GetAvailablePermits()); Assert.True(limiter.TryReplenish()); lease = await wait1; Assert.True(lease.IsAcquired); }
public override async Task CanAcquireResourceAsync_QueuesAndGrabsOldest() { var limiter = new FixedWindowRateLimiter(new FixedWindowRateLimiterOptions(1, QueueProcessingOrder.OldestFirst, 2, TimeSpan.Zero, autoReplenishment: false)); var lease = await limiter.WaitAsync(); Assert.True(lease.IsAcquired); var wait1 = limiter.WaitAsync(); var wait2 = limiter.WaitAsync(); Assert.False(wait1.IsCompleted); Assert.False(wait2.IsCompleted); lease.Dispose(); Assert.True(limiter.TryReplenish()); lease = await wait1; Assert.True(lease.IsAcquired); Assert.False(wait2.IsCompleted); lease.Dispose(); Assert.Equal(0, limiter.GetAvailablePermits()); Assert.True(limiter.TryReplenish()); lease = await wait2; Assert.True(lease.IsAcquired); }
public async Task AutoReplenish_ReplenishesCounters() { var limiter = new FixedWindowRateLimiter(new FixedWindowRateLimiterOptions(2, QueueProcessingOrder.OldestFirst, 1, TimeSpan.FromMilliseconds(1000), autoReplenishment: true)); Assert.Equal(2, limiter.GetAvailablePermits()); limiter.Acquire(2); var lease = await limiter.WaitAsync(1); Assert.True(lease.IsAcquired); }
public override async Task CanCancelWaitAsyncBeforeQueuing() { var limiter = new FixedWindowRateLimiter(new FixedWindowRateLimiterOptions(1, QueueProcessingOrder.OldestFirst, 1, TimeSpan.Zero, autoReplenishment: false)); var lease = limiter.Acquire(1); Assert.True(lease.IsAcquired); var cts = new CancellationTokenSource(); cts.Cancel(); var ex = await Assert.ThrowsAsync <TaskCanceledException>(() => limiter.WaitAsync(1, cts.Token).AsTask()); Assert.Equal(cts.Token, ex.CancellationToken); lease.Dispose(); Assert.True(limiter.TryReplenish()); Assert.Equal(1, limiter.GetAvailablePermits()); }
public override async Task CanAcquireResourcesWithWaitAsyncWithQueuedItemsIfNewestFirst() { var limiter = new FixedWindowRateLimiter(new FixedWindowRateLimiterOptions(2, QueueProcessingOrder.NewestFirst, 2, TimeSpan.Zero, autoReplenishment: false)); var lease = limiter.Acquire(1); Assert.True(lease.IsAcquired); var wait = limiter.WaitAsync(2); Assert.False(wait.IsCompleted); Assert.Equal(1, limiter.GetAvailablePermits()); lease = await limiter.WaitAsync(1); Assert.True(lease.IsAcquired); Assert.False(wait.IsCompleted); limiter.TryReplenish(); lease = await wait; Assert.True(lease.IsAcquired); }