使用Moq模拟IList.Add

时间:2019-04-21 18:47:14

标签: c# unit-testing entity-framework-6 moq

我正在尝试设置一个moq对象,以对使用EF6的业务层对象运行一些单元测试。 到目前为止,我可以测试简单的方法调用并检查是否调用了这些方法。但是,有没有一种方法可以测试对象是否确实插入了基础集合中。

这是我的模拟方法

private Mock<DbSet<T>> CreateMockDbSet<T>(IQueryable<T> entities) where T : class
{
    var mockSet = new Mock<DbSet<T>>();
    mockSet.As<IQueryable<T>>().Setup(m => m.Provider).Returns(entities.Provider);
    mockSet.As<IQueryable<T>>().Setup(m => m.Expression).Returns(entities.Expression);
    mockSet.As<IQueryable<T>>().Setup(m => m.ElementType).Returns(entities.ElementType);
    mockSet.As<IQueryable<T>>().Setup(m => m.GetEnumerator()).Returns(entities.GetEnumerator());
    IList<T> list = entities as IList<T>;
    mockSet.As<IList<T>>().Setup(m => m.Add(It.IsAny<T>())).Returns(list.Add(It.IsAny<T>()));
    mockSet.Setup(m => m.Include(It.IsAny<string>())).Returns(mockSet.Object);
    return mockSet;
}

我正在尝试从列表中模拟实际的Add方法,但如上所述,没有方法.Return

这甚至可能吗?

我需要验证是否在发生某种逻辑后将该对象插入到我的模拟集合中。

1 个答案:

答案 0 :(得分:3)

IList.Add方法返回void,因此您无法设置return,而是使用.Callback

mockSet.As<IList<T>>()
    .Setup(m => m.Add(It.IsAny<T>()))
    .Callback<T>(item => list.Add(item));

或使用Capture.In

mockSet.As<IList<T>>()
    .Setup(m => m.Add(Capture.In(list)));