c# - 模拟 DbContext.set<T>.Add() EF6

Question

我有以下类(其中 PilsnerContext 是 DbContext 类):

public abstract class ServiceBase<T> : IService<T> where T: class, IEntity
{
    protected readonly PilsnerContext Context;

    protected ServiceBase(PilsnerContext context)
    {
        Context = context;
    }

    public virtual T Add(T entity)
    {
        var newEntity = Context.Set<T>().Add(entity);
        Context.SaveChanges();
        return newEntity;
    }
}

public class ProspectsService : ServiceBase<Prospect>
{
    public ProspectsService(PilsnerContext context) : base(context){}

}

我正在尝试对模拟上下文的 Add 方法进行单元测试，例如:

[TestClass]
public class ProspectTest
{
    [TestMethod]
    public void AddProspect()
    {
        var mockProspect = new Mock<DbSet<Prospect>>();
        var mockContext = new Mock<PilsnerContext>();

        mockContext.Setup(m => m.Prospects).Returns(mockProspect.Object);

        var prospectService = new ProspectsService(mockContext.Object);

        var newProspect = new Prospect()
        {
            CreatedOn = DateTimeOffset.Now,
            Browser = "IE",
            Number = "1234567890",
            Visits = 0,
            LastVisitedOn = DateTimeOffset.Now
        };

        prospectService.Add(newProspect);

        mockProspect.Verify(m=>m.Add(It.IsAny<Prospect>()), Times.Once);
        mockContext.Verify(m=>m.SaveChanges(), Times.Once);
    }
}

但是断言:

mockProspect.Verify(m=>m.Add(It.IsAny<Prospect>()), Times.Once);

我认为失败是因为我在 Add 方法中使用了 Context.set().Add() 而不是 Context.Prospects.Add() 但是通过此测试的正确方法是什么？

异常(exception)情况是:

Expected invocation on the mock once, but was 0 times: m => m.Add(It.IsAny<Prospect>()) No setups configured. No invocations performed.

提前致谢。

Answer 1

看起来您只是缺少返回 DbSet 的设置:

mockContext.Setup(m => m.Set<Prospect>()).Returns(mockProspect.Object);