如何将单个自定义JUnit5扩展自动应用于所有测试类？

可以通过全局注册JUnit 5扩展+关闭自动检测的方式实现，无需修改任何测试类，同时避免第三方扩展被自动加载，完全支持Gradle且兼容IDEA。

步骤1：实现自定义清理扩展 #

先编写全局清理扩展，实现JUnit 5的AfterAllCallback接口，在测试类执行完成后执行清理逻辑：

import org.junit.jupiter.api.extension.AfterAllCallback
import org.junit.jupiter.api.extension.ExtensionContext
import io.mockk.unmockkAll

// 替换成你实际的包路径
class GlobalCleanupExtension : AfterAllCallback {
    override fun afterAll(context: ExtensionContext) {
        // 执行Mockk清理
        unmockkAll()
        // 后续可添加其他与Mocking无关的清理逻辑，比如释放资源、重置全局状态等
    }
}

步骤2：Gradle全局配置 #

在根项目的构建脚本中，为所有模块的测试任务统一配置JUnit平台参数：

Kotlin DSL（build.gradle.kts） #

subprojects {
    tasks.withType<Test> {
        useJUnitPlatform()
        // 关闭自动检测，避免加载第三方扩展
        systemProperty("junit.jupiter.extensions.autodetection.enabled", "false")
        // 全局注册自定义清理扩展
        systemProperty("junit.jupiter.extensions.default", "com.yourpackage.GlobalCleanupExtension")
    }
}

Groovy DSL（build.gradle） #

subprojects {
    tasks.withType(Test) {
        useJUnitPlatform()
        systemProperty 'junit.jupiter.extensions.autodetection.enabled', 'false'
        systemProperty 'junit.jupiter.extensions.default', 'com.yourpackage.GlobalCleanupExtension'
    }
}

关键说明 #

• 关闭自动检测后，第三方库（如Mockk）的扩展不会被自动加载，避免破坏现有测试；如果个别测试需要Mockk的扩展，可在对应测试类上手动添加@ExtendWith(MockKExtension::class)。
• IDEA兼容：导入Gradle项目后，IDEA会自动继承测试任务的系统属性配置，无论运行单个测试还是整个测试类，都会自动应用你的自定义扩展。
• 扩展性：后续要添加新的清理逻辑，只需修改GlobalCleanupExtension的afterAll方法，无需改动任何测试类。

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