3

我想使用NSubstitute通过模拟DbSet来对 Entity Framework 6.x 进行单元测试。幸运的是，Scott Xu提供了一个很好的单元测试库，EntityFramework.Testing.Moq使用Moq。所以，我修改了他的代码以适合 NSubstitute 并且到目前为止它看起来不错，直到我想测试DbSet<T>.Add()方法DbSet<T>.Remove()。这是我的代码位：

public static class NSubstituteDbSetExtensions
{
  public static DbSet<TEntity> SetupData<TEntity>(this DbSet<TEntity> dbset, ICollection<TEntity> data = null, Func<object[], TEntity> find = null) where TEntity : class
  {
    data = data ?? new List<TEntity>();
    find = find ?? (o => null);

    var query = new InMemoryAsyncQueryable<TEntity>(data.AsQueryable());

    ((IQueryable<TEntity>)dbset).Provider.Returns(query.Provider);
    ((IQueryable<TEntity>)dbset).Expression.Returns(query.Expression);
    ((IQueryable<TEntity>)dbset).ElementType.Returns(query.ElementType);
    ((IQueryable<TEntity>)dbset).GetEnumerator().Returns(query.GetEnumerator());

#if !NET40
    ((IDbAsyncEnumerable<TEntity>)dbset).GetAsyncEnumerator().Returns(new InMemoryDbAsyncEnumerator<TEntity>(query.GetEnumerator()));
    ((IQueryable<TEntity>)dbset).Provider.Returns(query.Provider);
#endif

    ...

    dbset.Remove(Arg.Do<TEntity>(entity =>
                                 {
                                   data.Remove(entity);
                                   dbset.SetupData(data, find);
                                 }));

    ...

    dbset.Add(Arg.Do<TEntity>(entity =>
                              {
                                data.Add(entity);
                                dbset.SetupData(data, find);
                              });

    ...

    return dbset;
  }
}

我创建了一个测试方法，如：

[TestClass]
public class ManipulationTests
{
  [TestMethod]
  public void Can_remove_set()
  {
    var blog = new Blog();
    var data = new List<Blog> { blog };

    var set = Substitute.For<DbSet<Blog>, IQueryable<Blog>, IDbAsyncEnumerable<Blog>>()
                        .SetupData(data);

    set.Remove(blog);

    var result = set.ToList();

    Assert.AreEqual(0, result.Count);
  }
}

public class Blog
{
   ...
}

当测试方法调用set.Remove(blog). 它抛出一个InvalidOperationException错误消息

收藏已修改；枚举操作可能无法执行。

这是因为在调用该方法data时，假对象已被修改。set.Remove(blog)但是，原始斯科特的使用方式Moq不会导致问题。

因此，我set.Remove(blog)用一个块包装了该方法，try ... catch (InvalidOperationException ex)让该catch块什么都不做，然后测试不会抛出异常（当然）并按预期通过。

我知道这不是解决方案，但是如何实现单元测试DbSet<T>.Add()和DbSet<T>.Remove()方法的目标？

4

1 回答 1

2

这里发生了什么事？

  1. set.Remove(blog); - 这会调用之前配置的 lambda。
  2. data.Remove(entity); - 该项目已从列表中删除。
  3. dbset.SetupData(data, find);- 我们再次调用 SetupData，用新列表重新配置 Substitute。
  4. SetupData跑...
  5. 在那里，dbSetup.Remove正在调用，以便重新配置下次调用 Remove 时发生的情况。

好的，我们这里有问题。dtSetup.Remove(Arg.Do<T....不会重新配置任何东西，而是将一种行为添加到 Substitute 的内部列表中，该列表应在您调用 Remove 时发生。因此，我们当前正在运行先前配置的删除操作 (1)，同时，在堆栈中，我们正在向列表中添加一个操作 (5)。当堆栈返回并且迭代器寻找下一个要调用的动作时，底层的模拟动作列表已经改变。迭代器不喜欢变化。

这导致了结论：我们不能修改替代品在其模拟操作之一运行时所做的事情。如果您考虑一下，阅读您的测试的任何人都不会认为会发生这种情况，因此您根本不应该这样做。

我们该如何解决？

public static DbSet<TEntity> SetupData<TEntity>(
    this DbSet<TEntity> dbset,
    ICollection<TEntity> data = null,
    Func<object[], TEntity> find = null) where TEntity : class
{
    data = data ?? new List<TEntity>();
    find = find ?? (o => null);

    Func<IQueryable<TEntity>> getQuery = () => new InMemoryAsyncQueryable<TEntity>(data.AsQueryable());

    ((IQueryable<TEntity>) dbset).Provider.Returns(info => getQuery().Provider);
    ((IQueryable<TEntity>) dbset).Expression.Returns(info => getQuery().Expression);
    ((IQueryable<TEntity>) dbset).ElementType.Returns(info => getQuery().ElementType);
    ((IQueryable<TEntity>) dbset).GetEnumerator().Returns(info => getQuery().GetEnumerator());

#if !NET40
    ((IDbAsyncEnumerable<TEntity>) dbset).GetAsyncEnumerator()
                                            .Returns(info => new InMemoryDbAsyncEnumerator<TEntity>(getQuery().GetEnumerator()));
    ((IQueryable<TEntity>) dbset).Provider.Returns(info => getQuery().Provider);
#endif

    dbset.Remove(Arg.Do<TEntity>(entity => data.Remove(entity)));
    dbset.Add(Arg.Do<TEntity>(entity => data.Add(entity)));

    return dbset;
}
  1. getQuerylambda 创建一个新查询。它总是使用捕获的列表data。
  2. 所有.Returns配置调用都使用 lambda。在那里，我们创建一个新的查询实例并将我们的调用委托给那里。
  3. Remove并且Add只修改我们捕获的列表。我们不必重新配置我们的 Substitute，因为每次调用都会使用 lambda 表达式重新评估查询。

虽然我真的很喜欢 NSubstitute，但我强烈建议您研究Effort，实体框架单元测试工具。

你会像这样使用它：

// DbContext needs additional constructor:
public class MyDbContext : DbContext
{
    public MyDbContext(DbConnection connection) 
        : base(connection, true)
    {
    }
}

// Usage:
DbConnection connection = Effort.DbConnectionFactory.CreateTransient();    
MyDbContext context = new MyDbContext(connection);

并且您有一个实际的 DbContext，您可以使用 Entity Framework 为您提供的所有内容，包括迁移，使用快速的内存数据库。

于 2014-12-14T21:02:27.200 回答