Spring Boot JUnit测试中如何启用基于配置属性的DataSource自动配置？

要在带有自定义Spring上下文的JUnit测试中，通过spring.datasource.*属性启用JDBC数据源的自动配置，按以下步骤操作：

1. 确保测试类启用自动配置机制 #

如果你的测试类通过@ContextConfiguration加载自定义配置类，需要额外添加@EnableAutoConfiguration注解触发Spring Boot的自动配置逻辑；或者直接使用@SpringBootTest（它默认包含自动配置能力），同时指定你的自定义配置类。

示例1：结合@SpringBootTest使用 #

@SpringBootTest(classes = {CustomConfig.class}) // 指定你的自定义配置类
@TestPropertySource(properties = {
    "spring.datasource.url=jdbc:mysql://localhost:3306/test_db",
    "spring.datasource.username=db_user",
    "spring.datasource.password=db_pass",
    "spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver"
})
public class DataSourceAutoConfigTest {

    @Autowired
    private DataSource dataSource;

    @Test
    void verifyDataSourceInitialization() {
        // 验证数据源是否成功初始化
        assertNotNull(dataSource);
    }
}

示例2：结合@ContextConfiguration使用 #

@ContextConfiguration(classes = {CustomConfig.class})
@EnableAutoConfiguration // 启用自动配置
@TestPropertySource(properties = {
    "spring.datasource.url=jdbc:h2:mem:testdb",
    "spring.datasource.username=sa",
    "spring.datasource.password="
})
public class CustomContextDataSourceTest {

    @Autowired
    private DataSource dataSource;

    @Test
    void testDataSourceConnection() throws SQLException {
        // 测试数据源连接有效性
        assertTrue(dataSource.getConnection().isValid(1));
    }
}

2. 配置数据源属性的两种方式 #

除了在@TestPropertySource中直接硬编码属性，还可以使用测试专用的配置文件：

• 在src/test/resources下创建application-test.properties文件：

spring.datasource.url=jdbc:postgresql://localhost:5432/test_db
spring.datasource.username=postgres
spring.datasource.password=postgres
spring.datasource.driver-class-name=org.postgresql.Driver

• 在测试类上添加@ActiveProfiles("test")激活该配置文件：

@SpringBootTest(classes = {CustomConfig.class})
@ActiveProfiles("test")
public class ProfileBasedDataSourceTest {
    // 测试逻辑
}

3. 避免配置冲突 #

如果你的自定义配置类中手动定义了DataSource Bean，Spring Boot的自动配置会被覆盖。若要启用自动配置，请移除自定义的DataSource Bean定义，让Spring根据spring.datasource.*属性自动创建数据源。

注意事项 #

• 确保项目依赖中包含spring-boot-starter-jdbc或对应数据库的starter依赖（如spring-boot-starter-data-jpa），否则自动配置逻辑不会触发。

内容的提问来源于stack exchange，提问作者Harold L. Brown