EntityFramework-Plus 并行测试中使用 InMemoryDatabase 的实践指南

背景介绍

在软件开发过程中，单元测试是保证代码质量的重要手段。当使用 EntityFramework 进行数据访问层开发时，我们经常需要编写针对数据库操作的测试用例。EntityFramework-Plus 是一个强大的 EF 扩展库，提供了批量操作、查询缓存等增强功能。

在实际测试场景中，特别是当测试用例需要并行执行时，如何正确配置和使用 InMemoryDatabase 成为一个技术挑战。本文将深入探讨这一问题的解决方案。

核心问题分析

在使用 EntityFramework-Plus 进行并行测试时，开发者会遇到一个关键问题：EntityFrameworkManager.ContextFactory 是一个全局静态属性。这意味着：

1. 所有测试用例共享同一个上下文工厂
2. 并行测试中修改工厂会影响其他正在运行的测试
3. 需要为每个测试类或测试用例提供独立的数据库实例

解决方案设计

经过实践验证，我们可以采用以下架构来解决并行测试中的数据库隔离问题：

1. 全局测试配置管理

创建一个全局测试配置管理器，负责维护所有测试用例的数据库配置信息。这个管理器应该是静态的，确保在整个测试生命周期中保持状态。

public static class TestConfigurationManager
{
    private static readonly ConcurrentDictionary<string, TestConfig> _configurations 
        = new ConcurrentDictionary<string, TestConfig>();
    
    public static void RegisterConfiguration(string dbName, TestConfig config)
    {
        _configurations.TryAdd(dbName, config);
    }
    
    public static TestConfig GetConfiguration(string dbName)
    {
        return _configurations[dbName];
    }
}

2. 上下文工厂实现

配置 EntityFrameworkManager.ContextFactory 使其能够根据当前测试用例动态返回正确的上下文实例：

EntityFrameworkManager.ContextFactory = context =>
{
    if (context is TestDbContext testContext)
    {
        var dbName = testContext.CurrentDatabaseName;
        var config = TestConfigurationManager.GetConfiguration(dbName);
        
        return new TestDbContext(config.Options, config.UserResolver);
    }
    return null;
};

3. 测试基类设计

创建一个测试基类，封装数据库初始化和清理逻辑：

public abstract class DatabaseTestBase : IDisposable
{
    protected readonly string DatabaseName;
    protected readonly TestDbContext Context;
    
    protected DatabaseTestBase()
    {
        DatabaseName = Guid.NewGuid().ToString();
        
        var options = new DbContextOptionsBuilder<TestDbContext>()
            .UseInMemoryDatabase(DatabaseName)
            .Options;
            
        var userResolver = new MockUserResolver();
        
        var config = new TestConfig(options, userResolver);
        TestConfigurationManager.RegisterConfiguration(DatabaseName, config);
        
        Context = new TestDbContext(options, userResolver)
        {
            CurrentDatabaseName = DatabaseName
        };
    }
    
    public void Dispose()
    {
        Context.Dispose();
    }
}

实现细节说明

1. 并发安全：使用 ConcurrentDictionary 确保在多线程环境下配置注册的安全访问
2. 隔离性：每个测试用例通过唯一的数据库名称(GUID)保证数据库完全隔离
3. 灵活性：可以扩展 TestConfig 类来包含测试所需的各种依赖项
4. 可维护性：通过基类封装公共逻辑，减少重复代码

最佳实践建议

1. 数据库命名策略：使用测试类名+测试方法名作为数据库名称，便于调试
2. 资源清理：确保在测试完成后正确释放数据库资源
3. 性能考量：虽然InMemoryDatabase轻量，但大量测试时仍需注意内存使用
4. 依赖注入：考虑与测试框架的DI容器集成，统一管理测试依赖

总结

通过上述方案，我们成功解决了 EntityFramework-Plus 在并行测试环境中的数据库隔离问题。这种设计不仅适用于 InMemoryDatabase，也可以扩展应用到其他测试数据库场景。关键在于：

1. 全局配置管理的线程安全实现
2. 上下文工厂的动态路由能力
3. 测试生命周期的合理管理

这种架构为复杂的EF核心应用提供了可靠的测试基础，确保测试的独立性和可重复性，是高质量软件开发的重要保障。