Polish JupiterConfigProvider (#312)

graalvm · Sep 23, 2022 · dda0a59 · dda0a59
2 parents b79c692 + 14c6a08
commit dda0a59
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/...native/src/main/java/org/graalvm/junit/platform/config/jupiter/JupiterConfigProvider.java b/...native/src/main/java/org/graalvm/junit/platform/config/jupiter/JupiterConfigProvider.java
@@ -85,6 +85,13 @@ public void onLoad(NativeImageConfiguration config) {
         for (String className : buildTimeInitializedClasses) {
             config.initializeAtBuildTime(className);
         }
+
+        try {
+            Class <?> executor = Class.forName("org.junit.jupiter.engine.extension.TimeoutExtension$ExecutorResource");
+            config.registerAllClassMembersForReflection(executor);
+        } catch (ClassNotFoundException e) {
+            debug("Failed to register class for reflection. Reason: %s", e);
+        }
     }
 
     @Override
@@ -100,13 +107,6 @@ public void onTestClassRegistered(Class<?> testClass, NativeImageConfiguration r
         AnnotationUtils.registerClassesFromAnnotationForReflection(testClass, registry, MethodSource.class, JupiterConfigProvider::handleMethodSource);
         AnnotationUtils.registerClassesFromAnnotationForReflection(testClass, registry, EnabledIf.class, JupiterConfigProvider::handleEnabledIf);
         AnnotationUtils.registerClassesFromAnnotationForReflection(testClass, registry, DisabledIf.class, JupiterConfigProvider::handleDisabledIf);
-
-        try {
-            Class <?> executor = Class.forName("org.junit.jupiter.engine.extension.TimeoutExtension$ExecutorResource");
-            registry.registerAllClassMembersForReflection(executor);
-        } catch (ClassNotFoundException e) {
-            debug("Failed to register class for reflection. Reason: %s", e);
-        }
     }
 
     private static Class<?>[] handleMethodSource(MethodSource annotation) {