diff --git a/src/Microsoft.AspNet.TestHost/ResponseFeature.cs b/src/Microsoft.AspNet.TestHost/ResponseFeature.cs index 5b7d230a13..9f2ebfd081 100644 --- a/src/Microsoft.AspNet.TestHost/ResponseFeature.cs +++ b/src/Microsoft.AspNet.TestHost/ResponseFeature.cs @@ -30,6 +30,8 @@ namespace Microsoft.AspNet.TestHost public Stream Body { get; set; } + public bool HeadersSent { get; set; } + public void OnSendingHeaders(Action callback, object state) { var prior = _sendingHeaders; @@ -43,6 +45,7 @@ namespace Microsoft.AspNet.TestHost public void FireOnSendingHeaders() { _sendingHeaders(); + HeadersSent = true; } } } diff --git a/test/Microsoft.AspNet.TestHost.Tests/ResponseFeatureTests.cs b/test/Microsoft.AspNet.TestHost.Tests/ResponseFeatureTests.cs index ebf601cb82..2564ee7112 100644 --- a/test/Microsoft.AspNet.TestHost.Tests/ResponseFeatureTests.cs +++ b/test/Microsoft.AspNet.TestHost.Tests/ResponseFeatureTests.cs @@ -15,6 +15,11 @@ namespace Microsoft.AspNet.TestHost // Assert Assert.Equal(200, responseInformation.StatusCode); + Assert.False(responseInformation.HeadersSent); + + responseInformation.FireOnSendingHeaders(); + + Assert.True(responseInformation.HeadersSent); } } } \ No newline at end of file