如何在Testcontainers中每个测试类后重置数据库？

我定义了如下基础测试类，其他测试类均继承该类：

public class AbstractContainerBaseTest {

    @ServiceConnection
    static final MySQLContainer<?> MY_SQL_CONTAINER = new MySQLContainer<>("mysql:8.0.32")

    static {
        MY_SQL_CONTAINER.start();
    }
}

class FirstTest extends AbstractContainerBaseTest {

    @Test
    void someTestMethod() {
        
    }
}

运行全部测试类时部分测试方法失败，但单独运行对应测试类则正常。原因是所有测试类共享同一容器，想咨询如何在每个测试类执行后重置数据库？

方法1：执行数据库重置脚本（推荐） #

在AbstractContainerBaseTest中添加@AfterClass方法，直接通过容器的连接执行清空表或恢复初始状态的SQL脚本，确保每个测试类执行完后数据库回到干净状态：

@AfterClass
static void resetDatabase() {
    try (Connection conn = MY_SQL_CONTAINER.createConnection("")) {
        Statement stmt = conn.createStatement();
        // 示例：清空所有业务表，根据实际表名调整
        stmt.execute("TRUNCATE TABLE user; TRUNCATE TABLE `order`;");
        // 如果有初始数据需要恢复，可以执行INSERT脚本或者导入备份文件
        // stmt.executeUpdate("source /docker-entrypoint-initdb.d/init.sql");
    } catch (SQLException e) {
        throw new RuntimeException("重置数据库失败", e);
    }
}

如果需要每个测试方法执行完都重置，把@AfterClass换成@AfterEach即可。

方法2：利用Spring事务回滚（适合Spring Boot测试） #

在测试类上添加@Transactional和@Rollback注解，这样每个测试方法的数据库操作都会被自动回滚，不会留下脏数据：

@Transactional
@Rollback
class FirstTest extends AbstractContainerBaseTest {
    @Test
    void someTestMethod() {
        // 测试逻辑，执行完自动回滚
    }
}

注意：这种方式仅对Spring管理的事务操作有效，直接JDBC或非事务的修改无法被回滚。

方法3：恢复数据库初始快照 #

在容器启动时导出初始数据快照，测试类结束后导入快照恢复：

private static String initialDataSnapshot;

static {
    MY_SQL_CONTAINER.start();
    // 导出初始数据库快照
    ExecResult result = MY_SQL_CONTAINER.execInContainer(
        "mysqldump", "-u", MY_SQL_CONTAINER.getUsername(),
        "-p" + MY_SQL_CONTAINER.getPassword(),
        MY_SQL_CONTAINER.getDatabaseName()
    );
    initialDataSnapshot = result.getStdout();
}

@AfterClass
static void restoreDatabase() {
    // 导入快照恢复初始状态
    MY_SQL_CONTAINER.execInContainer(
        "mysql", "-u", MY_SQL_CONTAINER.getUsername(),
        "-p" + MY_SQL_CONTAINER.getPassword(),
        MY_SQL_CONTAINER.getDatabaseName(),
        "-e", initialDataSnapshot
    );
}

方法4：重启容器（不推荐，效率较低） #

如果以上方法都不适用，可以在测试类结束后重启容器，缺点是会增加测试总耗时：

@AfterClass
static void restartContainer() {
    MY_SQL_CONTAINER.stop();
    MY_SQL_CONTAINER.start();
}

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