使用NUnit和NSubstitute对通用存储库进行单元测试

时间:2016-01-20 16:37:32

标签: c# entity-framework unit-testing nsubstitute

我如何使用NUNit和NSubstitute执行单元测试,我想测试插入假的" GamaItem"对象,并验证它是否有效,以及是否已触发SaveChanges。

我是单元测试中的新手,我不知道如何伪造dbContext对象。

提前致谢。

工作单位:

public class UnitOfWork: IUnitOfWork, IDisposable
{
    private SRColorContext context = new SRColorContext();
    private GenericEntityRepository<HairColorType> hairColorTypeRepository;
    private GenericEntityRepository<GamaItem> gamaItemRepository;

    public UnitOfWork()
    {
        if (context == null)
        {
            context = new SRColorContext();
        }
    }

    public UnitOfWork(SRColorContext context)
    {
        this.context = context;
    }


    public GenericEntityRepository<HairColorType> HairColorTypeRepository
    {
        get
        {
            if (this.hairColorTypeRepository == null)
            {
                this.hairColorTypeRepository = new GenericEntityRepository<HairColorType>(context);
            }
            return hairColorTypeRepository;
        }
    }

    public GenericEntityRepository<GamaItem> GamaItemRepository
    { 
        get
        {
            if (this.gamaItemRepository == null)
            {
                this.gamaItemRepository = new GenericEntityRepository<GamaItem>(context);
            }
            return gamaItemRepository;
        }
    }

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

    private bool disposed = false;


    protected virtual void Dispose(bool disposing)
    {
        if (!this.disposed)
        {
            if (disposing)
            {
                context.Dispose();
            }
        }
        this.disposed = true;
    }


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

通用存储库:

public class GenericEntityRepository<TEntity> where TEntity : class
{
    internal SRColorContext context;
    internal DbSet<TEntity> dbSet;

    public GenericEntityRepository(SRColorContext context)
    {
        this.context = context;
        this.dbSet = context.Set<TEntity>();
    }

    public virtual IEnumerable<TEntity> Get(
        Expression<Func<TEntity,bool>> filter = null,
        Func<IQueryable<TEntity>, IOrderedQueryable<TEntity>> orderBy = null,
        string includeProperties = "")
    {
        IQueryable<TEntity> query = dbSet;

        if(filter != null)
        {
            query = query.Where(filter);
        }

        foreach (var includeProperty in includeProperties.Split
            (new char[] { ',' }, StringSplitOptions.RemoveEmptyEntries))
        {
            query.Include(includeProperty);
        }

        if (orderBy != null)
        {
            return orderBy(query).ToList();
        }
        else
        {
            return query.ToList();
        }

    }

    public virtual TEntity GetById(object id)
    {
         return dbSet.Find(id);
    }

    public virtual void Insert(TEntity entity)
    {
        dbSet.Add(entity);
    }

    public virtual void Delete(object id)
    {
        TEntity entityToDelete = dbSet.Find(id);
        Delete(entityToDelete);
    }

    public virtual void Delete(TEntity entityToDelete)
    {
        if (context.Entry(entityToDelete).State == EntityState.Detached)
        {
            dbSet.Attach(entityToDelete);
        }
        dbSet.Remove(entityToDelete);
    }

    public virtual void Update(TEntity entityToUpdate)
    {
        dbSet.Attach(entityToUpdate);
        context.Entry(entityToUpdate).State = EntityState.Modified;
    }
}

1 个答案:

答案 0 :(得分:0)

我认为你应该伪造上下文对象以避免调用数据库而不是实体(除非你不能简单地创建一个)

如果您想为 UnitOfWork 类编写单元测试并且您正在使用NSubtitute(或任何其他.NET OSS Mocking框架),您必须使用依赖注入并替换上下文实例用假实例。

您可能还需要包装上下文类 - 如果它已被密封或使用的方法不是虚拟的。