如何借助TestServer与数据种子类共享内存数据库以运行集成测试

我之前做AspNet Core集成测试的时候也遇到过一模一样的问题——通过API接口去造测试数据不仅写起来麻烦，跑起来也慢，尤其是涉及到关联数据的时候。给你分享几个我实战中用的方案，都能直接绕开API，快速给内存数据库塞测试数据：

这是最直接的思路，既然已经用了内存数据库，完全可以在测试项目里专门写一个类，负责初始化不同测试场景需要的数据。

实现步骤： #

• 创建一个TestDataSeeder类，注入你的业务DbContext（TestServer的服务容器里已经注册了上下文，直接拿就行）
• 给每个测试场景写对应的初始化方法，比如SeedBasicUsers()、SeedOrderWithRelatedItems()
• 在测试用例的前置方法（比如xUnit的IClassFixture或者[SetUp]）里调用这些方法，提前准备好数据

代码示例： #

public class TestDataSeeder
{
    private readonly AppDbContext _dbContext;

    public TestDataSeeder(AppDbContext dbContext)
    {
        _dbContext = dbContext;
    }

    // 初始化基础用户数据
    public async Task SeedTestUsers()
    {
        if (!_dbContext.Users.Any())
        {
            _dbContext.Users.AddRange(new List<User>
            {
                new User { Id = 1, Name = "测试用户1", Email = "test1@example.com" },
                new User { Id = 2, Name = "测试用户2", Email = "test2@example.com" }
            });
            await _dbContext.SaveChangesAsync();
        }
    }

    // 初始化带关联商品的订单数据
    public async Task SeedOrderWithItems(int userId)
    {
        var testOrder = new Order
        {
            UserId = userId,
            OrderDate = DateTime.UtcNow,
            Items = new List<OrderItem>
            {
                new OrderItem { ProductName = "笔记本电脑", Quantity = 1, Price = 999.99m },
                new OrderItem { ProductName = "鼠标", Quantity = 2, Price = 29.99m }
            }
        };
        _dbContext.Orders.Add(testOrder);
        await _dbContext.SaveChangesAsync();
    }
}

在测试中使用： #

public class OrderApiTests : IClassFixture<WebApplicationFactory<Program>>
{
    private readonly WebApplicationFactory<Program> _factory;
    private readonly HttpClient _client;
    private readonly TestDataSeeder _dataSeeder;

    public OrderApiTests(WebApplicationFactory<Program> factory)
    {
        _factory = factory;
        // 从服务容器获取DbContext实例
        var dbContext = factory.Services.GetRequiredService<AppDbContext>();
        _dataSeeder = new TestDataSeeder(dbContext);
        _client = factory.CreateClient();
    }

    [Fact]
    public async Task GetUserOrders_ReturnsExpectedData()
    {
        // 先初始化测试数据
        await _dataSeeder.SeedTestUsers();
        await _dataSeeder.SeedOrderWithItems(1);

        // 发起API请求
        var response = await _client.GetAsync("/api/users/1/orders");
        response.EnsureSuccessStatusCode();

        // 断言返回结果符合预期
        var orders = await response.Content.ReadFromJsonAsync<List<OrderDto>>();
        Assert.Single(orders);
        Assert.Equal(2, orders[0].Items.Count);
    }
}

如果想让数据初始化逻辑更通用，不用每个测试类都重复写，可以扩展WebApplicationFactory，把初始化逻辑封装进去：

public class TestAppFactory : WebApplicationFactory<Program>
{
    protected override void ConfigureWebHost(IWebHostBuilder builder)
    {
        builder.ConfigureServices(services =>
        {
            // 这里你已经替换了内存数据库，继续注册测试数据初始化服务
            services.AddScoped<TestDataSeeder>();
        });
    }

    // 提供统一的方法来执行指定的数据初始化操作
    public async Task SeedTestData(Func<TestDataSeeder, Task> seedAction)
    {
        using var scope = Services.CreateScope();
        var seeder = scope.ServiceProvider.GetRequiredService<TestDataSeeder>();
        await seedAction(seeder);
    }
}

测试中使用自定义Factory： #

public class OrderApiTests : IClassFixture<TestAppFactory>
{
    private readonly TestAppFactory _factory;
    private readonly HttpClient _client;

    public OrderApiTests(TestAppFactory factory)
    {
        _factory = factory;
        _client = factory.CreateClient();
    }

    [Fact]
    public async Task GetUserOrders_ReturnsExpectedData()
    {
        // 通过Factory统一初始化数据
        await _factory.SeedTestData(async seeder =>
        {
            await seeder.SeedTestUsers();
            await seeder.SeedOrderWithItems(1);
        });

        // 发起请求并断言...
    }
}

如果需要大量随机模拟数据，手动写太麻烦，可以用Bogus这类库自动生成，配合上面的Seeder类：

先安装NuGet包：Install-Package Bogus

修改Seeder类，添加批量生成方法： #

public async Task SeedRandomUsers(int count)
{
    var userFaker = new Faker<User>()
        .RuleFor(u => u.Name, f => f.Person.FullName)
        .RuleFor(u => u.Email, f => f.Person.Email)
        .RuleFor(u => u.CreatedAt, f => f.Date.Past(1));

    var randomUsers = userFaker.Generate(count);
    _dbContext.Users.AddRange(randomUsers);
    await _dbContext.SaveChangesAsync();
}

这样就能快速生成批量测试数据，适合需要大数据量的测试场景。

• 测试隔离：每次测试后记得清理数据库，避免测试之间的数据污染。比如在xUnit的IAsyncLifetime接口的DisposeAsync方法里调用_dbContext.Database.EnsureDeletedAsync()
• 环境校验：确保测试环境下确实使用的是内存数据库，绝对不能误操作生产库
• 事务回滚：如果测试涉及到复杂的数据库操作，也可以用事务包裹测试逻辑，测试完成后回滚，不过内存数据库直接删除数据会更简单

内容的提问来源于stack exchange，提问作者Norcino