模拟Spring jdbc template For RowMapper的单元测试用例

import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.List;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.*;

public class JdbcTemplateTest {

    @Mock
    private JdbcTemplate jdbcTemplate;

    @Before
    public void setup() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testQueryForObject() {
        String sql = "SELECT * FROM users WHERE id = ?";
        Object[] params = {1};

        User expectedUser = new User(1, "John Doe");

        when(jdbcTemplate.queryForObject(eq(sql), eq(params), any(RowMapper.class))).thenReturn(expectedUser);

        User actualUser = jdbcTemplate.queryForObject(sql, params, new UserRowMapper());

        assertEquals(expectedUser, actualUser);
        verify(jdbcTemplate, times(1)).queryForObject(eq(sql), eq(params), any(RowMapper.class));
    }

    @Test
    public void testQuery() {
        String sql = "SELECT * FROM users";

        List<User> expectedUsers = List.of(
                new User(1, "John Doe"),
                new User(2, "Jane Smith")
        );

        when(jdbcTemplate.query(eq(sql), any(RowMapper.class))).thenReturn(expectedUsers);

        List<User> actualUsers = jdbcTemplate.query(sql, new UserRowMapper());

        assertEquals(expectedUsers, actualUsers);
        verify(jdbcTemplate, times(1)).query(eq(sql), any(RowMapper.class));
    }

    private static class User {
        private int id;
        private String name;

        public User(int id, String name) {
            this.id = id;
            this.name = name;
        }

        // Getters and setters
    }

    private static class UserRowMapper implements RowMapper<User> {
        @Override
        public User mapRow(ResultSet rs, int rowNum) throws SQLException {
            int id = rs.getInt("id");
            String name = rs.getString("name");
            return new User(id, name);
        }
    }
}