JUnit 5如何在所有测试类执行完毕后仅执行一次方法？

我希望在所有测试执行完成后运行一个执行特定逻辑的最终方法。我了解JUnit提供的扩展方法，但它们似乎都会在每个测试类执行后运行。

现有实现代码 #

Watcher 扩展类 #

public class Watcher implements BeforeAllCallback, AfterAllCallback {
   public static boolean started = false;

   @Override
   public void beforeAll(ExtensionContext ec) throws Exception {
      if(!started) {
         started = true;
         // 这里的逻辑通过静态布尔值正常工作
         System.out.println("Started all");
      }   
   }

   @Override
   public void afterAll(ExtensionContext ec) throws Exception {
      // 这段代码会在每个继承的测试类执行后运行
         System.out.println("Finished all");
   }
} 

ATest 测试类 #

public class ATest extends BaseTest {
   @Test
   public void A() {
      System.out.println("A");
      assertTrue(true);
   }
}

BTest 测试类 #

public class BTest extends BaseTest {
   @Test
   public void B() {
      System.out.println("B");
      assertTrue(true);
   }
}

BaseTest 基类 #

@ExtendWith(Watcher.class)
public class BaseTest {
   // 基类相关代码
}

当前输出 #

• Started all
• A
• Finished all
• B
• Finished all

期望输出 #

• Started all
• A
• B
• Finished all

解决方案 #

方法1：利用JVM关闭钩子（简单可靠） #

修改Watcher类的beforeAll方法，添加JVM关闭钩子。钩子会在所有测试执行完毕、JVM即将退出时触发，确保最终逻辑只执行一次：

public class Watcher implements BeforeAllCallback, AfterAllCallback {
   public static boolean started = false;

   @Override
   public void beforeAll(ExtensionContext ec) throws Exception {
      if(!started) {
         started = true;
         System.out.println("Started all");
         // 添加JVM关闭钩子，执行最终逻辑
         Runtime.getRuntime().addShutdownHook(new Thread(() -> {
             System.out.println("Finished all");
             // 在这里编写你的特定最终逻辑
         }));
      }   
   }

   @Override
   public void afterAll(ExtensionContext ec) throws Exception {
       // 移除原本每个测试类执行后都会触发的逻辑
   }
}

方法2：统计测试类数量，判断是否为最后一个执行的测试类 #

通过ExtensionContext获取本次运行的所有测试类总数，在afterAll中统计已完成的测试类数量，当数量匹配时执行最终逻辑：

public class Watcher implements BeforeAllCallback, AfterAllCallback {
   private static boolean started = false;
   private static int totalTestClasses;
   private static int completedTestClasses = 0;

   @Override
   public void beforeAll(ExtensionContext ec) throws Exception {
      if(!started) {
         started = true;
         // 获取根上下文下的所有测试类（即本次运行的所有测试类）
         totalTestClasses = ec.getRoot().getChildren().size();
         System.out.println("Started all");
      }   
   }

   @Override
   public void afterAll(ExtensionContext ec) throws Exception {
       completedTestClasses++;
       // 当所有测试类都执行完毕时，运行最终逻辑
       if (completedTestClasses == totalTestClasses) {
           System.out.println("Finished all");
           // 在这里编写你的特定最终逻辑
       }
   }
}

说明 #

• 方法1实现简单，无需统计测试类数量，适用于测试结束后JVM会正常退出的场景。
• 方法2更贴合JUnit的扩展机制，适合需要在JUnit生命周期内完成最终逻辑的场景，不受JVM关闭的影响。

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