EF模拟DbSet异步调用无法重复枚举问题

问题原因 #

核心问题在于模拟DbSet的枚举器被复用且仅初始化一次。当你通过EntityFrameworkMockHelper.GetMockDbSet创建模拟集合时，如果Helper内部直接返回同一个IEnumerator实例（而非每次生成新的枚举器），第一次枚举会耗尽这个枚举器，第二次调用时自然返回空结果。

解决方案 #

方案1：修复EntityFrameworkMockHelper的实现 #

检查你的GetMockDbSet方法，确保GetEnumerator()的设置使用工厂lambda表达式，而非直接返回枚举器实例。这样每次调用GetEnumerator()都会生成新的枚举器，避免复用耗尽的问题。

修改前的错误代码（可能存在于你的Helper中）：

mockSet.As<IQueryable<T>>()
       .Setup(m => m.GetEnumerator())
       .Returns(queryable.GetEnumerator()); // 仅返回一次枚举器，耗尽后无法再用

修改后的正确代码：

public static Mock<DbSet<T>> GetMockDbSet<T>(List<T> data) where T : class
{
    var queryable = data.AsQueryable();
    var mockSet = new Mock<DbSet<T>>();
    
    mockSet.As<IQueryable<T>>().Setup(m => m.Provider).Returns(queryable.Provider);
    mockSet.As<IQueryable<T>>().Setup(m => m.Expression).Returns(queryable.Expression);
    mockSet.As<IQueryable<T>>().Setup(m => m.ElementType).Returns(queryable.ElementType);
    // 使用lambda工厂，每次调用GetEnumerator()都生成新实例
    mockSet.As<IQueryable<T>>().Setup(m => m.GetEnumerator()).Returns(() => queryable.GetEnumerator());
    
    // 补充异步方法支持（适配ToListAsync）
    mockSet.As<IDbAsyncEnumerable<T>>()
           .Setup(m => m.GetAsyncEnumerator())
           .Returns(new TestDbAsyncEnumerator<T>(queryable.GetEnumerator()));
    
    mockSet.As<IQueryable<T>>()
           .Setup(m => m.Provider)
           .Returns(new TestDbAsyncQueryProvider<T>(queryable.Provider));
    
    return mockSet;
}

方案2：使用内存DbSet替代Moq模拟 #

如果你不想修改Helper，可以直接实现一个内存版的DbSet<T>，确保每次枚举都从原始集合读取：

public class InMemoryDbSet<T> : DbSet<T>, IQueryable, IEnumerable<T> where T : class
{
    private readonly List<T> _data;
    private readonly IQueryable _queryable;

    public InMemoryDbSet(List<T> data)
    {
        _data = data;
        _queryable = _data.AsQueryable();
    }

    public override T Add(T item)
    {
        _data.Add(item);
        return item;
    }

    // 按需实现其他DbSet方法（如Remove、Find等）

    public IEnumerator<T> GetEnumerator()
    {
        return _data.GetEnumerator();
    }

    IEnumerator IEnumerable.GetEnumerator()
    {
        return GetEnumerator();
    }

    Type IQueryable.ElementType => _queryable.ElementType;
    Expression IQueryable.Expression => _queryable.Expression;
    IQueryProvider IQueryable.Provider => _queryable.Provider;
}

测试中替换为内存集合：

var dummyData = new List<Person>
{
    new Person{Id=1, FirstName="Chris",LastName="Johnson"},
    new Person{Id=2, FirstName="Sally",LastName="Sue"},
};
var personSet = new InMemoryDbSet<Person>(dummyData);
var mockContext = new Mock<IMyContext>();
mockContext.Setup(x => x.People).Returns(personSet);
_context = mockContext.Object;

验证修改 #

修改后重新运行TestMethod1，两次调用GetCPeople都应该能正确返回1条记录，和真实数据库的行为一致。

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