HeadMessageHandler: Support empty content eg. 302

src/WebApiContrib/MessageHandlers/HeadMessageHandler.cs

@@ -22,7 +22,8 @@ protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage reques
                     .ContinueWith<HttpResponseMessage>(task => {
                         var response = task.Result;
                         response.RequestMessage.Method = HttpMethod.Head;
-                        response.Content = new HeadContent(response.Content);
+                        if (response.Content != null)
+                            response.Content = new HeadContent(response.Content);
                         return task.Result;
                     });
 

test/WebApiContribTests/MessageHandlers/HeadMessageHandlerTests.cs

@@ -25,7 +25,7 @@ public void Should_return_headers_and_no_body_when_method_is_head()
 
             var response = client.SendAsync(requestMessage).Result;
 
-            Assert.AreEqual("Boo",response.ReasonPhrase);
+            Assert.AreEqual("Boo", response.ReasonPhrase);
             Assert.AreEqual(0, response.Content.Headers.ContentLength);
         }
 
@@ -48,5 +48,27 @@ public void Should_do_nothing_when_method_is_get()
             Assert.AreEqual("Boo", response.ReasonPhrase);
             Assert.AreEqual("Hello world", response.Content.ReadAsStringAsync().Result);
         }
+
+        [Test]
+        public void Should_support_empry_bodies()
+        {
+            var messageHandler302 = new HeadMessageHandler();
+            var reasonPhrase = "302 from Controller.Redirect(url)";
+            messageHandler302.InnerHandler = new PrecannedMessageHandler(new HttpResponseMessage(HttpStatusCode.OK)
+            {
+                ReasonPhrase = reasonPhrase,
+                Content = null
+            });
+
+            var requestMessage = new HttpRequestMessage(HttpMethod.Head, "http://foo/bar");
+
+            var client = new HttpClient(messageHandler302);
+
+            var response = client.SendAsync(requestMessage).Result;
+
+            Assert.AreEqual(reasonPhrase, response.ReasonPhrase);
+            Assert.AreEqual(null, response.Content);
+
+        }
     }
 }