如何使用Mockito模拟Spring的JdbcTemplate.queryForList？

// Mockito依赖
testImplementation 'org.mockito:mockito-core:3.12.4'

// Spring的JdbcTemplate依赖
implementation 'org.springframework.boot:spring-boot-starter-jdbc:2.5.4'

@RunWith(MockitoJUnitRunner.class)
public class JdbcTemplateTest {

    @InjectMocks
    private JdbcTemplate jdbcTemplate;

    @Mock
    private DataSource dataSource;

    @Test
    public void testQueryForList() {
        // 创建模拟的结果集
        ResultSet resultSet = Mockito.mock(ResultSet.class);
        Mockito.when(resultSet.next()).thenReturn(true, true, false); // 模拟结果集的next方法返回值
        Mockito.when(resultSet.getString("column1")).thenReturn("value1", "value2"); // 模拟结果集的getString方法返回值

        // 创建模拟的PreparedStatement
        PreparedStatement preparedStatement = Mockito.mock(PreparedStatement.class);
        Mockito.when(preparedStatement.executeQuery()).thenReturn(resultSet); // 模拟PreparedStatement的executeQuery方法返回结果集

        // 创建模拟的Connection
        Connection connection = Mockito.mock(Connection.class);
        Mockito.when(connection.prepareStatement(Mockito.anyString())).thenReturn(preparedStatement); // 模拟Connection的prepareStatement方法返回PreparedStatement

        // 设置模拟的DataSource返回模拟的Connection
        Mockito.when(dataSource.getConnection()).thenReturn(connection);

        // 调用JdbcTemplate的queryForList方法进行测试
        List<Map<String, Object>> result = jdbcTemplate.queryForList("SELECT * FROM table");

        // 验证结果
        Assert.assertEquals(2, result.size());
        Assert.assertEquals("value1", result.get(0).get("column1"));
        Assert.assertEquals("value2", result.get(1).get("column1"));
    }
}