在实体框架6.1中模拟DbContext

Question

我发现许多示例（显然）显示了使用EF 6模拟DbContext的清晰工作示例，但是，似乎没有一个对我有用，我也不完全知道为什么。

这是设置模拟的我的单元测试代码；

var mockData = new List<User> { new User { Email = "my@email.com", Id = 1 } }.AsQueryable();

var mockSet = new Mock<DbSet<User>>();
    mockSet.As<IQueryable<User>>().Setup(m => m.Provider).Returns(mockData.Provider);
    mockSet.As<IQueryable<User>>().Setup(m => m.Expression).Returns(mockData.Expression);
    mockSet.As<IQueryable<User>>().Setup(m => m.ElementType).Returns(mockData.ElementType);
    mockSet.As<IQueryable<User>>().Setup(m => m.GetEnumerator()).Returns(mockData.GetEnumerator());

    var mockContext = new Mock<MyDbContext>();
    mockContext.Setup(c => c.Users).Returns(mockSet.Object);

然后调用我正在测试的服务；

var service = new UsersService(mockContext.Object);

var user = service.GetById(1);

由于基础DbSet始终为null，因此将引发NullReferenceException。 该代码执行以下操作；

在BaseClass中；

public IEnumerable<T> GetAll()
{
    return _dbSet.AsEnumerable();
}

在子类中；

  public User GetById(int id)
        {
            return GetAll().FirstOrDefault(x => x.Id == id);
        }

请注意，尽管关于SO的其他问题似乎与之相关，但它们并未涵盖EF 6。

作为参考，这是MSDN文章，该文章对相同的代码进行了修改以使其能够编译。

https://msdn.microsoft.com/zh-CN/data/dn314429.aspx

编辑：

降低了UserService的复杂性（它使用泛型/接口），现在的代码很简单；

public User GetById(int id)
        {
            return _dbContext.Set<User>().FirstOrDefault(x => x.Id == id);
        }

如果我将其进一步更改为；

   var dbSet = _dbContext.Set<User>();
        return dbSet.FirstOrDefault(x => x.Id == id);

我可以清楚地看到dbSet为null。

编辑2

根据wablab的建议，似乎模拟.Set解决了该问题。

还应感谢Vladyslav Kushnir的DbSet通用方法。

适用于可能需要它的任何人的工作代码；

 private static Mock<DbSet<T>> GetDbSetMock<T>(IEnumerable<T> items = null) where T : class
        {
            if (items == null)
            {
                items = new T[0];
            }

            var dbSetMock = new Mock<DbSet<T>>();
            var q = dbSetMock.As<IQueryable<T>>();

            q.Setup(x => x.GetEnumerator()).Returns(items.GetEnumerator);

            return dbSetMock;
        }



var mockContext = new Mock<Model1>();

var users = new List<User> { new User { Email = "my@email.com", Id = 1 } };

mockContext.Setup(x => x.Set<User>()).Returns(GetDbSetMock(users).Object);

var service = new UsersService(mockContext.Object);

var user = service.GetById(1);

Answer 1

我认为您需要在Set<User>()方法上创建一个设置来返回您的模拟。

Answer 2

    private Mock<DbSet<T>> GetDbSetMock<T>(IEnumerable<T> items = null) where T : class
    {
        if (items == null)
        {
            items = new T[0];
        }

        var dbSetMock = new Mock<DbSet<T>>();
        var q = dbSetMock.As<IQueryable<T>>();

        q.Setup(x => x.GetEnumerator()).Returns(items.GetEnumerator);

        return dbSetMock;
    }

这是我用来模拟DbContext的DbSet的工作得很好的通用方法。 该方法的实际调用是：

var contextMock = new Mock<MyContext>();
contextMock.Setup(x => x.MyDbEntities).Returns(GetDbSetMock<MyDbEntity>().Object);