如何写这个EF模拟设置代码为可重复使用通用的样板? [英] How to write this EF Mock setup code as a reusable Generic Boilerplate?
问题描述
我用的起订量,EF 6和的xUnit。我发现自己一遍又一遍地编写这些代码,也许以为我可以把它变成一个通用的方法,但遇到一些麻烦。
I am using moq, ef 6 and xunit. I find myself writing this code over and over and thought maybe I could make it into a generic method but having some trouble.
public static void CreateSalesMock(List<Sale> sales, Mock<DatabaseContext> dbContextMock)
{
var data = sales.AsQueryable();
var mockSet = new Mock<DbSet<Sale>>();
mockSet.As<IQueryable<Sale>>()
.Setup(x => x.Provider)
.Returns(data.Provider);
mockSet.As<IQueryable<Sale>>()
.Setup(x => x.Expression)
.Returns(data.Expression);
mockSet.As<IQueryable<Sale>>()
.Setup(x => x.ElementType)
.Returns(data.ElementType);
mockSet.As<IQueryable<Sale>>()
.Setup(x => x.GetEnumerator())
.Returns(data.GetEnumerator());
dbContextMock.Setup(x => x.Sales).Returns(mockSet.Object);
}
现在我有很多其他表在我的数据库,所以如果我可以写一个方法这将需要该数据的列表,并设定,让我可以模拟查询通过这将是巨大的。
Now I have many other tables in my database so if I could write one method that would take in a list of that data and set it up so I can mock query through it would be great.
public static void CreateMockSet<T, TA, TB>(T dataList, TA model,
Func<TB> lambda, Mock<DatabaseContext> dbContextMock)
where T : List<T>
where TA: Mock<DbSet<TA>>
{
var data = dataList.AsQueryable();
model.As<IQueryable<T>>()
.Setup(x => x.Provider)
.Returns(data.Provider);
model.As<IQueryable<T>>()
.Setup(x => x.Expression)
.Returns(data.Expression);
model.As<IQueryable<T>>()
.Setup(x => x.ElementType)
.Returns(data.ElementType);
model.As<IQueryable<T>>()
.Setup(x => x.GetEnumerator())
.Returns(data.GetEnumerator());
dbContextMock.Setup(x => lambda);
}
到目前为止,我有这一点,但我不知道是否会工作或没有。我被困传入拉姆达的一部分( X => x.Sales
为例),所以我甚至无法对其进行测试。
So far I have that but I am not sure if that will work or not. I am stuck passing in the "lambda" part(x => x.Sales
for example) so I can't even test it.
推荐答案
蒂姆·拉尔森已经提供了对的他的博客:
Tim Larson already offered a great solution for this boilerplate code in his blog:
public static class DbSetMocking
{
private static Mock<DbSet<T>> CreateMockSet<T>(IQueryable<T> data)
where T : class
{
var queryableData = data.AsQueryable();
var mockSet = new Mock<DbSet<T>>();
mockSet.As<IQueryable<T>>().Setup(m => m.Provider)
.Returns(queryableData.Provider);
mockSet.As<IQueryable<T>>().Setup(m => m.Expression)
.Returns(queryableData.Expression);
mockSet.As<IQueryable<T>>().Setup(m => m.ElementType)
.Returns(queryableData.ElementType);
mockSet.As<IQueryable<T>>().Setup(m => m.GetEnumerator())
.Returns(queryableData.GetEnumerator());
return mockSet;
}
public static IReturnsResult<TContext> ReturnsDbSet<TEntity, TContext>(
this IReturns<TContext, DbSet<TEntity>> setup,
TEntity[] entities)
where TEntity : class
where TContext : DbContext
{
Mock<DbSet<TEntity>> mockSet;
return ReturnsDbSet(setup, entities, out mockSet);
}
public static IReturnsResult<TContext> ReturnsDbSet<TEntity, TContext>(
this IReturns<TContext, DbSet<TEntity>> setup,
IQueryable<TEntity> entities)
where TEntity : class
where TContext : DbContext
{
Mock<DbSet<TEntity>> mockSet;
return ReturnsDbSet(setup, entities, out mockSet);
}
public static IReturnsResult<TContext> ReturnsDbSet<TEntity, TContext>(
this IReturns<TContext, DbSet<TEntity>> setup,
IEnumerable<TEntity> entities)
where TEntity : class
where TContext : DbContext
{
Mock<DbSet<TEntity>> mockSet;
return ReturnsDbSet(setup, entities, out mockSet);
}
public static IReturnsResult<TContext> ReturnsDbSet<TEntity, TContext>(
this IReturns<TContext, DbSet<TEntity>> setup,
TEntity[] entities, out Mock<DbSet<TEntity>> mockSet)
where TEntity : class
where TContext : DbContext
{
mockSet = CreateMockSet(entities.AsQueryable());
return setup.Returns(mockSet.Object);
}
public static IReturnsResult<TContext> ReturnsDbSet<TEntity, TContext>(
this IReturns<TContext, DbSet<TEntity>> setup,
IQueryable<TEntity> entities, out Mock<DbSet<TEntity>> mockSet)
where TEntity : class
where TContext : DbContext
{
mockSet = CreateMockSet(entities);
return setup.Returns(mockSet.Object);
}
public static IReturnsResult<TContext> ReturnsDbSet<TEntity, TContext>(
this IReturns<TContext, DbSet<TEntity>> setup,
IEnumerable<TEntity> entities, out Mock<DbSet<TEntity>> mockSet)
where TEntity : class
where TContext : DbContext
{
mockSet = CreateMockSet(entities.AsQueryable());
return setup.Returns(mockSet.Object);
}
}
然后在UT你使用它如下所示:
Then in the UT you use it as the following:
var context = new Mock<DatabaseContext>();
context.setup(x => x.Sales).ReturnsDbSet(new List<Sale>(){put here the items..});
修改
我更新的代码。现在有3个重载它允许在 DbSet<验证; T>
属性:
I updated the code. Now there are 3 more overloads which allows verification on DbSet<T>
property:
[TestMethod]
public void TestMethod1()
{
var sales = new List<Sale>
{
new Sale() {id = 1},
new Sale() {id = 6},
new Sale() {id = 5},
new Sale() {id = 4},
new Sale() {id = 3},
new Sale() {id = 2}
};
var fakeContest = new Mock<SalesContext>();
Mock<DbSet<Sale>> fakeSet;
fakeContest.Setup(context => context.Sales).ReturnsDbSet(sales, out fakeSet);
var itemsToRemove = sales.Where(sale => sale.id%2 == 0);
fakeContest.Object.Sales.RemoveRange(itemsToRemove);
fakeSet.Verify(set => set.RemoveRange(itemsToRemove));
}
这篇关于如何写这个EF模拟设置代码为可重复使用通用的样板?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!