2017-10-19 94 views
0

我正在开发一个NUnit,Moq,C#和.NET Framework 4.7的测试项目。在lambda表达式中使用泛型或如何使用泛型访问dbset

我有这两种方法是几乎等于:

using Microsoft.EntityFrameworkCore; 
using Moq; 
using NUnit.Framework; 
using System; 
using System.Collections.Generic; 
using System.Linq; 

private void MockProducts(Mock<TRZFDbContext> mockContext, List<Product> data) 
{ 
    IQueryable<Product> queryable = data.AsQueryable(); 

    var mockSet = new Mock<DbSet<Product>>(); 

    mockSet.As<IQueryable<Product>>().Setup(m => m.Provider).Returns(queryable.Provider); 
    mockSet.As<IQueryable<Product>>().Setup(m => m.Expression).Returns(queryable.Expression); 
    mockSet.As<IQueryable<Product>>().Setup(m => m.ElementType).Returns(queryable.ElementType); 
    mockSet.As<IQueryable<Product>>().Setup(m => m.GetEnumerator()).Returns(() => queryable.GetEnumerator()); 
    mockSet.Setup(d => d.Add(It.IsAny<Product>())).Callback<Product>((s) => data.Add(s)); 
    mockSet.Setup(d => d.Remove(It.IsAny<Product>())).Callback<Product>((s) => data.Remove(s)); 

    mockContext.Setup(m => m.Product).Returns(mockSet.Object); 
} 

private void MockEmvoProducts(Mock<TRZFDbContext> mockContext, List<EmvoProduct> data) 
{ 
    IQueryable<EmvoProduct> queryable = data.AsQueryable(); 

    var mockSet = new Mock<DbSet<EmvoProduct>>(); 

    mockSet.As<IQueryable<EmvoProduct>>().Setup(m => m.Provider).Returns(queryable.Provider); 
    mockSet.As<IQueryable<EmvoProduct>>().Setup(m => m.Expression).Returns(queryable.Expression); 
    mockSet.As<IQueryable<EmvoProduct>>().Setup(m => m.ElementType).Returns(queryable.ElementType); 
    mockSet.As<IQueryable<EmvoProduct>>().Setup(m => m.GetEnumerator()).Returns(() => queryable.GetEnumerator()); 
    mockSet.Setup(d => d.Add(It.IsAny<EmvoProduct>())).Callback<EmvoProduct>((s) => data.Add(s)); 
    mockSet.Setup(d => d.Remove(It.IsAny<EmvoProduct>())).Callback<EmvoProduct>((s) => data.Remove(s)); 

    mockContext.Setup(m => m.EmvoProduct).Returns(mockSet.Object); 
} 

我可以创建一个泛型方法改变的第二个参数data与通用。但我的问题是最后一句:

mockContext.Setup(m => m.Product).Returns(mockSet.Object); 

mockContext.Setup(m => m.EmvoProduct).Returns(mockSet.Object); 

m.ProductProduct一个DbSet和m.EmvoProductEmvoProduct一个DbSet。

有没有办法让lambda表达式使用泛型?或者也许有另一种方法。

+0

基本上你想创建功能设置? –

+0

是的。但问题是如何使它成为泛型。也许我必须像在这里一样将最新的句子移到外面:https://www.rhyous.com/2015/04/10/how-to-mock-an-entity-framework-dbcontext-and-its-dbset-属性/ – VansFannel

+0

链接中的哪个行号? –

回答

1

你应该能够与被像这样声明的公共静态类的扩展功能,这样做

public static class DBSetExtensions { 

    public static void Setup<T>(this DbSet<T> list, Action<T> lamda){ 
     lamda(objectFromDbSet); 
    } 

} 

现在任何类,你有一个DbSet泛型,你应该能够使用

mockSet.Setup(d => d.Add(It.IsAny<EmvoProduct>())) 

对于链接功能Callback,你需要他们在设置恢复任何对象从虚空上述改变对函数的返回类型。

+0

您可以添加一个关于如何使用你的答案?谢谢。 – VansFannel

+0

我已更新它。只要它是一个静态函数,它就可以在你使用DBSet对象的任何地方工作 –