如何moq简单添加使用工作单元和存储库模式的功能

时间:2015-01-26 17:00:18

标签: c# unit-testing moq

我的测试看起来像这样

        [Fact]
    public void SimpleAddTest()
    {
        // Arrange
        var authorizationsToBeAdded = new List<PatientPayerAuthInfo>
        {
            new PatientPayerAuthInfo (),
            new PatientPayerAuthInfo  ()
        }.ToList();
        var persistentAuthorizations = new List<PatientPayerAuthInfo>
        {
            new PatientPayerAuthInfo {PatientPayerAuthInfoId = 1 },
            new PatientPayerAuthInfo  {PatientPayerAuthInfoId = 2 },
             new PatientPayerAuthInfo  {PatientPayerAuthInfoId = 3 },
             new PatientPayerAuthInfo  {PatientPayerAuthInfoId = 4 }
        }.AsQueryable();

        var mockSet = new Mock<DbSet<PatientPayerAuthInfo>>();
        mockSet.As<IQueryable<PatientPayerAuthInfo>>().Setup(m => m.Provider).Returns(persistentAuthorizations.Provider);
        mockSet.As<IQueryable<PatientPayerAuthInfo>>().Setup(m => m.Expression).Returns(persistentAuthorizations.Expression);
        mockSet.As<IQueryable<PatientPayerAuthInfo>>().Setup(m => m.ElementType).Returns(persistentAuthorizations.ElementType);
        mockSet.As<IQueryable<PatientPayerAuthInfo>>().Setup(m => m.GetEnumerator()).Returns(persistentAuthorizations.GetEnumerator());

        var mockedUnitOfWork = new Mock<IUOW<DBContext>>();     

        var service = new PatientPayerService(mockedUnitOfWork.Object);

        // Act
        var sut = service.AddPatientPayerAuthInfos(authorizationsToBeAdded);

        // Assert

    }

服务层功能如下所示

        public void AddPatientPayerAuthInfos(IEnumerable<PatientPayerAuthInfo> patientPayerAuthInfos)
    {
        foreach (var patientPayerAuthInfo in patientPayerAuthInfos)
        {
            UOW.PatientPayerAuthInfos.Add(patientPayerAuthInfo);
        }
        UOW.SaveChanges();
    }

AND Repository实现

     public virtual void Add(T entity)
    {
        DbEntityEntry dbEntityEntry = DbContext.Entry(entity);
        if (dbEntityEntry.State != EntityState.Detached)
        {
            dbEntityEntry.State = EntityState.Added;
        }
        else
        {
            DbSet.Add(entity);
        }
    }

然后有一个调用EF的SaveChanges的提交方法。

我的问题是如何使用persistentAuthorizations设置mockedUnitOfWork,这样当我使用authorizationsToBeAdded添加两个对象时,persistentAuthorizations的总数将变为6,最初为4。

如果我走错了路,请纠正我。

  public interface IRepository<T> where T : class
  {
    void Add(T entity);
   }

public interface IUOW<U> where U : DbContext, IDisposable
{        
    IRepository<PatientPayerAuthInfo> PatientPayerAuthInfos { get; }
    void SaveChanges();
}

1 个答案:

答案 0 :(得分:2)

使用列表作为persistentAuthorizations的基础,例如:

var data = new List<PatientPayerAuthInfo>
{
    new PatientPayerAuthInfo {PatientPayerAuthInfoId = 1 },
    new PatientPayerAuthInfo {PatientPayerAuthInfoId = 2 },
    new PatientPayerAuthInfo {PatientPayerAuthInfoId = 3 },
    new PatientPayerAuthInfo {PatientPayerAuthInfoId = 4 }
};
var persistentAuthorizations = data.AsQueryable();

然后您可以像这样设置mockedUnitOfWork

var repositoy = new Mock<IRepository<PatientPayerAuthInfo>>();

// when adding data to the repository, add the item to 'data' 
repositoy.Setup(r => r.Add(It.IsAny<PatientPayerAuthInfo>()))
         .Callback(delegate(PatientPayerAuthInfo y)
                   {
                        data.Add(y);
                   });

// when accessing 'PatientPayerAuthInfos', use the repository mock
var mockedUnitOfWork = new Mock<IUOW<DBContext>>();
mockedUnitOfWork.SetupGet(x => x.PatientPayerAuthInfos).Returns(() => repositoy.Object);