.NET自动化测试中高效回滚应用/API产生的数据库变更（大型复杂数据库场景）

作为同样处理过大型.NET自动化测试项目的开发者，我完全理解你的痛点——2500+测试用例、复杂庞大的SQL Server数据库，既要保证测试隔离性，又不能让测试慢到无法接受。你遇到的核心问题是：测试进程的TransactionScope无法覆盖API独立进程的数据库连接，导致API的变更无法回滚。下面是几个针对大型数据库场景的高效解决方案，按可实现性和性能排序：

1. SQL Server数据库快照（推荐：平衡性能与实现成本） #

SQL Server的数据库快照是**写时复制（Copy-On-Write）**的只读镜像，创建快照的速度极快（仅需几秒，不管数据库多大），因为它只记录后续修改的数据页。对测试场景来说，流程非常简洁：

• 测试套件启动前：为目标数据库创建一个基准快照（比如TEST_DB_SNAPSHOT）
• 每个测试用例执行后：将数据库快速恢复到基准快照状态

具体实现（C#测试代码） #

// 测试套件初始化方法（全局仅执行一次）
[AssemblyInitialize]
public static void CreateTestSnapshot(TestContext context)
{
    var connectionString = ConfigurationManager.ConnectionStrings["TestDb"].ConnectionString;
    using (var conn = new SqlConnection(connectionString))
    {
        conn.Open();
        // 创建快照的SQL语句，注意文件名需对应你的数据库物理文件路径
        var createSnapshotSql = @"
        CREATE DATABASE TEST_DB_SNAPSHOT
        ON (NAME = TEST_DB, FILENAME = 'C:\SQLData\TEST_DB_SNAPSHOT.ss')
        AS SNAPSHOT OF TEST_DB;";
        using (var cmd = new SqlCommand(createSnapshotSql, conn))
        {
            cmd.ExecuteNonQuery();
        }
    }
}

// 每个测试用例执行后自动清理
[TestCleanup]
public void RevertToSnapshot()
{
    var masterConnString = ConfigurationManager.ConnectionStrings["TestDb"].ConnectionString
        .Replace("Initial Catalog=TEST_DB", "Initial Catalog=master");
    
    using (var conn = new SqlConnection(masterConnString))
    {
        conn.Open();
        // 先断开所有连接到测试库的活跃会话
        var killConnectionsSql = @"
        DECLARE @kill varchar(8000) = '';
        SELECT @kill = @kill + 'KILL ' + CONVERT(varchar(5), session_id) + ';'
        FROM sys.dm_exec_sessions
        WHERE database_id = DB_ID('TEST_DB') AND session_id <> @@SPID;
        EXEC(@kill);";
        using (var cmd = new SqlCommand(killConnectionsSql, conn))
        {
            cmd.ExecuteNonQuery();
        }
        
        // 恢复到基准快照
        var restoreSql = @"
        RESTORE DATABASE TEST_DB FROM DATABASE_SNAPSHOT = 'TEST_DB_SNAPSHOT';";
        using (var cmd = new SqlCommand(restoreSql, conn))
        {
            cmd.ExecuteNonQuery();
        }
    }
}

优缺点 #

• ✅ 性能极高：创建/恢复快照均为秒级操作，完全不受数据库大小影响
• ✅ 无业务代码侵入：完全在测试侧和数据库层面实现
• ❌ 测试环境独占：恢复快照会断开所有连接，因此测试环境不能同时运行其他业务
• ❌ 磁盘占用：快照会随测试修改的数据量递增，但测试后恢复会自动释放冗余空间

2. 跨进程事务传播（适合本地启动的API实例） #

如果你的API是本地启动的测试实例（而非远程服务），可以通过传递事务标识，让API的数据库连接主动加入测试的事务中，避免分布式事务的开销：

实现思路 #

1. 测试代码创建TransactionScope，获取当前事务的传播令牌
2. 发送API请求时，将令牌通过自定义请求头（比如X-Test-Transaction-Token）传递给API
3. API的数据库连接初始化逻辑中，检查请求头并将连接enlist到测试的事务中
4. 测试结束后，TransactionScope自动回滚所有操作（包括API的数据库变更）

关键代码示例 #

测试侧代码

[TestMethod]
public async Task TestApiCreateUser()
{
    using (var scope = new TransactionScope(TransactionScopeAsyncFlowOption.Enabled))
    {
        // 获取事务传播令牌
        var transactionToken = TransactionInterop.GetTransmitterPropagationToken(Transaction.Current);
        var tokenBase64 = Convert.ToBase64String(transactionToken);
        
        // 构造API请求，传递事务令牌
        var client = new HttpClient();
        client.DefaultRequestHeaders.Add("X-Test-Transaction-Token", tokenBase64);
        var response = await client.PostAsJsonAsync("https://localhost:5001/api/users", new UserDto { Name = "TestUser" });
        response.EnsureSuccessStatusCode();

        // 验证数据库状态（同事务内查询，结果可见）
        using (var conn = new SqlConnection(ConnectionString))
        {
            conn.Open();
            var userCount = (int)new SqlCommand("SELECT COUNT(*) FROM Users WHERE Name = 'TestUser'", conn).ExecuteScalar();
            Assert.AreEqual(1, userCount);
        }

        // 不调用Complete()，事务自动回滚
    }
}

API侧代码（数据库连接拦截）

public class TestAwareDbContext : DbContext
{
    private readonly IHttpContextAccessor _httpContextAccessor;

    public TestAwareDbContext(IHttpContextAccessor httpContextAccessor)
    {
        _httpContextAccessor = httpContextAccessor;
    }

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        var connectionString = Configuration.GetConnectionString("DefaultConnection");
        var connection = new SqlConnection(connectionString);
        
        // 检查测试事务令牌
        if (_httpContextAccessor.HttpContext?.Request.Headers.TryGetValue("X-Test-Transaction-Token", out var tokenBase64) == true)
        {
            var transactionToken = Convert.FromBase64String(tokenBase64);
            var transaction = TransactionInterop.GetTransactionFromTransmitterPropagationToken(transactionToken);
            connection.EnlistTransaction(transaction);
        }
        
        optionsBuilder.UseSqlServer(connection);
    }
}

优缺点 #

• ✅ 完全事务隔离：所有操作在同一个事务中，回滚彻底
• ✅ 无数据库层面开销：比快照更轻量
• ❌ 需修改API代码：添加测试专属逻辑，需做好环境判断（仅在测试环境启用）
• ❌ 仅支持本地API：远程服务的令牌传递会非常复杂，且可能需要MSDTC支持

3. 存储层面的快照（基础设施级方案） #

如果你的测试环境使用虚拟化或云存储（比如VMware、Azure Managed Disks），可以通过存储层面的快照快速恢复数据库状态：

• 测试前：对数据库所在的磁盘创建存储快照
• 每个测试后：将磁盘恢复到基准快照状态

这个方法性能是所有方案中最高的，但需要运维团队支持，且数据库需处于离线状态（或使用支持在线快照的存储系统）。

内容来源于stack exchange