Added tests

Author: davidni

test/Microsoft.AspNetCore.Server.WebListener.FunctionalTests/ResponseBodyTests.cs

@@ -36,6 +36,40 @@ public async Task ResponseBody_WriteNoHeaders_SetsChunked()
             }
         }
 
+        [ConditionalTheory]
+        [InlineData(true)]
+        [InlineData(false)]
+        public async Task ResponseBody_WriteNoHeaders_SetsChunked_LargeBody(bool enableKernelBuffering)
+        {
+            const int WriteSize = 1024 * 1024;
+            const int NumWrites = 32;
+
+            string address;
+            using (Utilities.CreateHttpServer(
+                baseAddress: out address,
+                app: httpContext =>
+                {
+                    for (int i = 0; i < NumWrites - 1; i++)
+                    {
+                        httpContext.Response.Body.Write(new byte[WriteSize], 0, WriteSize);
+                    }
+                    return httpContext.Response.Body.WriteAsync(new byte[WriteSize], 0, WriteSize);
+                },
+                enableKernelBuffering: enableKernelBuffering))
+            {
+                var response = await SendRequestAsync(address);
+                Assert.Equal(200, (int)response.StatusCode);
+                Assert.Equal(new Version(1, 1), response.Version);
+                IEnumerable<string> ignored;
+                Assert.False(response.Content.Headers.TryGetValues("content-length", out ignored), "Content-Length");
+                Assert.True(response.Headers.TransferEncodingChunked.HasValue, "Chunked");
+
+                var bytes = await response.Content.ReadAsByteArrayAsync();
+                Assert.Equal(WriteSize * NumWrites, bytes.Length);
+                Assert.True(bytes.All(b => b == 0));
+            }
+        }
+
         [ConditionalFact]
         public async Task ResponseBody_WriteNoHeadersAndFlush_DefaultsToChunked()
         {

test/Microsoft.AspNetCore.Server.WebListener.FunctionalTests/Utilities.cs

@@ -21,25 +21,46 @@ internal static class Utilities
         private static int NextPort = BasePort;
         private static object PortLock = new object();
 
-        internal static IServer CreateHttpServer(out string baseAddress, RequestDelegate app)
+        internal static IServer CreateHttpServer(out string baseAddress, RequestDelegate app, bool enableKernelBuffering = false)
         {
             string root;
-            return CreateDynamicHttpServer(string.Empty, AuthenticationSchemes.None, true, out root, out baseAddress, app);
+            return CreateDynamicHttpServer(
+                basePath: string.Empty,
+                authType: AuthenticationSchemes.None,
+                allowAnonymous: true,
+                enableKernelBuffering: enableKernelBuffering,
+                root: out root,
+                baseAddress: out baseAddress,
+                app: app);
         }
 
         internal static IServer CreateHttpServerReturnRoot(string path, out string root, RequestDelegate app)
         {
             string baseAddress;
-            return CreateDynamicHttpServer(path, AuthenticationSchemes.None, true, out root, out baseAddress, app);
+            return CreateDynamicHttpServer(
+                basePath: path,
+                authType: AuthenticationSchemes.None,
+                allowAnonymous: true,
+                enableKernelBuffering: false,
+                root: out root,
+                baseAddress: out baseAddress,
+                app: app);
         }
 
         internal static IServer CreateHttpAuthServer(AuthenticationSchemes authType, bool allowAnonymous, out string baseAddress, RequestDelegate app)
         {
             string root;
-            return CreateDynamicHttpServer(string.Empty, authType, allowAnonymous, out root, out baseAddress, app);
+            return CreateDynamicHttpServer(
+                basePath: string.Empty,
+                authType: authType,
+                allowAnonymous: allowAnonymous,
+                enableKernelBuffering: false,
+                root: out root,
+                baseAddress: out baseAddress,
+                app: app);
         }
 
-        internal static IServer CreateDynamicHttpServer(string basePath, AuthenticationSchemes authType, bool allowAnonymous, out string root, out string baseAddress, RequestDelegate app)
+        internal static IServer CreateDynamicHttpServer(string basePath, AuthenticationSchemes authType, bool allowAnonymous, bool enableKernelBuffering, out string root, out string baseAddress, RequestDelegate app)
         {
             lock (PortLock)
             {
@@ -55,6 +76,7 @@ internal static IServer CreateDynamicHttpServer(string basePath, AuthenticationS
                     server.Features.Get<IServerAddressesFeature>().Addresses.Add(baseAddress);
                     server.Listener.Settings.Authentication.Schemes = authType;
                     server.Listener.Settings.Authentication.AllowAnonymous = allowAnonymous;
+                    server.Listener.Settings.EnableKernelResponseBuffering = enableKernelBuffering;
                     try
                     {
                         server.Start(new DummyApplication(app));