如何在每个测试类后执行Flyway Clean和Migrate？静态方法依赖方案咨询

我来分享几个可行的方案，帮你搞定在测试类级别而非方法级别执行Flyway的clean()和migrate()操作的问题：

方案1：使用JUnit 5的TestInstance.Lifecycle.PER_CLASS（最推荐） #

这个方案最简洁，不需要处理静态方法的限制。JUnit 5允许通过@TestInstance(TestInstance.Lifecycle.PER_CLASS)指定测试类的生命周期为每个类仅实例化一次，这样@AfterAll就可以是非静态方法，直接使用@Autowired注入的Flyway实例：

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

@SpringBootTest
@TestInstance(TestInstance.Lifecycle.PER_CLASS) // 核心注解：修改测试实例生命周期
public class YourTestClass {

    @Autowired
    protected Flyway flyway;

    @AfterAll
    public void restoreDatabaseAfterClass() { // 现在可以是非静态方法
        flyway.clean();
        flyway.migrate();
    }

    // 你的测试方法写在这里...
}

为什么这个方案好用？ #

默认情况下JUnit 5的测试实例生命周期是PER_METHOD（每个测试方法创建一个新的测试类实例），所以@AfterAll必须是静态方法。而PER_CLASS模式下整个测试类只会创建一个实例，@AfterAll可以直接用非静态方法，完美适配Spring的依赖注入。

方案2：通过静态ApplicationContext获取Flyway实例 #

如果不想修改测试实例的生命周期，可以通过静态持有Spring的ApplicationContext，在@AfterAll静态方法中获取Flyway Bean：

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

@SpringBootTest
public class YourTestClass {

    private static ApplicationContext staticApplicationContext;

    @Autowired
    protected Flyway flyway;

    // 通过Autowired setter把ApplicationContext赋值给静态变量
    @Autowired
    public void setStaticApplicationContext(ApplicationContext applicationContext) {
        YourTestClass.staticApplicationContext = applicationContext;
    }

    @AfterAll
    public static void restoreDatabaseAfterClass() {
        Flyway staticFlyway = staticApplicationContext.getBean(Flyway.class);
        staticFlyway.clean();
        staticFlyway.migrate();
    }

    // 你的测试方法写在这里...
}

方案3：手动构建Flyway实例（不依赖Spring注入） #

如果不想依赖Spring的上下文管理，可以直接读取数据库配置，手动创建Flyway实例：

import org.junit.jupiter.api.AfterAll;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.core.env.Environment;
import org.flywaydb.core.Flyway;

@SpringBootTest
public class YourTestClass {

    private static String dbUrl;
    private static String dbUser;
    private static String dbPassword;

    // 提前读取数据库配置到静态变量
    @Autowired
    public void initDbProperties(Environment environment) {
        dbUrl = environment.getProperty("spring.datasource.url");
        dbUser = environment.getProperty("spring.datasource.username");
        dbPassword = environment.getProperty("spring.datasource.password");
    }

    @AfterAll
    public static void restoreDatabaseAfterClass() {
        // 手动配置并加载Flyway
        Flyway flyway = Flyway.configure()
                .dataSource(dbUrl, dbUser, dbPassword)
                .load();
        flyway.clean();
        flyway.migrate();
    }

    // 你的测试方法写在这里...
}

注意点 #

如果你的Flyway还有额外配置（比如脚本位置、禁用清理等），需要在手动构建时补充对应的配置项，比如.locations("classpath:db/migration")。

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