CreateServer with TestServerOptions in WAF (#64803)

Author: manandre

src/Mvc/Mvc.Testing/src/WebApplicationFactory.cs

@@ -16,6 +16,7 @@
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.DependencyModel;
 using Microsoft.Extensions.Hosting;
+using Microsoft.Extensions.Options;
 
 namespace Microsoft.AspNetCore.Mvc.Testing;
 
@@ -588,7 +589,16 @@ private static void EnsureDepsFile()
     /// </summary>
     /// <param name="serviceProvider">The <see cref="IServiceProvider"/> from the bootstrapped application.</param>
     /// <returns></returns>
-    protected virtual TestServer CreateServer(IServiceProvider serviceProvider) => new(serviceProvider);
+    protected virtual TestServer CreateServer(IServiceProvider serviceProvider)
+    {
+        var options = serviceProvider.GetService<IOptions<TestServerOptions>>();
+        if (options is not null)
+        {
+            return new(serviceProvider, options);
+        }
+
+        return new(serviceProvider);
+    }
 
     /// <summary>
     /// Creates the <see cref="IHost"/> with the bootstrapped application in <paramref name="builder"/>.

src/Mvc/test/Mvc.FunctionalTests/TestServerWithCustomConfigurationIntegrationTests.cs

@@ -0,0 +1,56 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System.Net;
+using Microsoft.AspNetCore.Hosting;
+using Microsoft.AspNetCore.Mvc.Testing;
+using Microsoft.AspNetCore.TestHost;
+
+namespace Microsoft.AspNetCore.Mvc.FunctionalTests;
+
+public class CustomWebApplicationFactory : WebApplicationFactory<SimpleWebSite.Startup>
+{
+    public Action<TestServerOptions> ConfigureOptions { get; set; }
+
+    protected override void ConfigureWebHost(IWebHostBuilder builder)
+    {
+        builder.UseTestServer(ConfigureOptions);
+    }
+}
+
+public class TestServerWithCustomConfigurationIntegrationTests : IClassFixture<CustomWebApplicationFactory>
+{
+    public CustomWebApplicationFactory Factory { get; }
+
+    public TestServerWithCustomConfigurationIntegrationTests(CustomWebApplicationFactory factory)
+    {
+        Factory = factory;
+    }
+
+    [Fact]
+    public async Task ServerConfigured()
+    {
+        // Arrange
+        var optionsConfiguredCounter = 0;
+        Factory.ConfigureOptions = options =>
+        {
+            options.AllowSynchronousIO = true;
+            options.PreserveExecutionContext = true;
+            optionsConfiguredCounter++;
+        };
+
+        // Act
+        Assert.Equal(0, optionsConfiguredCounter);
+        Factory.StartServer();
+
+        using var client = Factory.CreateClient();
+
+        using var response = await client.GetAsync("/");
+
+        // Assert
+        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
+        Assert.Equal(1, optionsConfiguredCounter);
+        Assert.True(Factory.Server.AllowSynchronousIO);
+        Assert.True(Factory.Server.PreserveExecutionContext);
+    }
+}