如何在使用MSTest运行所有测试前仅执行一次数据初始化

要实现所有测试类运行前仅执行一次数据初始化，核心是把数据库初始化逻辑从每次创建WebHost的流程中剥离，放到程序集级别的初始化方法中，同时确保测试共享同一个WebApplicationFactory和HttpClient实例。以下是具体修改方案：

1. 修复基类的程序集初始化方法 #

MsTest的[AssemblyInitialize]要求方法必须是静态方法且接收TestContext参数，否则不会执行。修改BaseTest，将数据库初始化逻辑移到这里，确保只执行一次：

public class BaseTest
{
    protected static CustomWebApplicationFactory<Program> _webAppFactory;
    protected static HttpClient _httpClient;
    protected static ApplicationContext _dbContext;

    [AssemblyInitialize]
    public static void Initialize(TestContext testContext)
    {
        // 初始化Web应用工厂和HttpClient
        _webAppFactory = new CustomWebApplicationFactory<Program>();
        _httpClient = _webAppFactory.CreateDefaultClient();

        // 仅执行一次数据库初始化
        var serviceProvider = _webAppFactory.Services;
        using (var scope = serviceProvider.CreateScope())
        {
            var scopedServices = scope.ServiceProvider;
            _dbContext = scopedServices.GetRequiredService<ApplicationContext>();
            _dbContext.Database.EnsureDeleted();
            _dbContext.Database.EnsureCreated();
            SeedData(_dbContext); // 执行数据种子填充
        }
    }

    [AssemblyCleanup]
    public static void Cleanup()
    {
        // 测试结束后统一释放资源
        _httpClient.Dispose();
        _webAppFactory.Dispose();
        _dbContext.Dispose();
    }
}

2. 修改CustomWebApplicationFactory #

移除原ConfigureWebHost中的数据库初始化代码，只保留DbContext的注册逻辑，避免每次创建WebHost时重复初始化：

public class CustomWebApplicationFactory<TProgram> : WebApplicationFactory<Program>
{
    protected override void ConfigureWebHost(IWebHostBuilder builder)
    {
        builder.ConfigureTestServices(services => 
        {
            // 移除默认的DbContext配置
            var descriptor = services.SingleOrDefault(d => d.ServiceType == typeof(DbContextOptions<ApplicationContext>));
            if (descriptor != null)
                services.Remove(descriptor);

            // 注册内存数据库（不在这里初始化）
            services.AddDbContext<ApplicationContext>(options => 
                options.UseInMemoryDatabase("testDB"));
        });
    }
}

3. 测试类保持原有结构 #

测试类继承BaseTest即可，无需额外修改，所有测试会共享同一个初始化好的数据库和HttpClient：

[TestClass]
public class VendorTest : BaseTest
{
    [TestMethod]
    public async Task AddVendor()
    {
        var content = JsonSerializer.Serialize(
            new
            {
                name = "VENDORTEST"
            }
        );

        var response = await _httpClient.PostAsync(
            "vendors",
            new StringContent(content, Encoding.UTF8, "application/json")
        );

        Assert.IsTrue(response.IsSuccessStatusCode);
    }

    [TestMethod]
    public async Task GetVendor()
    {
        var response = await _httpClient.GetAsync("vendors");
        string result = await response.Content.ReadAsStringAsync();
        Assert.IsNotNull(response);
    }
}

关键说明 #

• [AssemblyInitialize]标注的方法会在整个测试程序集运行前执行一次，确保数据库只初始化一次。
• 所有测试类共享同一个_webAppFactory和_httpClient实例，避免重复创建资源。
• 如果测试方法之间会修改数据库数据，可能导致测试污染，若需要隔离测试，可以在每个测试方法前添加重置数据的逻辑，但这会牺牲"仅初始化一次"的性能优势，需根据需求权衡。

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