C#如何使用Moq entityframework DbSet添加方法

时间:2016-11-08 10:19:03

标签: c# unit-testing c#-4.0 entity-framework-4 moq

我正在尝试创建一个测试实体框架Add方法的测试。任何人都可以帮助如何模拟DbSet.Add方法。我尝试过以下但没有工作。我做错了什么?

我得到的结果是null后的repository.Insert ...

test.cs中:

var productToCreate = new Product { Name = "Added", Description = "Added" };        

var result = repository.InsertAsync(objToCreate, userContext).Result;
Assert.AreEqual(result.Name, "Added");  

Mock.cs

internal static DbSet<T> GetMockedDataSet<T>(IEnumerable<T> data) where T : class
{
    // Create a mocked data set that contains the data
    var set = new Mock<DbSet<T>>();
    set.As<IDbAsyncEnumerable<T>>()
        .Setup(m => m.GetAsyncEnumerator())
        .Returns(new TestDbAsyncEnumerator<T>(data.GetEnumerator()));
    set.As<IQueryable<T>>()
        .Setup(m => m.Provider)
        .Returns(new TestDbAsyncQueryProvider<T>(data.AsQueryable().Provider));
    set.As<IQueryable<T>>().Setup(m => m.Expression).Returns(data.AsQueryable().Expression);
    set.As<IQueryable<T>>().Setup(m => m.ElementType).Returns(data.AsQueryable().ElementType);
    set.As<IQueryable<T>>().Setup(m => m.GetEnumerator()).Returns(data.GetEnumerator());

    set.Setup(x => x.AsNoTracking()).Returns(set.Object);
    set.Setup(x => x.Add(It.IsAny<T>())).Callback<T>((s) => data.Concat(new[] { s }));

    // Return the mock
    return set.Object;
}

存储库:

public async Task<Product> InsertAsync(Product input)
{
    using (var ctx = .....))
    {
        var added = ctx.Set<Product>().Add(input);

        await ctx.ValidateAndSaveAsync();

        return added;
    }
}

1 个答案:

答案 0 :(得分:5)

根据测试方法中Add方法的使用方式...

var added = ctx.Set<Product>().Add(input);

...设置中还应该有一个Returns,它返回输入的参数,如果这是所需的功能。

set.Setup(x => x.Add(It.IsAny<T>()))
   .Returns<T>(arg => arg)
   .Callback<T>((s) => data.Concat(new[] { s }));

但鉴于有关上下文依赖的信息未知......

using (var ctx = .....))

不确定所提供的解决方案是否会产生预期的效果。

此外,如果测试异步方法,请不要混合异步和同步调用。以下一行......

var result = repository.InsertAsync(objToCreate, userContext).Result;

...会导致死锁。

让测试方法始终保持异步。

[TestMethod]
public async Task InsertAsync_Should_Return_Product() {
    //...other code

    var expected = new Product { Name = "Added", Description = "Added" };        

    var actual = await repository.InsertAsync(expected, userContext);

    Assert.AreEqual(expected.Name, actual.Name);  
}