Fix TestWebSocketConnectionFeature that CancellationToken is not work…

…ing (#2906)
aspnet · Sep 5, 2018 · 201c058 · 201c058
1 parent d20d4f1
commit 201c058
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/Microsoft.AspNetCore.Http.Connections.Tests/TestWebSocketConnectionFeature.cs b/test/Microsoft.AspNetCore.Http.Connections.Tests/TestWebSocketConnectionFeature.cs
@@ -118,7 +118,7 @@ public override async Task<WebSocketReceiveResult> ReceiveAsync(ArraySegment<byt
                 {
                     if (_internalBuffer.Buffer == null || _internalBuffer.Buffer.Length == 0)
                     {
-                        await _input.WaitToReadAsync();
+                        await _input.WaitToReadAsync(cancellationToken);
 
                         if (_input.TryRead(out var message))
                         {