[#noissue] fix plugin-it engine discover

pinpoint-apm · Oct 23, 2023 · 2cc0627 · 2cc0627
1 parent b4ec301
commit 2cc0627
Showing 1 changed file with 20 additions and 14 deletions.
diff --git a/test/src/main/java/com/navercorp/pinpoint/test/plugin/junit5/engine/PluginTestEngine.java b/test/src/main/java/com/navercorp/pinpoint/test/plugin/junit5/engine/PluginTestEngine.java
@@ -94,23 +94,29 @@ public TestDescriptor discover(EngineDiscoveryRequest discoveryRequest, UniqueId
         // Plugin IT
         List<TestDescriptor> removedTestDescriptorList = new ArrayList<>();
         List<TestDescriptor> pluginTestDescriptorList = new ArrayList<>();
-        for (TestDescriptor testDescriptor : engineDescriptor.getChildren()) {
-            if (testDescriptor instanceof ClassTestDescriptor) {
-                final Class<?> testClass = ((ClassTestDescriptor) testDescriptor).getTestClass();
-                TestDescriptor pluginTestDescriptor = null;
-                if (isTestClassWithPluginTest.test(testClass)) {
-                    pluginTestDescriptor = addPluginTestDescriptor(testDescriptor, configuration);
-                } else if (isTestClassWithPluginForkedTest.test(testClass)) {
-                    pluginTestDescriptor = addPluginForkedTestDescriptor(testDescriptor, configuration);
-                } else if (isTestClassWithJunitAgent.test(testClass)) {
-                    pluginTestDescriptor = addPluginJunitTestDescriptor(testDescriptor, configuration);
-                }
 
-                if (pluginTestDescriptor != null) {
-                    pluginTestDescriptorList.add(pluginTestDescriptor);
-                    removedTestDescriptorList.add(testDescriptor);
+        try {
+            for (TestDescriptor testDescriptor : engineDescriptor.getChildren()) {
+                if (testDescriptor instanceof ClassTestDescriptor) {
+                    final Class<?> testClass = ((ClassTestDescriptor) testDescriptor).getTestClass();
+                    TestDescriptor pluginTestDescriptor = null;
+                    if (isTestClassWithPluginTest.test(testClass)) {
+                        pluginTestDescriptor = addPluginTestDescriptor(testDescriptor, configuration);
+                    } else if (isTestClassWithPluginForkedTest.test(testClass)) {
+                        pluginTestDescriptor = addPluginForkedTestDescriptor(testDescriptor, configuration);
+                    } else if (isTestClassWithJunitAgent.test(testClass)) {
+                        pluginTestDescriptor = addPluginJunitTestDescriptor(testDescriptor, configuration);
+                    }
+
+                    if (pluginTestDescriptor != null) {
+                        pluginTestDescriptorList.add(pluginTestDescriptor);
+                        removedTestDescriptorList.add(testDescriptor);
+                    }
                 }
             }
+        } catch (Exception e) {
+            System.out.println("Failed to discover");
+            e.printStackTrace();
         }
 
         for (TestDescriptor removedTestDescriptor : removedTestDescriptorList) {