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

Timeout all top-level awaits in TestUnixDomainSocket

This commit is contained in:
Stephen Halter 2019-08-08 13:21:09 -07:00
parent c6e1bcb66c
commit 55632872eb
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/Servers/Kestrel/test/FunctionalTests/UnixDomainSocketsTests.cs
View File

@ -84,20 +84,20 @@ namespace Microsoft.AspNetCore.Server.Kestrel.FunctionalTests
using (var host = hostBuilder.Build()) using (var host = hostBuilder.Build())
{ {
await host.StartAsync(); await host.StartAsync().DefaultTimeout();
using (var socket = new Socket(AddressFamily.Unix, SocketType.Stream, ProtocolType.Unspecified)) using (var socket = new Socket(AddressFamily.Unix, SocketType.Stream, ProtocolType.Unspecified))
{ {
await socket.ConnectAsync(new UnixDomainSocketEndPoint(path)); await socket.ConnectAsync(new UnixDomainSocketEndPoint(path)).DefaultTimeout();
var data = Encoding.ASCII.GetBytes("Hello World"); var data = Encoding.ASCII.GetBytes("Hello World");
await socket.SendAsync(data, SocketFlags.None); await socket.SendAsync(data, SocketFlags.None).DefaultTimeout();
var buffer = new byte[data.Length]; var buffer = new byte[data.Length];
var read = 0; var read = 0;
while (read < data.Length) while (read < data.Length)
{ {
read += await socket.ReceiveAsync(buffer.AsMemory(read, buffer.Length - read), SocketFlags.None); read += await socket.ReceiveAsync(buffer.AsMemory(read, buffer.Length - read), SocketFlags.None).DefaultTimeout();
} }
Assert.Equal(data, buffer); Assert.Equal(data, buffer);
@ -106,7 +106,7 @@ namespace Microsoft.AspNetCore.Server.Kestrel.FunctionalTests
// Wait for the server to complete the loop because of the FIN // Wait for the server to complete the loop because of the FIN
await serverConnectionCompletedTcs.Task.DefaultTimeout(); await serverConnectionCompletedTcs.Task.DefaultTimeout();
await host.StopAsync(); await host.StopAsync().DefaultTimeout();
} }
} }
finally finally