在WebApplicationFactory中覆盖启动不映射端点

基础概念

WebApplicationFactory 是 ASP.NET Core 中用于测试 Web API 和 MVC 应用程序的一个抽象基类。它允许你在隔离的环境中启动和配置应用程序，从而进行单元测试和集成测试。通过覆盖 WebApplicationFactory 中的方法，你可以自定义应用程序的启动过程，包括配置中间件、服务和路由。

相关优势

1. 隔离测试环境：在隔离的环境中运行测试，避免影响开发或生产环境。
2. 自定义配置：可以灵活地配置应用程序的中间件、服务和路由。
3. 集成测试：可以进行更全面的集成测试，确保各个组件协同工作。

类型

WebApplicationFactory 主要有以下几种类型：

• WebApplicationFactory<TStartup>：用于测试基于 Startup 类的应用程序。
• WebApplicationFactory<Program>：用于测试基于 Program 类的应用程序（ASP.NET Core 5 及以上版本）。

应用场景

当你需要对 Web API 或 MVC 应用程序进行集成测试时，可以使用 WebApplicationFactory 来启动和配置应用程序。例如，你可以测试某个端点的响应，或者验证中间件的行为。

问题：在 WebApplicationFactory 中覆盖启动不映射端点

原因

在某些情况下，你可能希望在测试环境中禁用某些端点的映射，以避免测试时访问这些端点。

解决方法

你可以通过覆盖 WebApplicationFactory 中的 CreateHostBuilder 方法来自定义应用程序的启动过程，并移除或禁用特定的端点。

以下是一个示例代码，展示如何在 WebApplicationFactory 中覆盖启动并禁用特定端点：

using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Mvc.Testing;
using Microsoft.Extensions.DependencyInjection;

public class CustomWebApplicationFactory<TStartup> : WebApplicationFactory<TStartup>
{
    protected override IWebHostBuilder CreateHostBuilder()
    {
        return base.CreateHostBuilder()
            .ConfigureServices((context, services) =>
            {
                // 移除或禁用特定端点
                var endpointRouteBuilder = new RouteEndpointRouteBuilder(services);
                endpointRouteBuilder.RemoveEndpoint(endpoint => endpoint.Pattern == "/api/sensitive-endpoint");

                services.AddRouting(options =>
                {
                    options.Routes.Add(endpointRouteBuilder.Build());
                });
            });
    }
}

在这个示例中，我们通过 RemoveEndpoint 方法移除了 /api/sensitive-endpoint 端点。你可以根据需要修改这个方法来禁用其他端点。

参考链接

• ASP.NET Core Integration Tests
• WebApplicationFactory Class

通过这种方式，你可以在测试环境中灵活地控制端点的映射，确保测试的准确性和安全性。