JUnit测试套件执行完毕后，如何移除运行中的Docker容器？

你遇到的核心问题是：JUnit 5 的 @Suite 类本身不是测试类，它仅用于组织测试套件的结构，因此类中标记的 @AfterAll 生命周期注解不会被触发执行。同时，主套件上的 @SpringBootTest 注解也无效，该注解仅适用于具体的测试类，而非套件类。

以下是几种可行的实现方式，按推荐程度排序：

方案一：自定义 JUnit 扩展实现套件级清理 #

编写一个 JUnit 扩展类，实现套件执行完成后的清理逻辑，再将该扩展绑定到主套件上。

步骤1：编写清理扩展 #

import org.junit.jupiter.api.extension.AfterAllCallback;
import org.junit.jupiter.api.extension.ExtensionContext;
import java.io.IOException;

public class DockerCleanupExtension implements AfterAllCallback {
    @Override
    public void afterAll(ExtensionContext context) throws Exception {
        // 注意：仅删除测试启动的特定容器，避免误删其他运行中的容器
        String cleanupScript = "docker rm -f oracle-test elasticsearch-test";
        ProcessBuilder processBuilder = new ProcessBuilder("powershell.exe", "-Command", cleanupScript);
        
        try {
            Process process = processBuilder.start();
            int exitCode = process.waitFor();
            if (exitCode != 0) {
                System.err.println("Docker 清理脚本执行失败，退出码：" + exitCode);
            }
        } catch (IOException | InterruptedException e) {
            e.printStackTrace();
        }
    }
}

步骤2：绑定扩展到主套件 #

@Suite
@SelectClasses({
    IntegrationTestSuite1.class,
    IntegrationTestSuite2.class,
    IntegrationTestSuite3.class,
})
@SuiteDisplayName("Full integration test suite")
@ExtendWith(DockerCleanupExtension.class) // 应用自定义清理扩展
public class FullIntegrationTestSuite {
    // 无需额外代码，扩展会自动执行清理
}

方案二：用独立测试类执行清理 #

创建一个仅包含清理逻辑的测试类，将其放在套件的最后执行位置，利用测试类的 @AfterAll 完成清理。

步骤1：编写清理测试类 #

import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Test;

public class DockerCleanupTest {
    // 空测试方法，确保类被识别为测试类
    @Test
    void dummyTest() {}

    @AfterAll
    public static void cleanupDockerContainers() {
        String cleanupScript = "docker rm -f oracle-test elasticsearch-test";
        ProcessBuilder processBuilder = new ProcessBuilder("powershell.exe", "-Command", cleanupScript);
        
        try {
            Process process = processBuilder.start();
            process.waitFor();
        } catch (IOException | InterruptedException e) {
            e.printStackTrace();
        }
    }
}

步骤2：添加到主套件末尾 #

@Suite
@SelectClasses({
    IntegrationTestSuite1.class,
    IntegrationTestSuite2.class,
    IntegrationTestSuite3.class,
    DockerCleanupTest.class // 放在最后，确保所有测试完成后执行清理
})
@SuiteDisplayName("Full integration test suite")
public class FullIntegrationTestSuite {
}

方案三：Spring Boot 测试上下文监听（适用于 Spring 环境） #

如果所有子测试都基于 @SpringBootTest，可以编写一个 TestExecutionListener，监听测试上下文关闭事件，执行一次性清理。

步骤1：编写监听类 #

import org.springframework.test.context.TestContext;
import org.springframework.test.context.TestExecutionListener;
import java.io.IOException;

public class DockerCleanupListener implements TestExecutionListener {
    private static boolean cleanupExecuted = false;

    @Override
    public void afterTestClass(TestContext testContext) throws Exception {
        // 确保清理逻辑仅执行一次
        if (!cleanupExecuted) {
            String cleanupScript = "docker rm -f oracle-test elasticsearch-test";
            ProcessBuilder processBuilder = new ProcessBuilder("powershell.exe", "-Command", cleanupScript);
            
            try {
                Process process = processBuilder.start();
                process.waitFor();
            } catch (IOException | InterruptedException e) {
                e.printStackTrace();
            }
            cleanupExecuted = true;
        }
    }
}

步骤2：注册到测试类 #

在每个子测试套件或测试类上添加监听：

@SpringBootTest(classes = MainApplication.class)
@TestExecutionListeners(
    listeners = DockerCleanupListener.class,
    mergeMode = TestExecutionListeners.MergeMode.MERGE_WITH_DEFAULTS
)
public class IntegrationTestSuite1 {
    // 子测试逻辑
}

1. 避免误删容器：绝对不要使用 docker rm -f $(docker ps -q) 这类命令，它会删除所有运行中的容器，一定要通过容器名称、标签等唯一标识指定清理目标。
2. 脚本路径问题：执行 docker 命令无需切换到 resources 目录，直接使用绝对路径或确保 docker 命令在系统 PATH 中即可。
3. 错误处理：建议检查脚本执行的退出码，而非仅捕获异常，避免脚本执行失败但未被察觉的情况。

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