Support RunWith annotation for JUnit4

Author: dnestoro

common/junit-platform-native/src/main/java/org/graalvm/junit/platform/JUnitPlatformFeature.java

@@ -84,6 +84,11 @@ public static void debug(String format, Object... args) {
         }
     }
 
+    @Override
+    public void afterRegistration(AfterRegistrationAccess access) {
+        extensionConfigProviders.forEach(p -> p.initialize(access.getApplicationClassLoader(), nativeImageConfigImpl));
+    }
+
     public static boolean debug() {
         return ImageSingletons.lookup(JUnitPlatformFeature.class).debug;
     }

common/junit-platform-native/src/main/java/org/graalvm/junit/platform/config/core/NativeImageConfiguration.java

@@ -66,6 +66,11 @@ default void registerAllClassMembersForReflection(Class<?>... classes) {
             registerForReflection(clazz.getDeclaredConstructors());
             registerForReflection(clazz.getDeclaredMethods());
             registerForReflection(clazz.getDeclaredFields());
+
+            Class<?>[] declaredClasses = clazz.getDeclaredClasses();
+            for (Class<?> cls : declaredClasses) {
+                registerAllClassMembersForReflection(cls);
+            }
         }
     }
 }

common/junit-platform-native/src/main/java/org/graalvm/junit/platform/config/core/PluginConfigProvider.java

@@ -41,10 +41,18 @@
 
 package org.graalvm.junit.platform.config.core;
 
-public interface PluginConfigProvider {
+public abstract class PluginConfigProvider {
 
-    void onLoad(NativeImageConfiguration config);
+    protected ClassLoader applicationClassLoader;
+    protected NativeImageConfiguration nativeImageConfigImpl;
 
-    void onTestClassRegistered(Class<?> testClass, NativeImageConfiguration registry);
 
+    public abstract void onLoad(NativeImageConfiguration config);
+
+    public abstract void onTestClassRegistered(Class<?> testClass, NativeImageConfiguration registry);
+
+    public final void initialize(ClassLoader classLoader, NativeImageConfiguration nic) {
+        applicationClassLoader = classLoader;
+        nativeImageConfigImpl = nic;
+    }
 }

common/junit-platform-native/src/main/java/org/graalvm/junit/platform/config/jupiter/JupiterConfigProvider.java

@@ -63,7 +63,7 @@
 
 import static org.graalvm.junit.platform.JUnitPlatformFeature.debug;
 
-public class JupiterConfigProvider implements PluginConfigProvider {
+public class JupiterConfigProvider extends PluginConfigProvider {
 
     @Override
     public void onLoad(NativeImageConfiguration config) {

common/junit-platform-native/src/main/java/org/graalvm/junit/platform/config/platform/PlatformConfigProvider.java

@@ -47,7 +47,7 @@
 import org.junit.platform.launcher.TestIdentifier;
 
 
-public class PlatformConfigProvider implements PluginConfigProvider {
+public class PlatformConfigProvider extends PluginConfigProvider {
 
     @Override
     public void onLoad(NativeImageConfiguration config) {

common/junit-platform-native/src/main/java/org/graalvm/junit/platform/config/vintage/VintageConfigProvider.java

@@ -46,7 +46,10 @@
 import org.graalvm.nativeimage.hosted.RuntimeReflection;
 import org.graalvm.nativeimage.hosted.RuntimeSerialization;
 
-public class VintageConfigProvider implements PluginConfigProvider {
+import java.lang.annotation.Annotation;
+import java.lang.reflect.Method;
+
+public class VintageConfigProvider extends PluginConfigProvider {
 
     @Override
     public void onLoad(NativeImageConfiguration config) {
@@ -60,5 +63,23 @@ public void onLoad(NativeImageConfiguration config) {
 
     @Override
     public void onTestClassRegistered(Class<?> testClass, NativeImageConfiguration registry) {
+        registerAnnotationClassesForReflection("org.junit.runner.RunWith", "value", testClass);
+        registerAnnotationClassesForReflection("org.junit.runners.Parameterized.UseParametersRunnerFactory", "value", testClass);
+    }
+
+    @SuppressWarnings("unchecked")
+    private void registerAnnotationClassesForReflection(String annotationClass, String classProviderMethod, Class<?> testClass) {
+        try {
+            Class<Annotation> annotation = (Class<Annotation>) applicationClassLoader.loadClass(annotationClass);
+            Method classProvider = annotation.getDeclaredMethod(classProviderMethod);
+
+            Annotation classAnnotation = testClass.getAnnotation(annotation);
+            if (classAnnotation != null) {
+                Class<?> annotationArgument = (Class<?>) classProvider.invoke(classAnnotation);
+                nativeImageConfigImpl.registerAllClassMembersForReflection(annotationArgument);
+            }
+        } catch (ReflectiveOperationException e) {
+             // intentionally ignored
+        }
     }
 }