如何在 C# 中使用 Moq 测试 Add 方法

Question

我尝试为存储库类中的 Add 方法编写一个单元测试。我正在使用 EF6 和起订量。我的测试方法看起来：

public static Mock<DbSet<T>> CreateDbSetMock<T>(IEnumerable<T> elements) where T : class
{
    var elementsAsQueryable = elements.AsQueryable();
    var dbSetMock = new Mock<DbSet<T>>();

    dbSetMock.As<IQueryable<T>>().Setup(m => m.Provider).Returns(elementsAsQueryable.Provider);
    dbSetMock.As<IQueryable<T>>().Setup(m => m.Expression).Returns(elementsAsQueryable.Expression);
    dbSetMock.As<IQueryable<T>>().Setup(m => m.ElementType).Returns(elementsAsQueryable.ElementType);
    dbSetMock.As<IQueryable<T>>().Setup(m => m.GetEnumerator()).Returns(elementsAsQueryable.GetEnumerator());

    return dbSetMock;
}

[Test()]
public void AddTest()
{
    // Arrange
    Mock<DbSet<Tytul>> titlesMock = CreateDbSetMock(new List<Tytul>());
    Mock<OzinDbContext> titlesContextMock = new Mock<OzinDbContext>();
    titlesContextMock.Setup(x => x.Tytuly).Returns(titlesMock.Object);
    titlesMock.Setup(x => x.Add(It.IsAny<Tytul>())).Returns((Tytul t) => t);
    IRepository<Tytul> tytulRepository = TytulRepository(titlesContextMock.Object);

    Tytul tytul = new Tytul
    {
        Identyfikator = "ABC"               
    };

    // Act
    tytulRepository.Add(tytul);
         // in Add method:
         //dbContext.Tytuly.Add(entity);
         //dbContext.SaveChanges();
    Tytul tytulInDb = tytulRepository.GetDetail(t => t.Identyfikator == "ABC");
         // in GetDetail method:
         //return dbContext.Tytuly.FirstOrDefault(predicate);

    // Assert
    Assert.AreEqual(tytulInDb.Identyfikator, tytul.Identyfikator);
}

我的问题是 GetDetail 方法返回 null，但我期望 tytulInDb 对象。怎么了？如何正确地编写这个测试？

Answer 1

将 DbSet 的虚假数据源提取到本地变量中，以便稍后可以在测试设置中进行交互。Callback在设置中添加 aAdd以将传递的参数添加到数据源，以便模拟在调用时可以对实际数据进行操作。

// Arrange
var data = new List<Tytul>(); //<<< local variable
Mock<DbSet<Tytul>> titlesMock = CreateDbSetMock(data);
var titlesContextMock = new Mock<OzinDbContext>();
titlesContextMock.Setup(x => x.Tytuly).Returns(titlesMock.Object);
titlesMock
    .Setup(x => x.Add(It.IsAny<Tytul>()))
    .Returns((Tytul t) => t)
    .Callback((Tytul t) => data.Add(t)); //<<< for when mocked Add is called.
IRepository<Tytul> tytulRepository = TytulRepository(titlesContextMock.Object);

//...Code removed for brevity

此外，在设置 DbSet Mock 时，请使用委托来Returns允许多个枚举，因为仅返回值将仅允许仅向前枚举器传递一次。

dbSetMock.As<IQueryable<T>>()
    .Setup(m => m.GetEnumerator())
    .Returns(() => elementsAsQueryable.GetEnumerator()); //<<< note delegate