Added test for checking if the custom headers are copied over when compressing content

Author: azzlack

dist/Microsoft.AspNet.WebApi.MessageHandlers.Compression.1.2.2.nupkg

@@ -6,15 +6,22 @@
 
     using global::Tests.Models;
 
-    using Microsoft.AspNet.WebApi.MessageHandlers.Compression.Attributes;
-
     public class TestController : ApiController
     {
         public async Task<HttpResponseMessage> Get()
         {
             return this.Request.CreateResponse(new TestModel("Get()"));
         }
 
+        [Route("api/test/customheader")]
+        public async Task<HttpResponseMessage> GetCustomHeader()
+        {
+            var response = this.Request.CreateResponse(new TestModel("Get()"));
+            response.Headers.Add("DataServiceVersion", "3.0");
+
+            return response;
+        }
+
         public async Task<HttpResponseMessage> Get(string id)
         {
             return this.Request.CreateResponse(new TestModel("Get(" + id + ")"));

dist/Microsoft.AspNet.WebApi.MessageHandlers.Compression.1.2.2.symbols.nupkg

@@ -41,6 +41,19 @@ public async Task Get_WhenMessageHandlerIsConfigured_ShouldReturnCompressedConte
             Assert.AreEqual("Get()", result.Data);
         }
 
+        public async Task Get_WhenGivenCustomHeader_ShouldReturnCompressedContentWithCustomHeader()
+        {
+            var response = await this.client.GetAsync("http://localhost:55399/api/test/customheader");
+
+            Console.WriteLine("Content: {0}", await response.Content.ReadAsStringAsync());
+
+            Assert.IsTrue(response.Content.Headers.ContentEncoding.Contains("gzip"));
+
+            Console.WriteLine("Content-Length: {0}", response.Content.Headers.ContentLength);
+
+            Assert.IsTrue(response.Headers.Contains("DataServiceVersion"), "The response did not contain the DataServiceVersion header");
+        }
+
         public async Task GetImage_WhenMessageHandlerIsConfigured_ShouldReturnCompressedContent()
         {
 

src/Tests/Controllers/TestController.cs

@@ -51,6 +51,12 @@ public async void Get_WhenMessageHandlerIsConfigured_ShouldReturnCompressedConte
             await this.testFixture.Get_WhenMessageHandlerIsConfigured_ShouldReturnCompressedContent();
         }
 
+        [Test]
+        public async void Get_WhenGivenCustomHeader_ShouldReturnCompressedContentWithCustomHeader()
+        {
+            await this.testFixture.Get_WhenGivenCustomHeader_ShouldReturnCompressedContentWithCustomHeader();
+        }
+
         [Test]
         public async void GetImage_WhenMessageHandlerIsConfigured_ShouldReturnCompressedContent()
         {

src/Tests/Tests/Common/TestFixture.cs

@@ -62,6 +62,12 @@ public async void Get_WhenMessageHandlerIsConfigured_ShouldReturnCompressedConte
             await this.testFixture.Get_WhenMessageHandlerIsConfigured_ShouldReturnCompressedContent();
         }
 
+        [Test]
+        public async void Get_WhenGivenCustomHeader_ShouldReturnCompressedContentWithCustomHeader()
+        {
+            await this.testFixture.Get_WhenGivenCustomHeader_ShouldReturnCompressedContentWithCustomHeader();
+        }
+
         [Test]
         public async void GetImage_WhenMessageHandlerIsConfigured_ShouldReturnCompressedContent()
         {