huahaofeng
/
aspnetcore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

OrTimeout Extension

This commit is contained in:
Dylan Dmitri Gray 2019-05-17 10:40:19 -07:00
parent 04ce6ef4c8
commit 9363eff2a8
2 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Middleware/RequestThrottling/sample/Startup.cs
View File

@ -26,7 +26,9 @@ namespace RequestThrottlingSample
{
app.Run(async context =>
{
Console.WriteLine("HEWWWO?");
await context.Response.WriteAsync("Hello world!");
Console.WriteLine("GOODBWWYWE!");
});
}

17
src/Middleware/RequestThrottling/test/SemaphoreWrapperTests.cs
View File

@ -10,6 +10,13 @@ using Microsoft.AspNetCore.Testing;
namespace Microsoft.AspNetCore.RequestThrottling.Tests
{
public static class TaskExtensions
{
public static Task OrTimeout(this Task task, int seconds = 30)
{
return task.TimeoutAfter(TimeSpan.FromSeconds(seconds));
}
}
public class SemaphoreWrapperTests
{
[Fact]
@ -18,7 +25,7 @@ namespace Microsoft.AspNetCore.RequestThrottling.Tests
using var s = new SemaphoreWrapper(1);
Assert.Equal(1, s.Count);
await s.EnterQueue();
await s.EnterQueue().OrTimeout();
Assert.Equal(0, s.Count);
s.LeaveQueue();
@ -44,13 +51,13 @@ namespace Microsoft.AspNetCore.RequestThrottling.Tests
public async Task WaitsIfNoSpaceAvailible()
{
using var s = new SemaphoreWrapper(1);
await s.EnterQueue();
await s.EnterQueue().OrTimeout();
var waitingTask = s.EnterQueue();
Assert.False(waitingTask.IsCompleted);
s.LeaveQueue();
await waitingTask.TimeoutAfter(TimeSpan.FromSeconds(30));
await waitingTask.OrTimeout();
}
[Fact]
@ -59,8 +66,8 @@ namespace Microsoft.AspNetCore.RequestThrottling.Tests
using var s1 = new SemaphoreWrapper(1);
using var s2 = new SemaphoreWrapper(1);
await s1.EnterQueue();
await s2.EnterQueue().TimeoutAfter(TimeSpan.FromSeconds(30));
await s1.EnterQueue().OrTimeout();
await s2.EnterQueue().OrTimeout();
}
}
}