Mockito用模拟对象测试DAO

时间:2014-01-30 17:36:11

标签: java junit mocking mockito dao

我想测试这个DAO方法

//in GrabDao.java
public WrapperProject getChildren(Integer entityId, String entityType){

    EntityDao entityDao = new EntityDao();
    UserDao userDao = new UserDao();

    EntityBase entity = entityDao.getEntityById(entityId, entityType);
    Date dateProjet = userDao.getOrganismeFromSession().getExercice().getDateProjet();

    return new Wrapper(dateProjet, entity);
}

这是我到目前为止所尝试的

    //in GrabDaoTest.java
    Integer paramEntityId = 1; 
    String paramEntityType = "type";

    EntityBase entityBase = Mockito.mock(EntityBase.class);

    EntityDao entityDao = Mockito.mock(EntityDao.class);
    when(entityDao.getEntityById(paramEntityId, paramEntityType)).thenReturn(entityBase);

    UserDao userDao = Mockito.mock(UserDao.class);
    Organisme organisme = Mockito.mock(Organisme.class);
    Exercice excercice = Mockito.mock(Exercice.class);

    when(userDao.getOrganismeFromSession()).thenReturn(organisme);
    when(organisme.getExercice()).thenReturn(excercice);
    when(userDao.getOrganismeFromSession().getExercice().getDateProjet()).thenReturn(new GregorianCalendar(2000, 01, 01).getTime());

现在我想测试带有假参数 paramEntityId getChildren paramEntityType 将正确返回WrapperProject 1和01/01 / 2000 使用模拟方法但我无法弄清楚如何启动read方法告诉她使用模拟dao

1 个答案:

答案 0 :(得分:4)

您的代码不是测试友好的,特别是这两行对测试非常不利:

EntityDao entityDao = new EntityDao();
UserDao userDao = new UserDao();

此代码应从此方法移至Factory或使用类似Spring(依赖注入)的内容注入。

单凭Mockito无法测试这样的代码。你的方法应该只做一件事,创建Daos是其他工作。

我会向你推荐两部来自GoogleTechTalks的电影: