使用Moq和存储库模式进行单元测试

时间:2013-11-04 11:36:08

标签: c# unit-testing

我是单元测试的新手,我想要一些帮助。我首先使用存储库模式的代码。我有一个通用的存储库,其中包含称为Generic Repository的通用CRUD操作(请参阅打击)

 public abstract class GenericRepository<T> where T : class
    {
        private HolidayDatabaseContext _dataContext;
        private readonly IDbSet<T> _dbset;
        protected GenericRepository(IDbFactory databaseFactory)
        {
            DatabaseFactory = databaseFactory;
            _dbset = DataContext.Set<T>();
        }
        protected IDbFactory DatabaseFactory
        {
            get;
            private set;
        }
        protected HolidayDatabaseContext DataContext
        {
            get { return _dataContext ?? (_dataContext = DatabaseFactory.Get()); }
        }

        public virtual void Add(T entity)
        {
            _dbset.Add(entity);
        }
        public virtual void Update(T entity)
        {
            _dataContext.Entry(entity).State = EntityState.Modified;
        }
        public virtual void Delete(T entity)
        {
            _dbset.Remove(entity);
        }
        public virtual IEnumerable<T> Enumerable()
        {
            return _dbset.AsEnumerable<T>();
        }

        public virtual IQueryable<T> List()
        {
            return _dbset.AsQueryable<T>();
        }

        public virtual T GetSingleById(int id)
        {
            return _dbset.Find(id);
        }

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

    }

然后我将其继承到用户存储库并创建了一些特定的方法。见下文

public class UserRepository : GenericRepository<User>, IUserRepository
    {
        public UserRepository(IDbFactory databaseFactory)
            : base(databaseFactory) { }


        public int HolidayEntitlement(int userId)
        {
           return HolidayEntitlement(userId, DateTime.Now);
        }
        public int HolidayEntitlement(int userId, DateTime dateTime)
        {
           //Get the User
            var user = this.GetSingleById(userId);

            //Work Total Entitlement
            int entitlement = user.BaseHolidayEntitlement;

            //Years in Service
            entitlement += (dateTime - user.EmploymentStartDate).Days / 365;

            return entitlement;



        }


         public int RemainingHolidayEntitlement(int userId)
        {
            return RemainingHolidayEntitlement(userId, DateTime.Now);
        }

         public int RemainingHolidayEntitlement(int userId, DateTime dateTime)
        {
            return int.MinValue;
        }
    }

我想单元测试HolidayEntitlement(int userId,DateTime dateTime),但我需要在方法中模拟GetSingleById部分

我把它写成了一个测试,但它没有编译。

 [TestMethod]
        public void GetHolidayEntitlement25()
        {
            //How to write this Unit test


            //Setup
            var user = new User { AnnualHolidayIncrement = 1, BaseHolidayEntitlement = 25, EmploymentStartDate = new DateTime(2013, 1, 1),Id=1 };

            Mock<UserRepository> mock = new Mock<UserRepository>();
            mock.Setup(m => m.GetSingleById(1)).Returns(user);

            Assert.AreEqual(25, mock.Object.HolidayEntitlement(1));
        }

任何帮助将不胜感激

2 个答案:

答案 0 :(得分:13)

你似乎在说你只想模仿部分界面。当你开始遇到这种情况时,它表明你正在混淆你的担忧,并可能在某处做错事。

在这种情况下,您的存储库正在执行 MUCH ,而不仅仅是CRUD,因此具有多个职责(它应该只有一个,查找SOLID编程)。您正在存储库中执行业务逻辑,它不应该存在!除了简单的CRUD操作之外的任何操作都应该移到业务逻辑层中。即你的HolidayEntitlement方法通过应用一些逻辑来计算某些东西,因此不是CRUD /存储库操作!

所以......你应该做的是将业务逻辑位移到一个新类中,比如UserLogic。在UserLogic类中,您将使用注入的IUserRepository与存储库进行通信。在UserLogic您可以使用HolidayEntitlement方法进行IUserRepository.GetSingleById调用UserLogic。所以,当你测试你的IUserRepository课时,你会注入你的模拟GetSingleById,它会有IUserRepository的存根版本,然后你会回到正确的轨道上!

我希望这有意义/有帮助吗?!

- 原始发布 -

P.S。我的原帖说明你应该模拟接口,而不是实例,所以这仍然适用,我将离开这里作为参考:

你应该嘲笑UserRepository UserRepository

这是因为IUserRepositoryUserRepository的实现。你想说你给它一个新的实现,即你的模拟。目前您正在使用 ACTUAL 类{{1}}。

答案 1 :(得分:5)

当您需要提供假的依赖时,通常会使用模拟,在这种情况下,您似乎正在尝试模拟被测系统(SUT),这实际上没有意义 - 确实如此没有意义,因为你的测试实际上没有告诉你UserRepository的行为;你正在做的就是测试你是否正确设置了Mock,这不是很有用!

您提供的测试代码似乎表明您要测试UserRepository.HolidayEntitlement

我更倾向于将这样的函数从Repository类中移出到一个单独的业务逻辑类型中。通过这种方式,您可以测试在完全隔离中计算用户假期权利的逻辑,这是单元测试的主要原则。

为了测试此函数是否执行了它应该执行的操作(即根据User的属性执行计算),您需要确保在其中操作的任何User实例函数是100%隔离并且在你的控制下 - 使用用户的Mock或Fake(Stub)实例,在这种情况下,Mocks是一个很好的选择,因为你只需要实现你的SUT将需要的依赖部分。

所以,你能做的就是:

为用户定义界面

public interface IUser
{
  int BaseHolidayEntitlement{get;set;}
  DateTime EmploymentStartDate {get;set;}
  //other properties for a User here
}

在您的用户类

上实现此功能
public class User:IUser
{
   //implemement your properties here
   public int BaseHolidayEntitlement{get;set;}
   public DateTime EmploymentStartDate {get;set;}
   //and so on
}

为用户逻辑创建一个类

public class UserRules
{
  public int GetHolidayEntitlement(IUser user,DateTime dateTime)
  {
     //perform your logic here and return the result
  }
}

现在您的测试变得更加简单,甚至不需要存储库

 [TestMethod]
 public void GetHolidayEntitlement_WithBase25_Returns25()
 {
    //Arrange
    var user = new Mock<IUser>();
    //setup known, controlled property values on the mock:
    user.SetupGet(u=>u.BaseHolidayEntitlement).Returns(25);
    user.SetupGet(u=>u.EmploymentStartDate).Returns(new DateTime(2013,1,1));
    var sut = new UserRules();
    int expected = 25;
    //Act
    int actual = sut.GetHolidayEntitlement(user.Object,DateTime.UtcNow);
    //Assert
    Assert.AreEqual(expected,actual,"GetHolidayEntitlement isn't working right...");
 }