From 5621a2c2c7326fc285b9a148e37c3298d25cff68 Mon Sep 17 00:00:00 2001
From: Hao Kung <haok@microsoft.com>
Date: Thu, 25 Jun 2015 17:04:16 -0700
Subject: [PATCH] React to HttpChanges

---
 src/Microsoft.AspNet.TestHost/ResponseFeature.cs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/Microsoft.AspNet.TestHost/ResponseFeature.cs b/src/Microsoft.AspNet.TestHost/ResponseFeature.cs
index 7ae2111959..25d9b3260d 100644
--- a/src/Microsoft.AspNet.TestHost/ResponseFeature.cs
+++ b/src/Microsoft.AspNet.TestHost/ResponseFeature.cs
@@ -4,6 +4,7 @@
 using System;
 using System.Collections.Generic;
 using System.IO;
+using System.Threading.Tasks;
 using Microsoft.AspNet.Http.Features;
 
 namespace Microsoft.AspNet.TestHost
@@ -33,7 +34,7 @@ namespace Microsoft.AspNet.TestHost
 
         public bool HasStarted { get; set; }
 
-        public void OnResponseStarting(Action<object> callback, object state)
+        public void OnStarting(Func<object, Task> callback, object state)
         {
             var prior = _responseStarting;
             _responseStarting = () =>
@@ -43,7 +44,7 @@ namespace Microsoft.AspNet.TestHost
             };
         }
 
-        public void OnResponseCompleted(Action<object> callback, object state)
+        public void OnCompleted(Func<object, Task> callback, object state)
         {
             var prior = _responseCompleted;
             _responseCompleted = () =>