EF Core集成测试：自动回滚事务下时态表测试方案问询

针对你在集成测试中遇到的「自动回滚事务无法生成时态表历史记录」的问题，以下是几种无需手动清理的可行方案：

方案1：手动模拟时态表历史记录（保留自动回滚） #

由于SQL Server时态表仅在事务提交时才会自动写入历史记录，我们可以在测试环境中临时禁用系统自动时态表，改为手动维护历史记录，这样事务回滚时会同时回滚主表和手动写入的历史表数据。

步骤1：修改测试用DbContext配置 #

// 测试专用DbContext，继承自你的业务DbContext
public class TestAppDbContext : AppDbContext
{
    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);
        
        // 移除原始的时态表配置
        var productEntity = modelBuilder.Entity<Product>();
        productEntity.ToTable("Products");
        
        // 手动关联历史表（如果你的业务中有历史表实体）
        productEntity.HasOne<ProductHistory>().WithMany().HasForeignKey(p => p.ProductId);
    }
}

步骤2：测试中手动生成历史记录 #

using (var transaction = new AutoRollbackTransaction())
using (var context = new TestAppDbContext())
{
    // 准备测试数据
    var testProduct = new Product { Id = 1, Price = 50 };
    context.Products.Add(testProduct);
    context.SaveChanges();

    // 执行修改操作
    testProduct.Price = 100;
    context.SaveChanges();

    // 手动生成历史记录并写入
    var historyRecord = new ProductHistory
    {
        ProductId = testProduct.Id,
        Price = 50, // 原始值
        ValidFrom = DateTime.UtcNow.AddSeconds(-1),
        ValidTo = DateTime.UtcNow
    };
    context.ProductHistories.Add(historyRecord);
    context.SaveChanges();

    // 断言验证历史记录
    Assert.Single(context.ProductHistories.Where(h => h.ProductId == 1));
}

事务回滚后，主表和历史表的测试数据会被自动清除，无需额外清理操作。

方案2：使用测试专用临时数据库 #

为每个测试实例创建独立的临时数据库（比如通过Docker快速启动SQL Server容器、使用本地DB文件或SQL Server LocalDB），测试完成后直接销毁整个数据库。这种方式允许你正常提交事务生成时态表历史记录，同时测试结束后数据库被销毁，不会留下任何测试数据。

方案3：提交事务后自动清理（兼顾时态表验证与自动清理） #

修改事务管理类，先提交主事务生成时态表历史记录，再自动清理测试相关的主表和历史表数据：

修改事务管理类 #

internal class TemporalSafeAutoRollback : IDisposable
{
    private TransactionScope _transaction;
    private readonly AppDbContext _dbContext;
    private List<int> _testEntityIds = new();

    public TemporalSafeAutoRollback(AppDbContext dbContext)
    {
        _dbContext = dbContext;
        _transaction = new TransactionScope(TransactionScopeAsyncFlowOption.Enabled);
    }

    // 标记测试数据ID，用于后续清理
    public void TrackTestEntity(int entityId)
    {
        _testEntityIds.Add(entityId);
    }

    public void Dispose()
    {
        if (_transaction == null) return;

        try
        {
            // 提交主事务，生成时态表历史记录
            _transaction.Complete();
            // 启动新事务执行清理
            using var cleanupTx = new TransactionScope(TransactionScopeAsyncFlowOption.Enabled);
            // 清理历史表
            _dbContext.ProductHistories
                .Where(h => _testEntityIds.Contains(h.ProductId))
                .ExecuteDelete();
            // 清理主表
            _dbContext.Products
                .Where(p => _testEntityIds.Contains(p.Id))
                .ExecuteDelete();
            _dbContext.SaveChanges();
            cleanupTx.Complete();
        }
        finally
        {
            _transaction.Dispose();
        }
    }
}

测试代码示例 #

using (var context = new AppDbContext())
using (var tx = new TemporalSafeAutoRollback(context))
{
    // 创建测试数据并标记追踪
    var testProduct = new Product { Name = "Test_Item", Price = 50 };
    context.Products.Add(testProduct);
    context.SaveChanges();
    tx.TrackTestEntity(testProduct.Id);

    // 执行修改操作
    testProduct.Price = 60;
    context.SaveChanges();

    // 此时事务已在Dispose时提交，可直接查询时态表历史记录
    var history = context.ProductHistories
        .Where(h => h.ProductId == testProduct.Id)
        .FirstOrDefault();
    Assert.NotNull(history);
    Assert.Equal(50, history.Price);
}
// Dispose时自动清理测试数据及对应历史记录

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