如何使用Entity Framework和Moq进行单元测试？

Question

我是Moq的新手，想要将它用作数据的后备存储 - 但不触及实时数据库。

我的设置如下：

• 一个的UnitOfWork包含所有存储库，并用于整个应用程序的数据访问。
• 存储库表示直接挂钩到由DbContext提供的DbSet。
• DbContext包含所有的DbSets。

这里是我的测试，到目前为止：

 // ARRANGE 
     var user = new User() 
     { 
      FirstName = "Some", 
      LastName = "Guy", 
      EmailAddress = "some.guy@mockymoqmoq.com", 
     }; 

     var mockSet = new MockDbSet<User>(); 
     var mockContext = new Mock<WebAPIDbContext>(); 

     mockContext.Setup(c => c.Set<User>()).Returns(mockSet.Object); 

     // ACT 
     using (var uow = UnitOfWork.Create(mockContext.Object)) 
     { 
      uow.UserRepository.Add(user); 
      uow.SaveChanges(); 
     } 

     // ASSERT 
     mockSet.Verify(u => u.Add(It.IsAny<User>()), Times.Once()); 

我的测试似乎是成功的，因为它可以验证用户添加到模拟DbSet - 但我需要做的实际上得到的是那些数据会被回传并对其进行进一步的断言（这只是一个临时测试）。

请指教，测试框架正在做我的工作。另外，如果易于使用，我可以选择移动到其他测试框架。

谢谢。

更新：这是我的工作代码。

单元测试

 // ARRANGE 
     var user = new User() 
     { 
      FirstName = "Some", 
      LastName = "Guy", 
      EmailAddress = "some.guy@mockymoqmoq.com", 
     }; 

     var mockSet = new MockDbSet<User>(); 
     var mockContext = new Mock<WebAPIDbContext>(); 

     mockContext.Setup(c => c.Set<User>()).Returns(mockSet.Object); 

     // ACT 
     using (var uow = UnitOfWork.Create(mockContext.Object)) 
     { 
      uow.UserRepository.Add(user); 
      uow.SaveChanges(); 
     } 

     // ASSERT 
     mockSet.Verify(u => u.Add(It.IsAny<User>()), Times.Once()); 

     // TODO: Further assertations can now take place by accessing mockSet.BackingStore. 
    } 

MockDbSet

class MockDbSet<TEntity> : Mock<DbSet<TEntity>> where TEntity : class 
{ 
    public ICollection<TEntity> BackingStore { get; set; } 

    public MockDbSet() 
    { 
     var queryable = (this.BackingStore ?? (this.BackingStore = new List<TEntity>())).AsQueryable(); 

     this.As<IQueryable<TEntity>>().Setup(e => e.Provider).Returns(queryable.Provider); 
     this.As<IQueryable<TEntity>>().Setup(e => e.Expression).Returns(queryable.Expression); 
     this.As<IQueryable<TEntity>>().Setup(e => e.ElementType).Returns(queryable.ElementType); 
     this.As<IQueryable<TEntity>>().Setup(e => e.GetEnumerator()).Returns(() => queryable.GetEnumerator()); 

     // Mock the insertion of entities 
     this.Setup(e => e.Add(It.IsAny<TEntity>())).Returns((TEntity entity) => 
     { 
      this.BackingStore.Add(entity); 

      return entity; 
     }); 

     // TODO: Other DbSet members can be mocked, such as Remove(). 
    } 
} 

Answer 1

你只需要创建一个集合作为后备存储，并嘲笑枚举dB设置与衬里收集

public class MockDbSet<TEntity> : Mock<DbSet<TEntity>> where TEntity : class { 
    public MockDbSet(List<TEntity> dataSource = null) { 
     var data = (dataSource ?? new List<TEntity>()); 
     var queryable = data.AsQueryable(); 

     this.As<IQueryable<TEntity>>().Setup(e => e.Provider).Returns(queryable.Provider); 
     this.As<IQueryable<TEntity>>().Setup(e => e.Expression).Returns(queryable.Expression); 
     this.As<IQueryable<TEntity>>().Setup(e => e.ElementType).Returns(queryable.ElementType); 
     this.As<IQueryable<TEntity>>().Setup(e => e.GetEnumerator()).Returns(() => queryable.GetEnumerator()); 
     //Mocking the insertion of entities 
     this.Setup(_ => _.Add(It.IsAny<TEntity>()).Returns((TEntity arg) => { 
      data.Add(arg); 
      return arg; 
     }); 

     //...the same can be done for other members like Remove 
    } 
} 

所以，现在你可以使用列表来保存数据

// ARRANGE 
var dataSource = new List<User>(); //<-- this will hold data 
var user = new User() 
{ 
    FirstName = "Some", 
    LastName = "Guy", 
    EmailAddress = "some.guy@mockymoqmoq.com", 
}; 

var mockSet = new MockDbSet<User>(dataSource); 
var mockContext = new Mock<WebAPIDbContext>(); 

mockContext.Setup(c => c.Set<User>()).Returns(mockSet.Object); 

// ACT 
using (var uow = UnitOfWork.Create(mockContext.Object)) 
{ 
    uow.UserRepository.Add(user); 
    uow.SaveChanges(); 


    // ASSERT 
    mockSet.Verify(u => u.Add(It.IsAny<User>()), Times.Once()); 
    Assert.IsTrue(dataSource.Contains(user)); //<-- shows mock actually added item 
    Assert.IsTrue(uow.UserRepository.Any(u => u == user)); //<-- show you can actually query mock DbSet 
}