EF Core自定义初始化逻辑能否指定主键植入测试数据？

针对你在集成测试里需要重复植入数据且指定主键的场景，这里提供几种比手动写SET IDENTITY_INSERT更简洁的方案：

1. 临时修改实体主键生成策略 #

EF Core允许你在运行时临时调整实体的主键生成规则，不用写原生SQL。示例代码如下：

// 植入数据前，临时把Notification的主键设为不自动生成
context.Model.FindEntityType(typeof(Notification))
    .Property(nameof(Notification.Id))
    .SetValueGenerated(ValueGenerated.Never);

// 正常添加数据并保存
await context.Notification.AddRangeAsync(data);
await context.SaveChangesAsync();

// 如果后续测试需要恢复自增规则，再改回去
context.Model.FindEntityType(typeof(Notification))
    .Property(nameof(Notification.Id))
    .SetValueGenerated(ValueGenerated.OnAdd);

这种方式完全用EF Core的API操作，避免了硬编码SQL语句，还能适配不同数据库（比如SQL Server以外的数据库可能不需要IDENTITY_INSERT）。

2. 封装IDENTITY_INSERT操作成扩展方法 #

如果还是依赖SQL Server的IDENTITY_INSERT机制，可以把重复的事务和开关逻辑封装成扩展方法，让代码更干净：

public static async Task WithIdentityInsertAsync<T>(this DbContext context, Func<Task> dataSeedingAction)
    where T : class
{
    var entityType = context.Model.FindEntityType(typeof(T));
    var tableName = entityType.GetTableName() ?? throw new InvalidOperationException("无法获取实体对应的表名");

    await using var transaction = await context.Database.BeginTransactionAsync(System.Data.IsolationLevel.Unspecified);
    try
    {
        await context.Database.ExecuteSqlRawAsync($"SET IDENTITY_INSERT {tableName} ON");
        await dataSeedingAction();
        await context.Database.ExecuteSqlRawAsync($"SET IDENTITY_INSERT {tableName} OFF");
        await transaction.CommitAsync();
    }
    catch
    {
        await transaction.RollbackAsync();
        throw;
    }
}

使用时只需要一行代码调用：

await context.WithIdentityInsertAsync<Notification>(async () =>
{
    await context.Notification.AddRangeAsync(data);
    await context.SaveChangesAsync();
});

这样每次植入数据时不用重复写事务和SQL，复用性更强。

3. 用测试专用的DbContext #

如果你的测试场景中大部分实体都需要指定主键植入，可以创建一个测试专用的DbContext，在配置里直接修改主键生成策略：

public class TestNotificationDbContext : YourAppDbContext
{
    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);
        // 把测试需要的实体主键设为不自动生成
        modelBuilder.Entity<Notification>()
            .Property(n => n.Id)
            .ValueGeneratedNever();
    }
}

集成测试时直接使用这个测试上下文，植入数据时直接指定主键即可，无需额外操作，代码最简洁。

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