我的 Mocked DbContext 使用 Moq 在单元测试中写入 db

Question

我是模拟和使用 Moq 的新手。这是我第一次尝试模拟添加功能。我有 mocks 设置，它适用于读取函数，但是当我尝试对 add 方法进行单元测试时，它会将更改保存到 db。

如何模拟 add 方法？

我想我必须模拟具有 SaveChanges() 方法的 UnitOfWork 类。我想知道如何设置模拟来拦截对SaveChanges() 的调用而不是保存到数据库。

这是我到目前为止的模拟

[SetUp]
    public void SetUp()
    {
        addCount = 0;

        IEnumerable<Platform> platformList = new List<Platform>(){
            new Platform() { Id = 1, Name = "Unknown"},
            new Platform() { Id =2, Name = "Amazon"},
            new Platform() { Id = 3, Name = "Prime Pantry"}
        };
        var platformData = platformList.AsQueryable();

        var mockPlatformSet = new Mock<DbSet<Platform>>();
        mockPlatformSet.As<IQueryable<Platform>>().Setup(m => m.Provider).Returns(platformData.Provider);
        mockPlatformSet.As<IQueryable<Platform>>().Setup(m => m.Expression).Returns(platformData.Expression);
        mockPlatformSet.As<IQueryable<Platform>>().Setup(m => m.ElementType).Returns(platformData.ElementType);
        mockPlatformSet.As<IQueryable<Platform>>().Setup(m => m.GetEnumerator()).Returns(platformData.GetEnumerator());
        mockPlatformSet.Setup(m => m.Add(It.IsAny<Platform>())).Callback(() => addCount++);

        var mockContext = new Mock<ApplicationDbContext>(){ CallBase = true };
        mockContext.Setup(m => m.Platforms).Returns(mockPlatformSet.Object);
        mockContext.Setup(m => m.Platforms.Add(It.IsAny<Platform>()));
        mockContext.Setup(m => m.Platforms.Add(It.IsAny<Platform>())).Callback(() => addCount++);


        unitOfWork = new UnitOfWork(mockContext.Object);
        platformRepo = new PlatformRepository(mockContext.Object);

        controller = new PlatformController(platformRepo, unitOfWork);
    }

添加 UnitOfWork 代码

public class UnitOfWork : IUnitOfWork 
{
    private readonly DbContext _context;
    private bool _isDisposed = false;

    public UnitOfWork(DbContext context)
    {
        _context = context;
    }

    public void SaveChanges()
    {
        _context.SaveChanges();
    }

    protected virtual void Dispose(bool disposing)
    {
        if (!_isDisposed)
        {
            if (disposing)
            {
                _context.Dispose();
            }
        }
        _isDisposed = true;
    }

    public void Dispose()
    {
        Dispose(true);
        GC.SuppressFinalize(this);
    }

}

更新问题

我正在尝试对 PlatformController 的 Create 方法进行单元测试。在这种方法中，我在存储库上调用Add 函数，然后调用UnitOfWork 的SaveChanges 函数。我想验证我的 Platform 对象是否已添加到 DbSet，但“拦截”对 SaveChanges() 的调用以使其不写入 db。

我该怎么做？

Answer 1

当我将我的mockContext 的CallBase 值设置为IUnitOfWork 为false 时，这似乎解决了我的单元测试写入数据库的问题。

这行代码： mockContext.As<IUnitOfWork>().CallBase = false;

这是我的Setup函数的代码

[SetUp]
    public void SetUp()
    {
        addCount = 0;

        IEnumerable<Platform> platformList = new List<Platform>(){
            new Platform() { Id = 1, Name = "Unknown"},
            new Platform() { Id =2, Name = "Amazon"},
            new Platform() { Id = 3, Name = "Prime Pantry"}
        };
        var platformData = platformList.AsQueryable();

        var mockPlatformSet = new Mock<DbSet<Platform>>();
        mockPlatformSet.As<IQueryable<Platform>>().Setup(m => m.Provider).Returns(platformData.Provider);
        mockPlatformSet.As<IQueryable<Platform>>().Setup(m => m.Expression).Returns(platformData.Expression);
        mockPlatformSet.As<IQueryable<Platform>>().Setup(m => m.ElementType).Returns(platformData.ElementType);
        mockPlatformSet.As<IQueryable<Platform>>().Setup(m => m.GetEnumerator()).Returns(platformData.GetEnumerator());
        mockPlatformSet.Setup(m => m.Add(It.IsAny<Platform>())).Callback(() => addCount++);

        var mockContext = new Mock<ApplicationDbContext>(){ CallBase = true };
        mockContext.Setup(m => m.Platforms).Returns(mockPlatformSet.Object);
        mockContext.Setup(m => m.Platforms.Add(It.IsAny<Platform>()));
        mockContext.Setup(m => m.Platforms.Add(It.IsAny<Platform>())).Callback(() => addCount++);
        mockContext.Setup(m => m.Set<Platform>()).Returns(mockPlatformSet.Object);
        mockContext.As<IUnitOfWork>().CallBase = false;

        unitOfWork = new UnitOfWork(mockContext.Object);
        platformRepo = new PlatformRepository(mockContext.Object);

        controller = new PlatformController(platformRepo, unitOfWork);
    }

Answer 2

我不确定这是否是您正在寻找的答案，但是使用工作单元抽象 EF DbContext 是一个糟糕的主意。原因是上下文已经是一个工作单元实现。根据msdn中类的描述：

表示工作单元和存储库模式的组合 并使您能够查询数据库并将更改组合在一起 然后将作为一个单元写回商店。

一旦您删除了不必要的抽象，模拟上下文应该相当容易，特别是如果您使用最新版本的实体框架。

Answer 3

您正在尝试在这里进行功能测试，因此拥有一个功能数据库是明智之举。

EF 可以使用测试连接字符串在设置和拆卸方法中重新创建和销毁数据库。这将为您的测试提供一个真实的功能测试环境，以模拟真实环境。

例如：

    [TestFixtureSetUp]
    public static void SetupFixture() //create database
    {
        using (var context = new XEntities())
        {
            context.Setup();
        }
    }

    [TestFixtureTearDown]
    public void TearDown() //drop database
    {
        using (var context = new XEntities())
        {
            context.Database.Delete();
        }
    }

    [SetUp]
    public void Setup() //Clear entities before each test so they are independent
    {
        using (var context = new XEntities())
        {
            foreach (var tableRow in context.Table)
            {
                context.Table.Remove(tableRow);
            }
            context.SaveChanges();
        }
    }

将您的测试项目中的连接字符串更改为指向“DbNameTest”，这样您就可以在测试中使用您的 XEntities 类，并且设置和添加要与之交互的测试数据是很清楚的。