From 201c058b9379b584317b912a4928a83ad7b569bf Mon Sep 17 00:00:00 2001
From: "Liangying.Wei" <lianwei@microsoft.com>
Date: Thu, 6 Sep 2018 00:31:05 +0800
Subject: [PATCH] Fix TestWebSocketConnectionFeature that CancellationToken is
 not working (#2906)

---
 .../TestWebSocketConnectionFeature.cs                           | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/Microsoft.AspNetCore.Http.Connections.Tests/TestWebSocketConnectionFeature.cs b/test/Microsoft.AspNetCore.Http.Connections.Tests/TestWebSocketConnectionFeature.cs
index c0711f4ace..f67dd94003 100644
--- 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))
                         {