import java.io.IOException;

import java.nio.file.Files;

import java.util.HashSet;

import java.util.Set;

import java.util.stream.Collectors;

import jdk.jpackage.test.*;

import jdk.jpackage.test.Annotations.Test;

 * @compile RuntimePackageTest.java

 * @run main/othervm/timeout=720 -Xmx512m jdk.jpackage.test.Main

 *  --jpt-run=RuntimePackageTest

    @Test

    public static void test() {

        new PackageTest()

        .addInitializer(cmd -> {

            cmd.addArguments("--runtime-image", Optional.ofNullable(

                    JPackageCommand.DEFAULT_RUNTIME_IMAGE).orElse(Path.of(

                            System.getProperty("java.home"))));

            // Remove --input parameter from jpackage command line as we don't

            // create input directory in the test and jpackage fails

            // if --input references non existant directory.

            cmd.removeArgumentWithValue("--input");

        })

        .addInstallVerifier(cmd -> {

            Set<Path> srcRuntime = listFiles(Path.of(cmd.getArgumentValue("--runtime-image")));

            Set<Path> dstRuntime = listFiles(cmd.appRuntimeDirectory());

            Set<Path> intersection = new HashSet<>(srcRuntime);

            intersection.retainAll(dstRuntime);

            srcRuntime.removeAll(intersection);

            dstRuntime.removeAll(intersection);

            assertFileListEmpty(srcRuntime, "Missing");

            assertFileListEmpty(dstRuntime, "Unexpected");

        })

        .run();

    }

    private static Set<Path> listFiles(Path root) throws IOException {

        try (var files = Files.walk(root)) {

            return files.map(root::relativize).collect(Collectors.toSet());

        }

    }

    private static void assertFileListEmpty(Set<Path> paths, String msg) {

        TKit.assertTrue(paths.isEmpty(), String.format(

                "Check there are no %s files in installed image",

                msg.toLowerCase()), () -> {

            String msg2 = String.format("%s %d files", msg, paths.size());

            TKit.trace(msg2 + ":");

            paths.stream().map(Path::toString).sorted().forEachOrdered(

                    TKit::trace);

            TKit.trace("Done");