From b0b1173bcc4d85de0f54443ae41b73256815549b Mon Sep 17 00:00:00 2001
From: Severin Gehwolf <sgehwolf@redhat.com>
Date: Fri, 29 Nov 2024 12:43:23 +0100
Subject: [PATCH] Add basic jlink smoke test

Basically runs `jlink --add-modules java.base --output <output>
--verbose`. This is to ensure that signing of builds don't break the
jlink functionality. Especially when linking from the run-time image
(JEP 493 in JDK 24+).
---
 .../src/net/adoptium/test/FeatureTests.java   | 83 +++++++++++++++++--
 1 file changed, 76 insertions(+), 7 deletions(-)

diff --git a/test/functional/buildAndPackage/src/net/adoptium/test/FeatureTests.java b/test/functional/buildAndPackage/src/net/adoptium/test/FeatureTests.java
index 976e92b74..b51b30d5e 100644
--- a/test/functional/buildAndPackage/src/net/adoptium/test/FeatureTests.java
+++ b/test/functional/buildAndPackage/src/net/adoptium/test/FeatureTests.java
@@ -1,6 +1,6 @@
 /*
  * ********************************************************************************
- * Copyright (c) 2021 Contributors to the Eclipse Foundation
+ * Copyright (c) 2021, 2024 Contributors to the Eclipse Foundation
  *
  * See the NOTICE file(s) with this work for additional
  * information regarding copyright ownership.
@@ -15,20 +15,27 @@
 
 package net.adoptium.test;
 
-import org.testng.annotations.BeforeTest;
-import org.testng.annotations.Test;
+import static org.testng.Assert.assertEquals;
+import static org.testng.Assert.assertTrue;
 
 import java.io.IOException;
+import java.nio.file.FileVisitResult;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.SimpleFileVisitor;
+import java.nio.file.attribute.BasicFileAttributes;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Locale;
 import java.util.logging.Logger;
 import java.util.regex.Pattern;
 
-import static net.adoptium.test.JdkPlatform.Architecture;
-import static net.adoptium.test.JdkPlatform.OperatingSystem;
-import static org.testng.Assert.assertEquals;
-import static org.testng.Assert.assertTrue;
+import org.testng.annotations.BeforeTest;
+import org.testng.annotations.Test;
+
+import net.adoptium.test.JdkPlatform.Architecture;
+import net.adoptium.test.JdkPlatform.OperatingSystem;
 
 /**
  * Tests the availability of various features like garbage collectors, flight recorder, that need to be enabled via
@@ -104,6 +111,68 @@ public void testLinkableRuntimeJDK24Plus() {
         }
     }
 
+    /**
+     * Tests whether basic jlink works using the default module path. The only
+     * included module in the output image is {@code java.base}.
+     */
+    @Test
+    public void testJlinkJdk11AndBetter() throws IOException {
+        // Only JDK 11 (JDK 9, really) and better have jlink
+        if (jdkVersion.isNewerOrEqual(11)) {
+            Path output = Paths.get("java.base-image");
+            ensureOutputDirectoryDeleted(output);
+            List<String> command = new ArrayList<>();
+            command.add(String.format("%s/bin/jlink", testJdkHome));
+            command.add("--add-modules");
+            command.add("java.base");
+            command.add("--output");
+            command.add(output.toString());
+            command.add("--verbose");
+
+            try {
+                ProcessBuilder processBuilder = new ProcessBuilder(command);
+                processBuilder.inheritIO();
+                Process process = processBuilder.start();
+
+                if (process.waitFor() != 0) {
+                    throw new AssertionError("Basic jlink smoke test failed! " + command);
+                }
+                LOGGER.info("Basic jlink smoke test passed. Command was: " + command);
+            } catch (InterruptedException | IOException e) {
+                throw new RuntimeException("Failed to launch JVM", e);
+            } finally {
+                ensureOutputDirectoryDeleted(output);
+            }
+        }
+    }
+
+    private static void ensureOutputDirectoryDeleted(Path path) throws IOException {
+        if (Files.exists(path)) {
+            deleteDirRecursively(path);
+        }
+    }
+
+    private static void deleteDirRecursively(Path path) throws IOException {
+        Files.walkFileTree(path, new SimpleFileVisitor<Path>() {
+
+            @Override
+            public FileVisitResult postVisitDirectory(Path dir, IOException e) throws IOException {
+                Files.delete(dir);
+                return FileVisitResult.CONTINUE;
+            }
+
+            @Override
+            public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
+                if (Files.isDirectory(file)) {
+                    // deleted in post-visit
+                    return FileVisitResult.CONTINUE;
+                }
+                Files.delete(file);
+                return FileVisitResult.CONTINUE;
+            }
+        });
+    }
+
     private boolean isVendorAdoptium() {
         return System.getProperty("java.vendor", "").toLowerCase(Locale.US).contains("adoptium");
     }