如何模拟多个级别的DbSet.Include lambdas?

时间:2015-06-15 16:23:11

标签: c# entity-framework unit-testing mocking moq

我正在使用Moq编写使用Entity Framework 6 DbSetDbContext对象的单元测试。我有一个带级联/多级Include的服务方法,我无法弄清楚如何设置它进行测试。服务方法如下所示:

return DataContext.Cars
    .Include(p => p.Model)
    .Include(p => p.Model.Make)
    .Select(c => new 
         {
             Key = c.CarId, 
             Value = string.Format("{0} {1} {2}", c.Model.Make.Name, c.Model.Name, c.Trim)
         }
    ).ToArray();

我知道我必须设置Include才能返回模拟对象,如下所示:

mockCarDbSet.Setup(m => m.Include(It.IsAny<string>())).Returns(mockCarSet.Object);

但是我从级联.Include(p => p.Model.Make)获得了一个空引用异常。如何设置Moq以处理Include的多个级别?

修改
好的,事实证明我不能使用It.IsAny<string>进行使用lambdas而不是字符串的Include调用,所以现在我有两个问题:

  1. 如何设置带有接受lambda的Include的模拟?
  2. 上面的设置会级联到多个级别吗?

2 个答案:

答案 0 :(得分:2)

include()是一种静态方法(扩展方法)。 Moq不支持静态方法mock(read this link)。

要测试您的代码,您需要将mockCarDbSet设置为返回IQueryable<Car>

var carQuery = new List<Car>
{
    //add cars
}
IQueryable<Post> query = carQuery.AsQueryable();
由于query

返回DataContext.Cars

这些步骤将解决静态方法问题。

答案 1 :(得分:0)

非常感谢@Old Fox提醒我Moq不能与静态成员合作,我找到了一种方法来使用Microsoft Fakes。垫片允许您填充静态方法。我使用Moq为每个实体设置Mock<DbSet>个对象:

var carData = new List<Car>{new Car{ Trim = "Whatever" }};  
var mockCarSet = new Mock<DbSet<Car>>();
mockCarSet.As<IQueryable<Car>>().Setup(m => m.Provider).Returns(carData.Provider);
mockCarSet.As<IQueryable<Car>>().Setup(m => m.Expression).Returns(carData.Expression);
mockCarSet.As<IQueryable<Car>>().Setup(m => m.ElementType).Returns(carData.ElementType);
mockCarSet.As<IQueryable<Car>>().Setup(m => m.GetEnumerator()).Returns(carData.GetEnumerator);
var mockMakeSet = new Mock<DbSet<Make>>();
//do the same stuff as with Car for IQueryable Setup
var mockModelSet = new Mock<DbSet<Model>>();
//do the same stuff as with Car for IQueryable Setup
using(ShimsContext.Create())
{
    //hack to return the first, since this is all mock data anyway
    ShimModel.AllInstances.MakeGet = model => mockMakeSet.Object.First();
    ShimCar.AllInstances.ModelGet = car => mockModelSet.Object.First();
    //run the test
}