28

我将 ASP.NET MVC 4 与 Entity Framework 5 一起使用。我有模型类和实体映射来将现有表映射到这些模型类。所有这些设置都很好，效果很好。

现在我想嘲笑这个。我创建了采用 DataContext 并使用通用存储库的工作单元。在那之后，我构建了能够一次从许多存储库中获取数据的服务，并且只需要一个 DataContext 实例。这也很好用。

现在问题来了：我想用模拟数据测试服务。当我创建工作单元实例时，我希望能够插入一个模拟的 DataContext 而不是真正的 DataContext。

我试图创建一个 IContext 接口并让真实的和模拟的 DataContext 实现它，但遇到了 DbSet 的问题。我尝试使用 IDbSet 并创建 FakeDbSet 但没有成功。我还在互联网上读到使用 IDbSet 模拟上下文并使用 FakeDbSet 是一种不好的方法。

您知道实现这一目标的最佳方法是什么吗？我现在拥有的是我想保留的行为，但真的希望能够模拟来自 DataContext 中的模型类的数据。

我知道实体框架已经带有工作单元行为，并且您不需要在此之上添加额外的行为。但我想将它包装在另一个跟踪所有存储库的类（称为 UnitOfWork 类）中。

编辑：我写了两篇文章来解释我使用 LINQ 和实体框架的解决方案。

http://gaui.is/how-to-mock-the-datacontext-linq/

http://gaui.is/how-to-mock-the-datacontext-entity-framework/

这是我的代码：

IRepository.cs

public interface IRepository<T> where T : class
{
    void Add(T entity);
    void Delete(T entity);
    void Update(T entity);
    T GetById(long Id);
    IEnumerable<T> All();
    IEnumerable<T> Find(Expression<Func<T, bool>> predicate);
}

IUnitOfWork.cs

public interface IUnitOfWork : IDisposable
{
    IRepository<TEntity> GetRepository<TEntity>() where TEntity : class;
    void Save();
}

存储库.cs

public class Repository<T> : IRepository<T> where T : class
{
    private readonly IDbContext _context;
    private readonly IDbSet<T> _dbset;

    public Repository(IDbContext context)
    {
        _context = context;
        _dbset = context.Set<T>();
    }

    public virtual void Add(T entity)
    {
        _dbset.Add(entity);
    }

    public virtual void Delete(T entity)
    {
        var entry = _context.Entry(entity);
        entry.State = System.Data.EntityState.Deleted;
    }

    public virtual void Update(T entity)
    {
        var entry = _context.Entry(entity);
        _dbset.Attach(entity);
        entry.State = System.Data.EntityState.Modified;
    }

    public virtual T GetById(long id)
    {
        return _dbset.Find(id);
    }

    public virtual IEnumerable<T> All()
    {
        return _dbset;
    }

    public IEnumerable<T> Find(Expression<Func<T, bool>> predicate)
    {
        return _dbset.Where(predicate);
    }
}

UnitOfWork.cs

public class UnitOfWork<TContext> : IUnitOfWork where TContext : IDbContext, new()
{
    private readonly IDbContext _ctx;
    private Dictionary<Type, object> _repositories;
    private bool _disposed;

    public UnitOfWork()
    {
        _ctx = new TContext();
        _repositories = new Dictionary<Type, object>();
        _disposed = false;
    }

    public IRepository<TEntity> GetRepository<TEntity>() where TEntity : class
    {
        if (_repositories.Keys.Contains(typeof(TEntity)))
            return _repositories[typeof(TEntity)] as IRepository<TEntity>;

        var repository = new Repository<TEntity>(_ctx);
        _repositories.Add(typeof(TEntity), repository);
        return repository;
    }

    public void Save()
    {
        _ctx.SaveChanges();
    }

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

    protected virtual void Dispose(bool disposing)
    {
        if (!this._disposed)
        {
            if (disposing)
            {
                _ctx.Dispose();
            }

            this._disposed = true;
        }
    }
}

示例服务.cs

public class ExampleService
{
    private IRepository<Example> m_repo;

    public ExampleService(IUnitOfWork uow)
    {
        m_repo = uow.GetRepository<Example>();
    }

    public void Add(Example Example)
    {
        m_repo.Add(Example);
    }

    public IEnumerable<Example> getAll()
    {
        return m_repo.All();
    }
}

ExampleController.cs

public IEnumerable<Example> GetAll()
{
    // Create Unit Of Work object
    IUnitOfWork uow = new UnitOfWork<AppDataContext>();

    // Create Service with Unit Of Work attached to the DataContext
    ExampleService service = new ExampleService(uow);

    return service.getAll();
}
4

2 回答 2

10

你的ExampleService类是期待IUnitOfWork的，这意味着你只需要另一个IUnitOfWork是 Mock 并且它的GetRepository()方法将返回一个IRepositoryMock。

例如（不是真正的 Mock，而是 In-Memory 存根）：

  public InMemoryRepository<T> : IRepository<T> where T : class
  {
        ........
  }

  public InMemoryUnitOfWork : IUnitOfWork
  {
       public IRepository<TEntity> GetRepository<TEntity>() where TEntity : class
       {
            return new InMemoryRepository<TEntity>();
       }
  }

然后：

public IEnumerable<Example> GetAll()
{
    // Create Unit Of Work object
    IUnitOfWork uow = new InMemoryUnitOfWork();

    // Create Service with Unit Of Work
    ExampleService service = new ExampleService(uow);

    return service.getAll();
}
于 2013-05-23T14:58:14.140 回答
0

您可以点击以下链接，它非常有帮助。

具有实体框架的 MVC3 应用程序中的通用存储库模式

实体框架和数据模式

于 2013-12-28T01:05:15.633 回答