Jdbc模板的Mockito测试用例和spring的keyHolder

Question

我编写了一个测试用例，但在return keyholder.getKey()行得到一个空指针异常。

我的测试用例如下所示：

@InjectMocks
private UserDAOImpl userDAO;

@Mock
private JdbcTemplate jdbcTemplate;

@Mock
private KeyHolderFactory keyHolderFactory;

@Before
public void setup() {
    MockitoAnnotations.initMocks(this);
    ReflectionTestUtils.setField(userDAO, "jdbcTemplate", jdbcTemplate);
}

@Test
public void testCreateUser() {

    KeyHolder keyHolder = mock(GeneratedKeyHolder.class);

    when(keyHolderFactory.newKeyHolder()).thenReturn(keyHolder);

    User user = getUserInfo();

    Map<String,Object> map = new HashMap<>();
    map.put("id",1L);
    when(keyHolder.getKeys()).thenReturn(map);
    when(keyHolder.getKey()).thenReturn(1L);

    when(jdbcTemplate.update(Mockito.any(PreparedStatementCreator.class), Mockito.any(KeyHolder.class))).thenReturn(1);

    assertEquals(1L, userDAO.createUser(user));
}

方法如下所示：

 @Override
  public long createUser(User user) {

KeyHolder keyHolder = new GeneratedKeyHolder();

jdbcTemplate.update(connection -> {
  PreparedStatement ps = connection
      .prepareStatement("insert into user (address, first_name,last_name, email, password, phone_number, is_init, is_system_admin, created_at)"
              + " values( ?, ?, ?, ?, ?, ?, ?, ?,?)",
          Statement.RETURN_GENERATED_KEYS);
  ps.setString(1, user.getAddress());
  ps.setString(2, user.getFirstName());
  ps.setString(3, user.getLastName());
  ps.setString(4, user.getEmail());
  ps.setString(5, user.getPassword());
  ps.setString(6, user.getPhone());
  ps.setBoolean(7, user.isActive());
  ps.setBoolean(8, user.isSystemAdmin());
  ps.setDate(9, new Date(Date.from((user.getCreatedAt().toInstant())).getTime()));

  return ps;
}, keyHolder);

return (long) keyHolder.getKey();
  }

我创建了一个接口KeyHolderFactory:

public interface KeyHolderFactory {
KeyHolder newKeyHolder();
}

并且这个接口的实现如下：

public class GeneratedKeyHolderFactory implements KeyHolderFactory {

public KeyHolder newKeyHolder() {
    return new GeneratedKeyHolder();
  }
}

有人能帮我找到解决办法吗？

Answer 1

接下来是您的问题：

在测试类中，您已经嘲弄了binkeyholderFactory，这是正确的。然后创建模拟的keyholder，描述它的行为，并使keyholderFactory在被询问时返回keyholder。除了下一个：

在代码类方法中，使用new GeneratedKeyHolder()创建新的keyholder并且这个新的keyholder与测试类中的模拟实例不相关。它是绝对不同的对象。在您的测试类中的一个是模拟的，将遵循您的场景。但是，在您的测试类方法中，这个对象是另一个对象，当您执行keyholder.getKey()时，它当然不知道该做什么。

解决方案：

您已经在测试中注入了keyholderFactory并很好地配置了它的行为。只需将其注入测试类，并在方法中使用KeyHolderFactory.NewKeyHolder()而不是New GeneratedKeyHolder()创建KeyHolder