如何在JUnit5扩展中获取传入会话的特定测试用例UniqueId

我们开发了一款JUnit5测试数据生成扩展，功能类似@ParameterizedTest。当在IntelliJ等IDE中运行单个特定测试用例时，该用例的UniqueId会通过JUnit5的Launcher API封装进LauncherDiscoveryRequest传入，最终流转到TestTemplateTestDescriptor的过滤逻辑中，只运行匹配该UniqueId的用例。

当前扩展需要遍历所有生成的测试用例才能定位目标用例，不仅耗时，还会受非确定性用例生成逻辑的影响。我们希望直接利用传入Launcher的UniqueId生成对应测试用例，但需要在扩展中合规地提前获取这个UniqueId。

• 尝试使用TestExecutionListener.dynamicTestRegistered回调：调用时机晚于过滤和用例生成流程，无法提前获取UniqueId。
• 尝试从TestTemplateInvocationContextProvider.provideTestTemplateInvocationContexts传入的ExtensionContext中获取：该UniqueId对应整个测试类，内部目标UniqueId只能通过反射获取，存在兼容性风险。

可以通过实现JUnit5的DiscoveryPostProcessor接口，在测试发现阶段直接从LauncherDiscoveryRequest中提取目标UniqueId，具体步骤如下：

1. 实现DiscoveryPostProcessor接口
   这个接口允许你在测试发现流程中拦截并处理LauncherDiscoveryRequest，从中筛选出目标UniqueId：

   import org.junit.platform.engine.discovery.UniqueIdSelector;
   import org.junit.platform.launcher.DiscoveryPostProcessor;
   import org.junit.platform.launcher.LauncherDiscoveryRequest;
   import org.junit.platform.launcher.LauncherDiscoveryResult;
   
   public class TargetUniqueIdExtractor implements DiscoveryPostProcessor {
       private static final ThreadLocal<String> TARGET_UNIQUE_ID = new ThreadLocal<>();
   
       @Override
       public void postProcessDiscovery(LauncherDiscoveryRequest request, LauncherDiscoveryResult result) {
           // 从请求选择器中筛选出UniqueIdSelector
           request.getSelectors().stream()
                   .filter(selector -> selector instanceof UniqueIdSelector)
                   .map(UniqueIdSelector.class::cast)
                   .findFirst()
                   .ifPresent(selector -> TARGET_UNIQUE_ID.set(selector.getUniqueId().toString()));
       }
   
       // 提供静态方法供扩展后续获取UniqueId
       public static String getTargetUniqueId() {
           return TARGET_UNIQUE_ID.get();
       }
   
       // 测试结束后清理线程本地存储
       public static void clear() {
           TARGET_UNIQUE_ID.remove();
       }
   }
   

2. 注册DiscoveryPostProcessor
   在扩展的@Extension类中，通过@RegisterExtension注解或者服务加载机制（在META-INF/services/org.junit.platform.launcher.DiscoveryPostProcessor文件中写入处理器全类名）注册这个处理器，确保它在测试发现阶段被执行。

3. 在测试用例生成阶段使用UniqueId
   在TestTemplateInvocationContextProvider.provideTestTemplateInvocationContexts方法中，调用TargetUniqueIdExtractor.getTargetUniqueId()获取目标UniqueId，直接生成对应的测试用例，无需遍历所有生成结果：

   @Override
   public Stream<TestTemplateInvocationContext> provideTestTemplateInvocationContexts(ExtensionContext context) {
       String targetUniqueId = TargetUniqueIdExtractor.getTargetUniqueId();
       if (targetUniqueId != null) {
           // 根据UniqueId直接生成对应的测试用例上下文
           return Stream.of(createContextForUniqueId(targetUniqueId));
       }
       // 没有目标UniqueId时，生成所有测试用例
       return generateAllTestContexts();
   }
   

4. 清理线程本地存储
   为避免线程污染，建议在TestExecutionListener.afterTestExecution回调或者扩展的销毁方法中调用TargetUniqueIdExtractor.clear()。

• 完全基于JUnit5公开的扩展API实现，无反射操作，兼容性和稳定性有保障。
• 在测试发现阶段提前获取目标UniqueId，正好赶在用例生成和过滤流程之前，满足需求。

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