huahaofeng
/
aspnetcore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Fixes #7635] Provide a way to configure clients in WebApplicationFactory

This commit is contained in:
Javier Calvarro Nelson 2018-04-13 10:31:26 -07:00
parent 22510e1377
commit 1e26571ba6
1 changed files with 25 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/Microsoft.AspNetCore.Mvc.Testing/WebApplicationFactory.cs
View File

@ -95,6 +95,7 @@ namespace Microsoft.AspNetCore.Mvc.Testing
CreateServer,
CreateWebHostBuilder,
GetTestAssemblies,
CreateDefaultClient,
builder =>
{
_configuration(builder);
@ -300,20 +301,11 @@ namespace Microsoft.AspNetCore.Mvc.Testing
/// <param name="handlers">A list of <see cref="DelegatingHandler"/> instances to set up on the
/// <see cref="HttpClient"/>.</param>
/// <returns>The <see cref="HttpClient"/>.</returns>
public HttpClient CreateDefaultClient(params DelegatingHandler[] handlers) =>
CreateDefaultClient(new Uri("http://localhost"), handlers);
/// <summary>
/// Creates a new instance of an <see cref="HttpClient"/> that can be used to
/// send <see cref="HttpRequestMessage"/> to the server.
/// </summary>
/// <param name="baseAddress">The base address of the <see cref="HttpClient"/> instance.</param>
/// <param name="handlers">A list of <see cref="DelegatingHandler"/> instances to set up on the
/// <see cref="HttpClient"/>.</param>
/// <returns>The <see cref="HttpClient"/>.</returns>
public HttpClient CreateDefaultClient(Uri baseAddress, params DelegatingHandler[] handlers)
public virtual HttpClient CreateDefaultClient(params DelegatingHandler[] handlers)
{
EnsureServer();
var baseAddress = new Uri("http://localhost");
if (handlers == null || handlers.Length == 0)
{
var client = _server.CreateClient();
@ -342,6 +334,22 @@ namespace Microsoft.AspNetCore.Mvc.Testing
}
}
/// <summary>
/// Creates a new instance of an <see cref="HttpClient"/> that can be used to
/// send <see cref="HttpRequestMessage"/> to the server.
/// </summary>
/// <param name="baseAddress">The base address of the <see cref="HttpClient"/> instance.</param>
/// <param name="handlers">A list of <see cref="DelegatingHandler"/> instances to set up on the
/// <see cref="HttpClient"/>.</param>
/// <returns>The <see cref="HttpClient"/>.</returns>
public HttpClient CreateDefaultClient(Uri baseAddress, params DelegatingHandler[] handlers)
{
var client = CreateDefaultClient(handlers);
client.BaseAddress = baseAddress;
return client;
}
/// <inheritdoc />
public void Dispose()
{
@ -386,18 +394,21 @@ namespace Microsoft.AspNetCore.Mvc.Testing
private readonly Func<IWebHostBuilder, TestServer> _createServer;
private readonly Func<IWebHostBuilder> _createWebHostBuilder;
private readonly Func<IEnumerable<Assembly>> _getTestAssemblies;
private readonly Func<DelegatingHandler[], HttpClient> _createDefaultClient;
public DelegatedWebApplicationFactory(
WebApplicationFactoryClientOptions options,
Func<IWebHostBuilder, TestServer> createServer,
Func<IWebHostBuilder> createWebHostBuilder,
Func<IEnumerable<Assembly>> getTestAssemblies,
Func<DelegatingHandler[], HttpClient> createClient,
Action<IWebHostBuilder> configureWebHost)
{
ClientOptions = new WebApplicationFactoryClientOptions(options);
_createServer = createServer;
_createWebHostBuilder = createWebHostBuilder;
_getTestAssemblies = getTestAssemblies;
_createDefaultClient = createClient;
_configuration = configureWebHost;
}
@ -408,6 +419,8 @@ namespace Microsoft.AspNetCore.Mvc.Testing
protected override IEnumerable<Assembly> GetTestAssemblies() => _getTestAssemblies();
protected override void ConfigureWebHost(IWebHostBuilder builder) => _configuration(builder);
public override HttpClient CreateDefaultClient(params DelegatingHandler[] handlers) => _createDefaultClient(handlers);
}
}
}