![](/img/trans.png)
[英]Unit Testing Generic Unit of Work and Repository Pattern framework using Moq
[英]Unit test an Entity Framework generic repository using Moq
问题/问题
我无法通过测试,因为通用存储库类this.dbSet = context.Set<T>();
始终为null
。 正如您在下面的代码中看到的,我DbSet
了DbSet
和上下文。 我还设置了DbSet
上下文以返回DbSet
。 EnityRepository
构造函数按预期采用EnityRepository
上下文,但this.dbSet = context.Set<T>();
没有拿起我嘲笑的DbSet
。 我不确定我做错了什么。 我不是在以正确的方式嘲笑吗?
结构:
IService
通用存储库
public class EntityRepository<T> : IEntityRepository<T> where T : class
{
internal MyDB_Entities context;
internal DbSet<T> dbSet;
public EntityRepository(MyDB_Entities context)
{
this.context = context;
this.dbSet = context.Set<T>();
}
public virtual T GetByID(object id)
{
return dbSet.Find(id);
}
// more code
}
通用存储库接口
public interface IEntityRepository<T> where T : class
{
IEnumerable<T> Get(Expression<Func<T, bool>> filter = null, Func<IQueryable<T>, IOrderedQueryable<T>> orderBy = null, string includeProperties = "");
T GetByID(object id);
// more code
}
工作单位
public class UnitOfWork : IUnitOfWork, IDisposable
{
MyDB_Entities _context;
public IEntityRepository<Customer> customerRepository { get; set; }
public IEntityRepository<Product> productRepository { get; set; }
public UnitOfWork(MyDB_Entities context)
{
_context = context;
customerRepository = new EntityRepository<Customer>(_context);
productRepository = new EntityRepository<Product>(_context);
}
public void Save()
{
_context.SaveChanges();
}
// more code
}
工作单元接口
public interface IUnitOfWork
{
IEntityRepository<Customer> customerRepository { get; set; }
IEntityRepository<Product> productRepository { get; set; }
void Dispose();
void Save();
}
服务
public class SomeService : ISomeService
{
readonly IUnitOfWork _unitOfWork;
public SomeService (IUnitOfWork unitOfWork)
{
_unitOfWork = unitOfWork;
}
// DoSomethingMethod
}
服务接口
public interface ISomeService
{
// IDoSomethingMethod
}
扩展
public static class MockDBSetExtension
{
public static void SetSource<T>(this Mock<DbSet<T>> mockSet, IList<T> source) where T : class
{
var data = source.AsQueryable();
mockSet.As<IQueryable<T>>().Setup(m => m.Provider).Returns(data.Provider);
mockSet.As<IQueryable<T>>().Setup(m => m.Expression).Returns(data.Expression);
mockSet.As<IQueryable<T>>().Setup(m => m.ElementType).Returns(data.ElementType);
mockSet.As<IQueryable<T>>().Setup(m => m.GetEnumerator()).Returns(data.GetEnumerator());
}
}
测试班
[TestClass]
public class My_Test
{
Mock<DbSet<Product>> _mockProductDBSet;
Mock<MyDB_Entities> mockContext;
[TestInitialize]
public void TestInitialize()
{
_mockProductDBSet = new Mock<DbSet<Product>>();
mockContext = new Mock<MyDB_Entities>();
mockContext.Setup(s => s.Products).Returns(_mockProductDBSet.Object);
}
[TestMethod]
public void TestMocking()
{
var prod = new Product() { ProductName= "AAA", ProductID = 1 };
_mockProductDBSet.SetSource(new List<Product> { prod });
// more code here (new up the service, then test the service method, etc)
}
}
假设您有一个IProuctService
定义为
public interface IProductService {
string GetProductName(int productId);
}
具体实现取决于IUnitOfWork
public class ProductService : IProductService {
readonly IUnitOfWork _unitOfWork;
public ProductService(IUnitOfWork unitOfWork) {
_unitOfWork = unitOfWork;
}
public string GetProductName(int productId) {
var item = _unitOfWork.productRepository.GetByID(productId);
if (item != null) {
return item.ProductName;
}
throw new ArgumentException("Invalid product id");
}
}
如果被测方法是IProductService.GetProductName
,这里是一个可以完成的测试示例。
[TestMethod]
public void ProductService_Given_Product_Id_Should_Get_Product_Name() {
//Arrange
var productId = 1;
var expected = "AAA";
var product = new Product() { ProductName = expected, ProductID = productId };
var productRepositoryMock = new Mock<IEntityRepository<Product>>();
productRepositoryMock.Setup(m => m.GetByID(productId)).Returns(product).Verifiable();
var unitOfWorkMock = new Mock<IUnitOfWork>();
unitOfWorkMock.Setup(m => m.productRepository).Returns(productRepositoryMock.Object);
IProductService sut = new ProductService(unitOfWorkMock.Object);
//Act
var actual = sut.GetProductName(productId);
//Assert
productRepositoryMock.Verify();//verify that GetByID was called based on setup.
Assert.IsNotNull(actual);//assert that a result was returned
Assert.AreEqual(expected, actual);//assert that actual result was as expected
}
在这种情况下,不需要模拟 DbSet 或 DbContext,因为 SUT 不需要相关接口的实现。 它们可以被模拟以供被测系统使用。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.