在与Asp的集成测试中设置虚拟IP地址。Net核心测试服务器

Question

我有一个C#Asp。Net Core（1.x）项目，实现web REST API及其相关的集成测试项目，其中在任何测试之前，都有类似以下设置：

// ...

IWebHostBuilder webHostBuilder = GetWebHostBuilderSimilarToRealOne()
    .UseStartup<MyTestStartup>();

TestServer server = new TestServer(webHostBuilder);
server.BaseAddress = new Uri("http://localhost:5000");

HttpClient client = server.CreateClient();

// ...

在测试期间，客户端用于将HTTP请求发送到Web API（测试中的系统）并检索响应。

在实际的测试系统中，有一些组件从每个请求中提取发送方IP地址，如下所示：

HttpContext httpContext = ReceiveHttpContextDuringAuthentication();

// edge cases omitted for brevity
string remoteIpAddress = httpContext?.Connection?.RemoteIpAddress?.ToString()

现在，在集成测试期间，这段代码无法找到IP地址，因为远程地址始终为空。

有没有办法从测试代码中将其设置为某个已知值？我一直在这里搜索，但没有找到类似的东西。助教

Answer 1

使用WebHost。CreateDefaultBuilder可能会搞乱你的应用程序配置。

除非绝对必要，否则没有必要仅仅为了适应测试而更改产品代码。

在不覆盖Startup类方法的情况下，添加自己的中间件的最简单方法是通过IStartupFilter添加中间件，正如Elliott的回答所建议的那样。

而不是使用WebHost。CreateDefaultBuilder，只需使用

base.CreateWebHostBuilder().ConfigureServices...

    public class CustomWAF : WebApplicationFactory<Startup>
    {
        protected override IWebHostBuilder CreateWebHostBuilder()
        {
            return base.CreateWebHostBuilder().ConfigureServices(services =>
            {
                services.AddSingleton<IStartupFilter, CustomStartupFilter>();
            });
        }
    }

Answer 2

根据ASP中的回答。NET内核，是否有任何方法可以从程序中设置中间件。反恐精英？

还可以从ConfigureServices配置中间件，这允许您创建自定义WebApplicationFactory，而无需StartupStub类：

public class CustomWebApplicationFactory : WebApplicationFactory<Startup>
{
    protected override IWebHostBuilder CreateWebHostBuilder()
    {
        return WebHost
            .CreateDefaultBuilder<Startup>(new string[0])
            .ConfigureServices(services =>
            {
                services.AddSingleton<IStartupFilter, CustomStartupFilter>();
            });
    }
}


public class CustomStartupFilter : IStartupFilter
{
    public Action<IApplicationBuilder> Configure(Action<IApplicationBuilder> next)
    {
        return app =>
        {
            app.UseMiddleware<FakeRemoteIpAddressMiddleware>();
            next(app);
        };
    }
}

Answer 3

您可以编写中间件来设置自定义IP地址，因为此属性是可写的：

public class FakeRemoteIpAddressMiddleware
{
    private readonly RequestDelegate next;
    private readonly IPAddress fakeIpAddress = IPAddress.Parse("127.168.1.32");

    public FakeRemoteIpAddressMiddleware(RequestDelegate next)
    {
        this.next = next;
    }

    public async Task Invoke(HttpContext httpContext)
    {
        httpContext.Connection.RemoteIpAddress = fakeIpAddress;

        await this.next(httpContext);
    }
}

然后您可以像这样创建StartupStub类：

public class StartupStub : Startup
{
    public StartupStub(IConfiguration configuration) : base(configuration)
    {
    }

    public override void Configure(IApplicationBuilder app, IHostingEnvironment env)
    {
        app.UseMiddleware<FakeRemoteIpAddressMiddleware>();
        base.Configure(app, env);
    }
}

并使用它创建一个TestServer：

new TestServer(new WebHostBuilder().UseStartup<StartupStub>());