如何在 Junit/Mockito 中覆盖以下所有行

Question

我尝试模拟 JdbcTemplate jdbcTemplate ，但这并没有涵盖里面的任何内容

新员工(.......);

请告诉我有没有办法覆盖 new Employee(...) 中的这些行？

public List<Employee> findByCustIdAndType(long Id, String type) 
    {
        return jdbcTemplate.query(SQL.getEmployee(Id, type),
                (rs, rowNum) -> new Employee(rs.getLong("CUSTOMER_ID"), 
                                             rs.getLong("ANCHOR_CUSTOMER_ID") ,
                                             rs.getString("SEGMENT"), 
                                             rs.getDate("END_TS")));
    }

Answer 1

尝试使用 Mockito 捕获 lambda，即 RowMapper<Employee>。然后使用模拟ResultSet 调用它以返回预期值，以便可以断言返回的Employee。这是一个例子：

@RunWith(MockitoJUnitRunner.class)
public class EmployeeDAOTest {
    private static final long CUSTOMER_ID = 1;
    private static final long ANCHOR_CUSTOMER_ID = 2;
    private static final String SEGMENT = "A";
    private static final Date END_TS = Date.valueOf(LocalDate.now());

    @InjectMocks
    private EmployeeDAO dao;

    @Mock   
    private JdbcTemplate jdbcTemplate;

    @Mock   
    private ResultSet resultSet;

    @Captor
    private ArgumentCaptor<RowMapper<Employee>> rowMapperCaptor;

    @Before
    public void prepareTest() throws SQLException {
        when(resultSet.getLong("CUSTOMER_ID")).thenReturn(CUSTOMER_ID);
        when(resultSet.getLong("ANCHOR_CUSTOMER_ID")).thenReturn(ANCHOR_CUSTOMER_ID);
        when(resultSet.getString("SEGMENT")).thenReturn(SEGMENT);
        when(resultSet.getDate("END_TS")).thenReturn(END_TS);
    }

    @Test
    public void test() throws SQLException {
        dao.findByCustIdAndType(0, null);

        verify(jdbcTemplate).query(anyString(), rowMapperCaptor.capture());
        RowMapper<Employee> rowMapper = rowMapperCaptor.getValue();
        Employee employee = rowMapper.mapRow(resultSet, 1);
        assertEquals(CUSTOMER_ID, employee.getCustomerId());
        assertEquals(ANCHOR_CUSTOMER_ID, employee.getAnchorCustomerId());
        assertEquals(SEGMENT, employee.getSegment());
        assertEquals(END_TS, employee.getEndTs());
    }

}