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

时间:2018-06-11 10:07:59

标签: c# entity-framework nunit moq

我尝试从存储库类中为Add方法编写单元测试。我正在使用EF6和Moq。我的测试方法看起来:

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对象。怎么了?怎么写这个测试是否正确?

1 个答案:

答案 0 :(得分:2)

将DbSet的伪数据源提取到局部变量中,以便稍后在测试设置中进行交互。向Callback设置添加Add以将传递的参数添加到数据源,以便在调用时模拟对实际数据执行操作。

// 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