ABP.io项目单元测试：SQLite替换为LocalDb的实现示例

实现思路 #

1. 替换EF Core模块依赖：移除AbpEntityFrameworkCoreSqliteModule，添加AbpEntityFrameworkCoreSqlServerModule以支持SQL Server/LocalDb
2. 配置LocalDb连接：使用LocalDb的连接字符串创建物理数据库（而非内存模式），适配空间数据需求
3. 调整数据库初始化逻辑：每次测试前生成全新数据库，确保测试环境干净
4. 适配资源管理：修改连接对象的创建与释放逻辑，测试完成后自动清理数据库

修改后的完整代码 #

using Microsoft.Data.SqlClient;
using Microsoft.EntityFrameworkCore;
using Volo.Abp.EntityFrameworkCore;
using Volo.Abp.EntityFrameworkCore.SqlServer;
using Volo.Abp.Modularity;
using System;

[DependsOn(
    typeof(ABPProjectEntityFrameworkCoreModule),
    typeof(ABPProjectTestBaseModule),
    typeof(AbpEntityFrameworkCoreSqlServerModule) // 替换为SQL Server模块
    )]
public class ABPProjectEntityFrameworkCoreTestModule : AbpModule
{
    private SqlConnection _sqlConnection;
    private string _dbName;

    public override void ConfigureServices(ServiceConfigurationContext context)
    {
        ConfigureLocalDb(context.Services);
    }

    private void ConfigureLocalDb(IServiceCollection services)
    {
        // 生成唯一数据库名，避免并行测试冲突
        _dbName = $"ABPProject_Test_{Guid.NewGuid():N}";
        _sqlConnection = CreateDatabaseAndGetConnection();

        services.Configure<AbpDbContextOptions>(options =>
        {
            options.Configure(context =>
            {
                context.DbContextOptions.UseSqlServer(_sqlConnection, sqlOptions =>
                {
                    // 启用空间数据支持
                    sqlOptions.UseNetTopologySuite();
                });
            });
        });
    }

    public override void OnApplicationShutdown(ApplicationShutdownContext context)
    {
        // 关闭连接后删除测试数据库
        _sqlConnection.Dispose();
        DeleteTestDatabase();
    }

    private SqlConnection CreateDatabaseAndGetConnection()
    {
        // LocalDb连接字符串，自动创建数据库
        var connectionString = $"Server=(localdb)\\mssqllocaldb;Database={_dbName};Trusted_Connection=True;";
        var connection = new SqlConnection(connectionString);
        connection.Open();

        // 创建数据库并生成表结构
        var options = new DbContextOptionsBuilder<ABPProjectDbContext>()
            .UseSqlServer(connection, sqlOptions => sqlOptions.UseNetTopologySuite())
            .Options;

        using (var context = new ABPProjectDbContext(options))
        {
            context.Database.EnsureCreated();
        }

        return connection;
    }

    private void DeleteTestDatabase()
    {
        // 连接到master数据库删除测试库
        var masterConnectionString = "Server=(localdb)\\mssqllocaldb;Database=master;Trusted_Connection=True;";
        using var masterConn = new SqlConnection(masterConnectionString);
        masterConn.Open();

        using var command = masterConn.CreateCommand();
        // 终止所有连接后删除数据库
        command.CommandText = $@"
            ALTER DATABASE [{_dbName}] SET SINGLE_USER WITH ROLLBACK IMMEDIATE;
            DROP DATABASE [{_dbName}];
        ";
        command.ExecuteNonQuery();
    }
}

关键说明 #

• 空间数据支持：通过UseNetTopologySuite()启用EF Core的空间数据功能，匹配SQL Server的空间类型
• 唯一数据库名：用Guid生成唯一数据库名，避免多线程测试或重复运行时的数据库冲突
• 自动清理：在应用关闭时自动删除测试数据库，避免LocalDb中残留大量无用测试库
• LocalDb实例：使用(localdb)\\mssqllocaldb作为默认实例，这是LocalDb的标准实例名

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