如何在spring JdbcTemplate中模拟数据源

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

import javax.sql.DataSource;

import static org.mockito.Mockito.*;

public class JdbcTemplateTest {

    @Mock
    private DataSource mockDataSource;

    @InjectMocks
    private JdbcTemplate jdbcTemplate;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        jdbcTemplate.setDataSource(mockDataSource);
    }

    @Test
    public void testQuery() {
        // 模拟数据源返回的结果
        String expectedValue = "mocked value";
        when(mockDataSource.getConnection()).thenReturn(null);
        when(jdbcTemplate.queryForObject(anyString(), any(), any())).thenReturn(expectedValue);

        // 调用被测试类的方法
        String result = jdbcTemplate.queryForObject("SELECT * FROM table", String.class);

        // 验证结果是否符合预期
        assertEquals(expectedValue, result);

        // 验证模拟对象的方法是否被调用
        verify(mockDataSource, times(1)).getConnection();
        verify(jdbcTemplate, times(1)).queryForObject(anyString(), any(), any());
    }
}