Added test for checking if the custom headers are copied over when co…

…mpressing content
azzlack · Jun 24, 2015 · e17c064 · e17c064
1 parent 57a5844
commit e17c064
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 2 deletions.
diff --git a/dist/Microsoft.AspNet.WebApi.MessageHandlers.Compression.1.2.2.nupkg b/dist/Microsoft.AspNet.WebApi.MessageHandlers.Compression.1.2.2.nupkg
diff --git a/dist/Microsoft.AspNet.WebApi.MessageHandlers.Compression.1.2.2.symbols.nupkg b/dist/Microsoft.AspNet.WebApi.MessageHandlers.Compression.1.2.2.symbols.nupkg
diff --git a/src/Tests/Controllers/TestController.cs b/src/Tests/Controllers/TestController.cs
@@ -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 + ")"));

diff --git a/src/Tests/Tests/Common/TestFixture.cs b/src/Tests/Tests/Common/TestFixture.cs
@@ -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()
         {
 

diff --git a/src/Tests/Tests/Facade/OwinHostTests.cs b/src/Tests/Tests/Facade/OwinHostTests.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()
         {

diff --git a/src/Tests/Tests/Facade/WebHostTests.cs b/src/Tests/Tests/Facade/WebHostTests.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()
         {