import java.io.IOException;

import jdk.jpackage.internal.IOUtils;

import jdk.jpackage.test.Test;

import jdk.jpackage.test.HelloApp;

import jdk.jpackage.test.JPackageCommand;

 * @summary jpackage with --win-console

 * @modules jdk.jpackage/jdk.jpackage.internal

public class WinConsoleTest {

    public static void main(String[] args) throws IOException {

        JPackageCommand cmd = JPackageCommand.helloAppImage();

        final Path launcherPath = cmd.appImage().resolve(

                cmd.launcherPathInAppImage());

        IOUtils.deleteRecursive(cmd.outputDir().toFile());

        cmd.execute().assertExitCodeIsZero();

        HelloApp.executeAndVerifyOutput(launcherPath);

        checkSubsystem(launcherPath, false);

        IOUtils.deleteRecursive(cmd.outputDir().toFile());

        cmd.addArgument("--win-console").execute().assertExitCodeIsZero();

        HelloApp.executeAndVerifyOutput(launcherPath);

        checkSubsystem(launcherPath, true);

    }

    private static void checkSubsystem(Path path, boolean isConsole) throws

            IOException {

        final int subsystemGui = 2;

        final int subsystemConsole = 3;

        final int bufferSize = 512;

        final int expectedSubsystem = isConsole ? subsystemConsole : subsystemGui;

            byte[] bytes = new byte[bufferSize];

            Test.assertEquals(bufferSize, inputStream.read(bytes),

                    String.format("Check %d bytes were read from %s file",

                            bufferSize, path));

            for (int i = 0; i < (bytes.length - 4); i++) {

                if (bytes[i] == 0x50 && bytes[i + 1] == 0x45

                        && bytes[i + 2] == 0x0 && bytes[i + 3] == 0x0) {

                    Test.assertEquals(expectedSubsystem, subsystem,

                            String.format("Check subsystem of PE [%s] file",

                                    path));

                    return;

        Test.assertUnexpected(String.format(

                "Subsystem not found in PE header of [%s] file", path));