使用Java8编写的DAO类方法的Mockito测试用例

Question

public class DBConnectionDAOImpl {

public List<String> getData{
    return jdbcTemplate.query("Select abc from table_name ", (ResultSet rs) -> getAllData(rs));
}

public List<String> getAllData(ResultSet rs){
    List<String> ls = new ArrayList<>();
    try{
        while(rs.next()){
            ls.add(rs.getString("pk"));
        }
        return ls;
    }catch(Exception e){


    }
}

}

public class DBConnectionDAOImplTest {

@InjectMock
DBConnectionDAOImpl dbConnectionDAOImpl;

@MOck
JdbcTemplate jdbcTemplate;

@Test
public void getDataTest(){
    ResultSet rs = Mockito.mock(ResultSet.class);
    when(rs.next().thenReturn(true).thenReturn(false));
    when(rs.getString("pk")).thenReturn("someVal");
    when(dbConnectionDAOImpl.getAllData(rs)).thenReturn(new ArrayList<String>());
    when(jdbcTemplate.query("Select abc from table_name ", (ResultSet rs) -> dbConnectionDAOImpl.getAllData(rs))).thenReturn(new ArrayList<String>());

    List<String> result = dbConnectionDAOImpl.getData();
    assertNotNull(result);
}

when(dbConnectionDAOImpl.getAllData(rs)).thenReturn(new ArrayList<String>());

---编辑1--更改

 when(dbConnectionDAOImpl.getAllData(rs)).thenReturn(new ArrayList<String>());

根据注释，断言是失败的。

Answer 1

在我的评论中提供的链接中，尝试更改：

when(dbConnectionDAOImpl.getAllData(rs)).thenReturn(new ArrayList<String>());

到

doReturn(new ArrayList<String>()).when(dbConnectionDAOImpl).getAllData(rs)