Spring Boot：使用@AutoConfigureTestDatabase时自定义H2数据库参数

要实现在每个测试中创建独立的自定义配置H2内存数据库，同时避免重复执行脚本导致的冲突，按以下步骤操作：

1. 结合@AutoConfigureTestDatabase与动态数据源配置 #

在测试类上保留@AutoConfigureTestDatabase，设置replace = AutoConfigureTestDatabase.Replace.NONE，同时通过@DynamicPropertySource动态生成唯一的H2数据库URL，以此确保每个测试实例使用独立的内存库，同时在URL中指定H2兼容模式。

示例代码：

import org.junit.jupiter.api.Test;
import org.springframework.boot.test.autoconfigure.jdbc.AutoConfigureTestDatabase;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.DynamicPropertyRegistry;
import org.springframework.test.context.DynamicPropertySource;

import java.util.UUID;

@SpringBootTest
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)
public class CustomH2IntegrationTest {

    @DynamicPropertySource
    static void setupDataSource(DynamicPropertyRegistry registry) {
        // 生成唯一数据库名，保证每个测试的内存库独立
        String uniqueDbName = "test_db_" + UUID.randomUUID();
        // 自定义H2 URL：添加兼容模式（示例为MySQL兼容）、关闭自动销毁等参数
        String h2Url = String.format(
            "jdbc:h2:mem:%s;DB_CLOSE_DELAY=-1;MODE=MYSQL;DB_CLOSE_ON_EXIT=FALSE",
            uniqueDbName
        );
        registry.add("spring.datasource.url", () -> h2Url);
    }

    @Test
    void testBusinessLogic1() {
        // 你的测试逻辑
    }

    @Test
    void testBusinessLogic2() {
        // 你的测试逻辑
    }
}

2. 关键配置说明 #

• 唯一数据库名：通过UUID生成的唯一名称，确保每个测试启动时创建全新的内存H2数据库，彻底避免多测试共享数据库导致的脚本重复执行冲突。
• H2兼容模式：在URL中通过MODE=XXX指定（比如MODE=MYSQL、MODE=ORACLE），直接实现H2的兼容模式自定义。
• DB_CLOSE_DELAY=-1：禁止H2在最后一个连接关闭时销毁内存数据库，保证测试过程中数据库可用；DB_CLOSE_ON_EXIT=FALSE避免JVM退出时自动关闭数据库。

3. JUnit 4兼容方案（若使用旧版本） #

如果项目仍在使用JUnit 4，无法使用@DynamicPropertySource，可以通过@TestPropertySource结合自定义数据源配置类实现：

import org.junit.Test;
import org.springframework.boot.test.autoconfigure.jdbc.AutoConfigureTestDatabase;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.TestPropertySource;

@SpringBootTest
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)
@TestPropertySource(properties = {
    "spring.datasource.url=jdbc:h2:mem:test_db_${random.uuid};DB_CLOSE_DELAY=-1;MODE=MYSQL;DB_CLOSE_ON_EXIT=FALSE"
})
public class JUnit4CustomH2Test {

    @Test
    void testLogic() {
        // 测试逻辑
    }
}

这里利用Spring的${random.uuid}占位符生成唯一数据库名，同样能实现每个测试使用独立内存库的效果。

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