如何添加一个项目到模拟DbSet(使用Moq)
本文关键字:模拟 使用 Moq 项目 DbSet 何添加 添加 一个 | 更新日期: 2023-09-27 18:02:20
我试图设置一个模拟DbSet用于测试目的。我在这里使用了教程http://www.loganfranken.com/blog/517/mocking-dbset-queries-in-ef6/并稍微修改了它,以便每次调用GetEnumerator都会返回一个新的枚举器(我遇到的另一个问题)。但是,我在向DbSet添加项目时遇到困难。
输出为preCount = 3 postCount = 3。但是,我希望它是precount = 3 postCount = 4。如有任何帮助,不胜感激。
static void Main(string[] args)
{
Debug.WriteLine("hello debug");
List<string> stringList = new List<string>
{
"a", "b", "c"
};
DbSet<string> myDbSet = GetQueryableMockDbSet(stringList);
int preCount = myDbSet.Count();
myDbSet.Add("d");
int postCount = myDbSet.Count();
Debug.WriteLine("preCount = " + preCount + " postCount = " + postCount);
}
private static DbSet<T> GetQueryableMockDbSet<T>(List<T> sourceList) where T : class
{
var queryable = sourceList.AsQueryable();
var dbSet = new Mock<DbSet<T>>();
dbSet.As<IQueryable<T>>().Setup(m => m.Provider).Returns(queryable.Provider);
dbSet.As<IQueryable<T>>().Setup(m => m.Expression).Returns(queryable.Expression);
dbSet.As<IQueryable<T>>().Setup(m => m.ElementType).Returns(queryable.ElementType);
dbSet.As<IQueryable<T>>().Setup(m => m.GetEnumerator()).Returns(() => queryable.GetEnumerator());
return dbSet.Object;
}
myDbSet
不是DbSet
的真正实现,而是一个模拟,这意味着它是假的,它需要为您需要的所有方法设置。Add
也不例外,所以需要设置它来做你需要做的事情,否则它什么也不做。
添加如下内容,当调用myDbSet.Add("d");
时,'d'被添加到列表中,可以稍后返回。
dbSet.Setup(d => d.Add(It.IsAny<T>())).Callback<T>((s) => sourceList.Add(s));
完整代码private static DbSet<T> GetQueryableMockDbSet<T>(List<T> sourceList) where T : class
{
var queryable = sourceList.AsQueryable();
var dbSet = new Mock<DbSet<T>>();
dbSet.As<IQueryable<T>>().Setup(m => m.Provider).Returns(queryable.Provider);
dbSet.As<IQueryable<T>>().Setup(m => m.Expression).Returns(queryable.Expression);
dbSet.As<IQueryable<T>>().Setup(m => m.ElementType).Returns(queryable.ElementType);
dbSet.As<IQueryable<T>>().Setup(m => m.GetEnumerator()).Returns(() => queryable.GetEnumerator());
dbSet.Setup(d => d.Add(It.IsAny<T>())).Callback<T>((s) => sourceList.Add(s));
return dbSet.Object;
}
输出hello debug
preCount = 3 postCount = 4
为了处理.Find()
,我们可以以类似的方式使用反射,在使用Find
时,对一些约定做一些假设。
(希望这没有出格,我已经把这个问题收藏了很多年,并且正在寻找Find
的实现…)
实现另一个帮助器:
static object find(IEnumerable<object> oEnumerable, object[] keys)
{
// assumptions: primary key of object is named ID
// primary key of object is an int
// keys passed to .Find() method is a single value of int type
foreach (var o in oEnumerable)
{
var t = o.GetType();
var prop = t.GetProperty("ID");
if (prop != null)
{
if (prop.PropertyType == typeof(int))
{
if ((int)prop.GetValue(o) == (int)keys[0])
{
return o;
}
}
}
}
return null;
}
,在上面Daniel提供的模拟设置示例中:
dbSet.Setup(d => d.Find(It.IsAny<object[]>())).Returns((object[] oArray) => find(sourceList, oArray) as T);
因为我们没有办法(或者通常不想)像EF那样确定主键,所以我假设"ID"是关键字段名(与我自己的约定相匹配),但这可以扩展为接受许多变体。我还假设只有一个整数被传递给Find
(这是我的标准约定),但是这也可以扩展为更强大的支持。