Detect native-image-compatible JDK using Gradle toolchains

Author: marcphilipp

platform-tooling-support-tests/platform-tooling-support-tests.gradle.kts

@@ -227,6 +227,7 @@ val test by testing.suites.getting(JvmTestSuite::class) {
 
 				val gradleJavaVersion = JavaVersion.current().majorVersion.toInt()
 				jvmArgumentProviders += JavaHomeDir(project, gradleJavaVersion, develocity.testDistribution.enabled)
+				jvmArgumentProviders += JavaHomeDir(project, gradleJavaVersion, develocity.testDistribution.enabled, nativeImage = true)
 				systemProperty("gradle.java.version", gradleJavaVersion)
 			}
 		}
@@ -253,14 +254,15 @@ class MavenRepo(project: Project, @get:Internal val repoDir: Provider<File>) : C
 	override fun asArguments() = listOf("-Dmaven.repo=${repoDir.get().absolutePath}")
 }
 
-class JavaHomeDir(project: Project, @Input val version: Int, testDistributionEnabled: Provider<Boolean>) : CommandLineArgumentProvider {
+class JavaHomeDir(project: Project, @Input val version: Int, testDistributionEnabled: Provider<Boolean>, @Input val nativeImage: Boolean = false) : CommandLineArgumentProvider {
 
 	@Internal
 	val javaLauncher: Property<JavaLauncher> = project.objects.property<JavaLauncher>()
 			.value(project.provider {
 				try {
 					project.javaToolchains.launcherFor {
 						languageVersion = JavaLanguageVersion.of(version)
+						nativeImageCapable = nativeImage
 					}.get()
 				} catch (e: Exception) {
 					null
@@ -276,7 +278,7 @@ class JavaHomeDir(project: Project, @Input val version: Int, testDistributionEna
 		}
 		val metadata = javaLauncher.map { it.metadata }
 		val javaHome = metadata.map { it.installationPath.asFile.absolutePath }.orNull
-		return javaHome?.let { listOf("-Djava.home.$version=$it") } ?: emptyList()
+		return javaHome?.let { listOf("-Djava.home.$version${if (nativeImage) ".nativeImage" else ""}=$it") } ?: emptyList()
 	}
 }
 

platform-tooling-support-tests/projects/graalvm-starter/gradle.properties

@@ -0,0 +1,2 @@
+org.gradle.java.installations.fromEnv=GRAALVM_HOME
+org.gradle.java.installations.auto-download=false

platform-tooling-support-tests/src/main/java/platform/tooling/support/Helper.java

@@ -82,4 +82,9 @@ public static Optional<Path> getJavaHome(int version) {
 		);
 		return sources.filter(Objects::nonNull).findFirst().map(Path::of);
 	}
+
+	public static Optional<Path> getJavaHomeWithNativeImageSupport(int version) {
+		var value = System.getProperty("java.home." + version + ".nativeImage");
+		return Optional.ofNullable(value).map(Path::of);
+	}
 }

platform-tooling-support-tests/src/test/java/platform/tooling/support/tests/GraalVmStarterTests.java

@@ -13,17 +13,19 @@
 import static java.util.concurrent.TimeUnit.MINUTES;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static platform.tooling.support.Helper.getJavaHomeWithNativeImageSupport;
+import static platform.tooling.support.ProcessStarters.getGradleJavaVersion;
 import static platform.tooling.support.tests.Projects.copyToWorkspace;
 
 import java.nio.file.Path;
 
 import org.junit.jupiter.api.Order;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.Timeout;
-import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
 import org.junit.jupiter.api.extension.DisabledOnOpenJ9;
 import org.junit.jupiter.api.io.TempDir;
 import org.junit.platform.tests.process.OutputFiles;
+import org.opentest4j.TestAbortedException;
 
 import platform.tooling.support.MavenRepo;
 import platform.tooling.support.ProcessStarters;
@@ -33,16 +35,20 @@
  */
 @Order(Integer.MIN_VALUE)
 @DisabledOnOpenJ9
-@EnabledIfEnvironmentVariable(named = "GRAALVM_HOME", matches = ".+")
 class GraalVmStarterTests {
 
 	@Test
 	@Timeout(value = 10, unit = MINUTES)
 	void runsTestsInNativeImage(@TempDir Path workspace, @FilePrefix("gradle") OutputFiles outputFiles)
 			throws Exception {
+
+		var graalVmHome = getJavaHomeWithNativeImageSupport(getGradleJavaVersion());
+
 		var result = ProcessStarters.gradlew() //
 				.workingDir(copyToWorkspace(Projects.GRAALVM_STARTER, workspace)) //
-				.addArguments("-Dmaven.repo=" + MavenRepo.dir()) //
+				.putEnvironment("GRAALVM_HOME",
+					graalVmHome.orElseThrow(TestAbortedException::new).toString()).addArguments(
+						"-Dmaven.repo=" + MavenRepo.dir()) //
 				.addArguments("javaToolchains", "nativeTest", "--no-daemon", "--stacktrace", "--no-build-cache",
 					"--warning-mode=fail") //
 				.redirectOutput(outputFiles) //