Spring Boot多数据库集成测试：SpringRunner适配不同SQL脚本方法咨询

当然可以！用SpringRunner做测试时，完全能实现针对MySQL、Oracle等不同数据库执行专属SQL脚本的需求，解决你现在遇到的通用脚本跨库不兼容问题。下面给你分享几个实用的实现方案：

方案1：利用@Sql的config属性自定义数据库适配逻辑 #

你可以通过自定义SqlConfig实现类，根据当前测试的数据库类型自动选择对应的脚本：

首先创建一个数据库专属的配置器：

public class DatabaseSpecificSqlConfig implements SqlConfig {
    @Override
    public DataSource dataSource() {
        // 获取当前测试上下文的数据源
        return TestContextManager.getCurrentTestContext().getApplicationContext().getBean(DataSource.class);
    }

    @Override
    public String[] scripts() {
        DataSource dataSource = dataSource();
        String dbType = getDatabaseType(dataSource);
        
        // 根据数据库类型返回对应脚本路径
        if ("ORACLE".equals(dbType)) {
            return new String[]{"classpath:tenantBeforeTestRun_oracle.sql"};
        } else if ("MYSQL".equals(dbType)) {
            return new String[]{"classpath:tenantBeforeTestRun_mysql.sql"};
        }
        // 兜底的默认脚本
        return new String[]{"classpath:tenantBeforeTestRun.sql"};
    }

    private String getDatabaseType(DataSource dataSource) {
        try (Connection conn = dataSource.getConnection()) {
            String productName = conn.getMetaData().getDatabaseProductName().toUpperCase();
            if (productName.contains("ORACLE")) {
                return "ORACLE";
            } else if (productName.contains("MYSQL")) {
                return "MYSQL";
            }
        } catch (SQLException e) {
            throw new RuntimeException("无法识别数据库类型", e);
        }
        return "UNKNOWN";
    }
}

然后在测试类中引用这个配置器：

@SqlGroup({
    @Sql(executionPhase = Sql.ExecutionPhase.BEFORE_TEST_METHOD, 
         config = @SqlConfig(configurer = DatabaseSpecificSqlConfig.class))
})
@RunWith(SpringRunner.class)
@SpringBootTest
public class TenantTest {
    // 你的测试方法
}

方案2：结合Spring Profiles实现动态脚本加载 #

这种方式更简洁，利用Spring的Profile机制，通过激活不同Profile来加载对应数据库的脚本：

首先在测试类中指定激活的Profile（也可以通过Maven/Gradle参数动态切换）：

@RunWith(SpringRunner.class)
@SpringBootTest
@ActiveProfiles("mysql") // 切换成"oracle"即可执行Oracle专属脚本
public class TenantTest {
    @Sql(executionPhase = Sql.ExecutionPhase.BEFORE_TEST_METHOD, 
         scripts = "classpath:tenantBeforeTestRun_${spring.profiles.active}.sql")
    @Test
    public void testTenantOperation() {
        // 测试逻辑
    }
}

然后创建两个对应脚本文件：tenantBeforeTestRun_mysql.sql（MySQL专属）和tenantBeforeTestRun_oracle.sql（Oracle专属）即可。

方案3：自定义TestExecutionListener实现更灵活的控制 #

如果需要更复杂的执行逻辑，可以自定义测试执行监听器，在测试方法前后手动执行对应脚本：

public class DatabaseSqlListener extends AbstractTestExecutionListener {
    @Override
    public void beforeTestMethod(TestContext testContext) throws Exception {
        DataSource dataSource = testContext.getApplicationContext().getBean(DataSource.class);
        String dbType = getDatabaseType(dataSource);
        String scriptPath = "classpath:tenantBeforeTestRun_" + dbType.toLowerCase() + ".sql";
        
        // 使用Spring的ScriptUtils执行SQL脚本
        ScriptUtils.executeSqlScript(dataSource.getConnection(), new ClassPathResource(scriptPath));
    }

    private String getDatabaseType(DataSource dataSource) throws SQLException {
        try (Connection conn = dataSource.getConnection()) {
            String productName = conn.getMetaData().getDatabaseProductName().toUpperCase();
            if (productName.contains("ORACLE")) {
                return "ORACLE";
            } else if (productName.contains("MYSQL")) {
                return "MYSQL";
            }
        }
        throw new IllegalArgumentException("不支持当前数据库类型");
    }
}

然后在测试类中注册这个监听器：

@RunWith(SpringRunner.class)
@SpringBootTest
@TestExecutionListeners(listeners = DatabaseSqlListener.class, mergeMode = MergeMode.MERGE_WITH_DEFAULTS)
public class TenantTest {
    // 你的测试方法
}

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