如何使用实体框架和 Moq 进行单元测试?

Question

我是Moq的新手，并希望将其用作数据的后备存储 - 但不接触实时数据库。

我的设置如下：

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

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

// ARRANGE
var user = new User()
{
FirstName = "Some",
LastName = "Guy",
EmailAddress = "[e-mail]",
};
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 = "[e-mail]",
};
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.
}

模拟数据库集

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

你只需要创建一个集合来充当后备存储，并使用后备集合模拟枚举数据库集

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 = "[e-mail]",
};
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
}