如何在Mockito中使用ArgumentCaptor优化JDBC插入方法的单元测试？

嘿，我懂你想靠ArgumentCaptor把单元测试质量提上去的需求，针对你这个JDBC插入用户的方法，咱们可以一步步写出靠谱的测试——既不用连真实数据库，还能把参数和SQL逻辑都验证到位。

核心思路 #

你的insert方法用了匿名内部类PreparedStatementCreator，直接mockJdbcTemplate的话没法直接验证里面的参数设置逻辑，这时候ArgumentCaptor就派上用场了：它能帮咱们捕获这个PreparedStatementCreator对象，然后手动触发它的方法，验证里面的细节。

完整测试步骤和代码示例 #

首先，先确保你的测试类配置好Mockito环境，然后按下面的步骤来：

1. 初始化Mockito依赖
   用@ExtendWith(MockitoExtension.class)启用Mockito支持，给JdbcTemplate加@Mock注解生成mock实例，给你的DAO类（假设insert方法在UserDao里）加@InjectMocks，让Mockito自动注入依赖。

2. 捕获PreparedStatementCreator参数
   定义ArgumentCaptor来捕获传给jdbcTemplate.update()的PreparedStatementCreator实例，这样就能拿到这个匿名内部类对象。

3. 模拟JDBC对象并验证逻辑
   mock出Connection和PreparedStatement，调用捕获到的PreparedStatementCreator的createPreparedStatement方法，然后验证参数设置和SQL语句是否正确。

下面是完整的测试代码：

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.PreparedStatementCreator;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;

import static org.mockito.Mockito.*;

@ExtendWith(MockitoExtension.class)
class UserDaoTest {

    @Mock
    private JdbcTemplate jdbcTemplate;

    @InjectMocks
    private UserDao userDao; // 替换成你实际的DAO类名

    @Test
    void insert_shouldSetCorrectUserIdAndValidSql() throws SQLException {
        // 准备测试用的用户对象
        User testUser = new User("user_001");

        // 调用要测试的insert方法
        userDao.insert(testUser);

        // 捕获PreparedStatementCreator参数
        ArgumentCaptor<PreparedStatementCreator> psCreatorCaptor = ArgumentCaptor.forClass(PreparedStatementCreator.class);
        verify(jdbcTemplate).update(psCreatorCaptor.capture());

        // mock JDBC的Connection和PreparedStatement
        Connection mockConn = mock(Connection.class);
        PreparedStatement mockPs = mock(PreparedStatement.class);
        when(mockConn.prepareStatement(anyString())).thenReturn(mockPs);

        // 触发捕获到的PreparedStatementCreator的逻辑
        psCreatorCaptor.getValue().createPreparedStatement(mockConn);

        // 验证参数设置是否正确
        verify(mockPs).setString(1, testUser.getId());
        // 验证SQL语句是否符合预期
        verify(mockConn).prepareStatement("INSERT INTO user (id) VALUES ?");
    }
}

关键细节解释 #

• 为什么用ArgumentCaptor？ 匿名内部类的逻辑是封装在方法里的，没法直接mock或验证，通过捕获这个对象，咱们能深入到内部逻辑里，确保参数和SQL都没写错。
• 完全脱离真实数据库：所有JDBC相关对象都是mock的，测试跑起来飞快，也不会污染测试数据。
• 双重验证：既确认了JdbcTemplate.update()被正确调用，又验证了SQL语句和参数的正确性，测试覆盖更全面，不会漏掉关键逻辑。

要是你的SQL语句更复杂（比如多参数），只需要多加几个verify(mockPs).setXXX()的断言就行，非常灵活。

内容的提问来源于stack exchange，提问作者Dixon Ivey