Spring Boot Kotlin项目用Kotest扩展自动重置Flyway数据库

方案一：自动注册JUnit 5扩展（推荐） #

这种方式无需修改现有测试类，就能让所有integrationTest目录下的测试自动应用Flyway清理迁移逻辑，同时通过Spring上下文注入Flyway实例。

1. 改造监听逻辑为JUnit 5 Extension #

将你的FlywayCleaningTestListener改为实现JUnit 5的BeforeEachCallback接口，借助SpringExtension获取Spring上下文里的Flyway Bean：

import org.junit.jupiter.api.extension.BeforeEachCallback
import org.junit.jupiter.api.extension.ExtensionContext
import org.springframework.test.context.junit.jupiter.SpringExtension
import org.flywaydb.core.Flyway

class FlywayCleaningExtension : BeforeEachCallback {
    override fun beforeEach(context: ExtensionContext) {
        // 从Spring测试上下文获取Flyway实例
        val applicationContext = SpringExtension.getApplicationContext(context)
        val flyway = applicationContext.getBean(Flyway::class.java)
        
        // 执行清理+迁移
        flyway.clean()
        flyway.migrate()
    }
}

2. 自动注册扩展到所有集成测试 #

在src/integrationTest/resources/META-INF/services/目录下创建文件org.junit.jupiter.api.extension.Extension，文件内容为你的扩展类全限定名：

com.your.project.path.FlywayCleaningExtension

JUnit 5会自动扫描这个文件，将扩展应用到所有integrationTest目录下的测试类。

方案二：抽象基类实现 #

如果偏好更直观的控制方式，可创建抽象基类，让所有集成测试继承它：

1. 编写抽象基类 #

在基类中注入Flyway，并在@BeforeEach方法中执行清理迁移逻辑：

import org.junit.jupiter.api.BeforeEach
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.boot.test.context.SpringBootTest
import org.flywaydb.core.Flyway

// 根据你的测试需求调整@SpringBootTest的参数，比如指定配置类、web环境等
@SpringBootTest
abstract class BaseIntegrationTest {

    @Autowired
    lateinit var flyway: Flyway

    @BeforeEach
    fun setupDatabase() {
        flyway.clean()
        flyway.migrate()
    }
}

2. 集成测试类继承基类 #

所有integrationTest下的测试类只需继承该基类，即可自动获得数据库初始化逻辑：

class UserRepositoryIntegrationTest : BaseIntegrationTest() {
    // 测试代码，无需重复写数据库初始化逻辑
}

注意事项 #

• 确保integrationTest的Spring配置文件（如application-integrationtest.yml）中正确配置了Flyway的数据库连接信息，避免权限或连接问题。
• Gradle的integrationTest source set需正确引入依赖：spring-boot-starter-test、flyway-core、junit-jupiter-api、junit-jupiter-engine等。

内容的提问来源于stack exchange，提问作者Rüdiger Schulz