如何在Entity Framework的DbSet Mock中配置Add方法修改对象属性值

核心逻辑是劫持DbSet<T>的Add/AddAsync方法执行流程，在方法返回前给实体主键属性赋值，完全复现EF调用Add方法后自动给主键分配取值的行为。以下以最常用的Moq框架为例给出实现：

• 基础场景实现（单实体、int自增主键）
  先定义测试用实体：

  public class Blog
  {
      public int Id { get; set; }
      public string Title { get; set; }
  }
  

  配置Mock DbSet的Add逻辑：

  // 维护自增主键的递增值，初始值和数据库自增起始值保持一致即可
  int nextAutoId = 1;
  var mockBlogDbSet = new Mock<DbSet<Blog>>();
  
  // 配置同步Add方法
  mockBlogDbSet.Setup(set => set.Add(It.IsAny<Blog>()))
      .Callback<Blog>(addedEntity =>
      {
          // 拦截传入的新增实体，给Id赋值
          addedEntity.Id = nextAutoId++;
      });
  
  // 项目中用到异步方法的话补充AddAsync配置
  mockBlogDbSet.Setup(set => set.AddAsync(It.IsAny<Blog>(), It.IsAny<CancellationToken>()))
      .Returns<Blog, CancellationToken>((addedEntity, token) =>
      {
          addedEntity.Id = nextAutoId++;
          return ValueTask.CompletedTask;
      });
  

• 非int主键场景适配
  如果主键是Guid类型，不需要维护递增变量，直接在回调里生成新Guid即可：

  mockBlogDbSet.Setup(set => set.Add(It.IsAny<Blog>()))
      .Callback<Blog>(addedEntity =>
      {
          addedEntity.Id = Guid.NewGuid();
      });
  

• 多实体通用配置封装
  项目中实体多的话可以抽成扩展方法，自动识别约定的主键属性（Id或类名+Id），不用每个DbSet重复写配置：

  public static Mock<DbSet<T>> ConfigureAutoGeneratedId<T>(this Mock<DbSet<T>> mockSet, int startId = 1) where T : class
  {
      int currentId = startId;
      // 自动查找实体的约定主键属性
      var keyProperty = typeof(T).GetProperty($"{typeof(T).Name}Id") 
                        ?? typeof(T).GetProperty("Id");
      if (keyProperty == null) return mockSet;
  
      // 配置同步Add
      mockSet.Setup(s => s.Add(It.IsAny<T>()))
          .Callback<T>(entity => keyProperty.SetValue(entity, currentId++));
  
      // 配置异步Add
      mockSet.Setup(s => s.AddAsync(It.IsAny<T>(), It.IsAny<CancellationToken>()))
          .Returns<T, CancellationToken>((entity, token) =>
          {
              keyProperty.SetValue(entity, currentId++);
              return ValueTask.CompletedTask;
          });
  
      return mockSet;
  }
  
  // 调用示例
  var mockUserSet = new Mock<DbSet<User>>().ConfigureAutoGeneratedId();
  var mockPostSet = new Mock<DbSet<Post>>().ConfigureAutoGeneratedId(startId: 100);
  

注意：如果测试逻辑需要严格对齐EF的真实行为（Add时生成临时主键，调用SaveChanges后才生成持久化正式主键），可以把Id赋值逻辑移到DbContext的SaveChanges/SaveChangesAsyncMock回调中，遍历追踪状态为Added的实体统一赋值即可，绝大多数单元测试场景不需要做这层区分，直接在Add回调里赋值就能满足测试要求。

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