From ee7825ecb803836e247567dfa8249abfb6f06e6b Mon Sep 17 00:00:00 2001
From: Chris R <Tratcher@outlook.com>
Date: Tue, 1 Sep 2015 16:05:37 -0700
Subject: [PATCH] React to HttpContext.Features API changes.

---
 src/Microsoft.AspNet.TestHost/WebSocketClient.cs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/Microsoft.AspNet.TestHost/WebSocketClient.cs b/src/Microsoft.AspNet.TestHost/WebSocketClient.cs
index 0c0f958fa0..1eaecbcb61 100644
--- a/src/Microsoft.AspNet.TestHost/WebSocketClient.cs
+++ b/src/Microsoft.AspNet.TestHost/WebSocketClient.cs
@@ -94,7 +94,7 @@ namespace Microsoft.AspNet.TestHost
                 HttpContext = new DefaultHttpContext(FeatureCollection);
 
                 // Request
-                HttpContext.SetFeature<IHttpRequestFeature>(new RequestFeature());
+                HttpContext.Features.Set<IHttpRequestFeature>(new RequestFeature());
                 var request = HttpContext.Request;
                 request.Protocol = "HTTP/1.1";
                 var scheme = uri.Scheme;
@@ -122,13 +122,13 @@ namespace Microsoft.AspNet.TestHost
                 request.Body = Stream.Null;
 
                 // Response
-                HttpContext.SetFeature<IHttpResponseFeature>(new ResponseFeature());
+                HttpContext.Features.Set<IHttpResponseFeature>(new ResponseFeature());
                 var response = HttpContext.Response;
                 response.Body = Stream.Null;
                 response.StatusCode = 200;
 
                 // WebSocket
-                HttpContext.SetFeature<IHttpWebSocketFeature>(this);
+                HttpContext.Features.Set<IHttpWebSocketFeature>(this);
             }
 
             public void PipelineComplete()