如何正确Mock DataContext中的ExecuteQuery方法？单元测试模拟数据库返回数据解惑

我完全懂这种困扰——模拟EF Core这类数据访问层的IQueryable返回确实容易踩坑，尤其是还要兼顾异常分支的测试。咱们一步步来拆解，用最常用的Moq框架举例子，帮你搞定这个GetBuildings方法的测试。

第一步：明确要模拟的依赖 #

你的GetBuildings方法里用了using(var c...)，这应该是在实例化一个数据库上下文（比如BuildingDbContext），所以我们要模拟的是这个上下文里的Buildings集合（DbSet<Building>），而不是直接模拟IQueryable本身。

第二步：准备模拟数据并配置Mock #

先准备好测试用的模拟Building数据，然后把它转换成IQueryable，再配置Moq让上下文返回这个模拟集合：

// 1. 准备测试数据
var testBuildings = new List<Building>
{
    new Building { Id = 1, PropertyId = 101, Name = "办公楼A" },
    new Building { Id = 2, PropertyId = 101, Name = "仓库B" },
    new Building { Id = 3, PropertyId = 102, Name = "门店C" }
}.AsQueryable();

// 2. 模拟DbSet<Building>
var mockDbSet = new Mock<DbSet<Building>>();
mockDbSet.As<IQueryable<Building>>()
         .Setup(m => m.Provider)
         .Returns(testBuildings.Provider);
mockDbSet.As<IQueryable<Building>>()
         .Setup(m => m.Expression)
         .Returns(testBuildings.Expression);
mockDbSet.As<IQueryable<Building>>()
         .Setup(m => m.ElementType)
         .Returns(testBuildings.ElementType);
mockDbSet.As<IQueryable<Building>>()
         .Setup(m => m.GetEnumerator())
         .Returns(testBuildings.GetEnumerator());

// 3. 模拟数据库上下文
var mockContext = new Mock<BuildingDbContext>();
mockContext.Setup(c => c.Buildings).Returns(mockDbSet.Object);

第三步：注入Mock并测试正常路径 #

把模拟的上下文注入到BuildingService（如果你的Service是通过构造函数注入上下文的话），然后调用GetBuildings验证返回结果：

// 实例化Service，传入模拟的上下文
var service = new BuildingService(mockContext.Object);

// 调用方法
var result = service.GetBuildings(101);

// 验证结果：应该返回PropertyId=101的两个建筑
Assert.Equal(2, result.Count());
Assert.All(result, b => Assert.Equal(101, b.PropertyId));

第四步：测试异常分支 #

如果你的方法里有条件触发SpecialException，比如当propertyId小于0时抛出，那我们可以在测试里验证这个异常逻辑：

// 测试异常分支：传入无效的propertyId
var invalidPropertyId = -1;
Assert.Throws<SpecialException>(() => service.GetBuildings(invalidPropertyId));

关键注意点 #

• 不要直接模拟IQueryable：因为DbSet实现了IQueryable，但还有EF Core的额外逻辑，所以必须模拟DbSet并转换成IQueryable来配置。
• 如果你的Service是在方法内部new上下文，得先重构代码——把上下文作为依赖注入，这样才能模拟，这也是依赖注入的核心好处之一。
• 复杂场景可以用EF Core的InMemory数据库：不用Moq，直接用内存数据库模拟真实查询行为，适合需要测试LINQ过滤、排序等逻辑的场景。

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