/*

 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/*

 * @test

 * @bug 8153654

 * @summary Tests for jdeps tool with multi-release jar files

 * @modules jdk.jdeps/com.sun.tools.jdeps

 * @library mrjar mrjar/base mrjar/9 mrjar/10 mrjar/v9 mrjar/v10

 * @build test.* p.* q.*

 * @run testng MultiReleaseJar

 */

import org.testng.Assert;

import org.testng.annotations.AfterClass;

import org.testng.annotations.BeforeClass;

import org.testng.annotations.Test;

import java.io.IOException;

import java.io.InputStream;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.concurrent.TimeUnit;

public class MultiReleaseJar {

    Path mrjar;

    String testJdk;

    String fileSep;

    Path cmdPath;

    @BeforeClass

    public void initialize() throws Exception {

        mrjar = Paths.get(System.getProperty("test.classes", "."), "mrjar");

        testJdk = System.getProperty("test.jdk");

        fileSep = System.getProperty("file.separator");

        cmdPath = Paths.get(testJdk, "bin");

    }

    @Test

    public void basic() throws Exception {

        // build the jar file

        Result r = run("jar -cf Version.jar -C base test --release 9 -C 9 test --release 10 -C 10 test");

        checkResult(r);

        // try out a bunch of things

        r = run("jdeps --multi-release 9  -v missing.jar");

        checkResult(r, false, "Warning: Path does not exist: missing.jar");

        r = run("jdeps -v Version.jar");

        checkResult(r, false, "the --multi-release option is not set");

        r = run("jdeps --multi-release base  -v Version.jar");

        checkResult(r, true,

                "Version.jar ->",

                "test.Version",

                "test.Version"

        );

        r = run("jdeps --multi-release 9  -v Version.jar");

        checkResult(r, true,

                "Version.jar ->",

                "9/test.NonPublic",

                "9/test.NonPublic",

                "9/test.Version",

                "9/test.Version",

                "9/test.Version",

                "9/test.Version"

        );

        r = run("jdeps --multi-release 10  -v Version.jar");

        checkResult(r, true,

                "Version.jar ->",

                "10/test.Version",

                "10/test.Version",

                "10/test.Version",

                "10/test.Version",

                "9/test.NonPublic",

                "9/test.NonPublic"

        );

        r = run("jdeps --multi-release 8  -v Version.jar");

        checkResult(r, false, "Error: invalid argument for option: 8");

        r = run("jdeps --multi-release 9.1  -v Version.jar");

        checkResult(r, false, "Error: invalid argument for option: 9.1");

        r = run("jdeps -v -R -cp Version.jar test/Main.class");

        checkResult(r, false, "the --multi-release option is not set");

        r = run("jdeps -v -R -cp Version.jar -multi-release 9 test/Main.class");

        checkResult(r, false,

                "Error: unknown option: -multi-release",

                "Usage: jdeps <options> <path",

                "use -h, -?, -help, or --help"

        );

        r = run("jdeps -v -R -cp Version.jar --multi-release 9 test/Main.class");

        checkResult(r, true,

                "Main.class ->",

                "Main.class ->",

                "test.Main",

                "test.Main",

                "test.Main",

                "Version.jar ->",

                "9/test.NonPublic",

                "9/test.NonPublic",

                "9/test.Version",

                "9/test.Version",

                "9/test.Version",

                "9/test.Version"

        );

        r = run("jdeps -v -R -cp Version.jar --multi-release 10 test/Main.class");

        checkResult(r, true,

                "Main.class ->",

                "Main.class ->",

                "test.Main",

                "test.Main",

                "test.Main",

                "Version.jar ->",

                "10/test.Version",

                "10/test.Version",

                "10/test.Version",

                "10/test.Version",

                "9/test.NonPublic",

                "9/test.NonPublic"

        );

        r = run("jdeps -v -R -cp Version.jar --multi-release base test/Main.class");

        checkResult(r, true,

                "Main.class ->",

                "Main.class ->",

                "test.Main",

                "test.Main",

                "test.Main",

                "Version.jar ->",

                "test.Version",

                "test.Version"

        );

        r = run("jdeps -v -R -cp Version.jar --multi-release 9.1 test/Main.class");

        checkResult(r, false, "Error: invalid argument for option: 9.1");

        // Rebuild jar without version 10

        r = run("jar -cf Version.jar -C base test --release 9 -C 9 test");

        checkResult(r);

        // but ask for version 10

        r = run("jdeps -v -R -cp Version.jar --multi-release 10 test/Main.class");

        checkResult(r, true,

                "Main.class ->",

                "Main.class ->",

                "test.Main",

                "test.Main",

                "test.Main",

                "Version.jar ->",

                "9/test.NonPublic",

                "9/test.NonPublic",

                "9/test.Version",

                "9/test.Version",

                "9/test.Version",

                "9/test.Version"

        );

    }

    @Test

    public void ps_and_qs() throws Exception {

        // build the jar file

        Result r = run("jar -cf PQ.jar -C base p --release 9 -C v9 p -C v9 q --release 10 -C v10 q");

        checkResult(r);

        r = run("jdeps -v -R -cp PQ.jar --multi-release base PQ.jar");

        checkResult(r, true,

                "PQ.jar -> java.base",

                "p.Foo"

        );

        r = run("jdeps -v -R -cp PQ.jar --multi-release 9 PQ.jar");

        checkResult(r, true,

                "PQ.jar -> java.base",

                "9/p.Foo",

                "9/p.Foo",

                "9/q.Bar"

        );

        r = run("jdeps -v -R -cp PQ.jar --multi-release 10 PQ.jar");

        checkResult(r, true,

                "PQ.jar -> java.base",

                "10/q.Bar",

                "10/q.Bar",

                "10/q.Gee",

                "9/p.Foo",

                "9/p.Foo"

        );

    }

    static class Result {

        final String cmd;

        final int rc;

        final String out;

        final String err;

        Result(String cmd, int rc, String out, String err) {

            this.cmd = cmd;

            this.rc = rc;

            this.out = out;

            this.err = err;

        }

    }

    Result run(String cmd) throws Exception {

        String[] cmds = cmd.split(" +");

        cmds[0] = cmdPath.resolve(cmds[0]).toString();

        ProcessBuilder pb = new ProcessBuilder(cmds);

        pb.directory(mrjar.toFile());

        Process p = pb.start();

        p.waitFor(10, TimeUnit.SECONDS);

        String out;

        try (InputStream is = p.getInputStream()) {

            out = new String(is.readAllBytes());

        }

        String err;

        try (InputStream is = p.getErrorStream()) {

            err = new String(is.readAllBytes());

        }

        return new Result(cmd, p.exitValue(), out, err);

    }

    void checkResult(Result r) throws Exception {

        System.out.println(r.cmd);

        System.out.println(r.out);

        if (r.rc != 0) {

            System.out.println(r.err);

            throw new Exception("rc=" + r.rc);

        }

        System.out.println();

    }

    void checkResult(Result r, boolean checkrc, String... lines) throws Exception {

        System.out.println(r.cmd);

        System.out.println(r.out);

        if (checkrc && r.rc != 0) {

            System.out.println(r.err);

            throw new Exception("rc=" + r.rc);

        }

        String[] out = r.out.split("\r?\n");

        Assert.assertEquals(out.length, lines.length);

        int n = 0;

        for (String line : lines) {

            Assert.assertTrue(out[n++].contains(line), "\"" + line + "\"");

        }

        System.out.println();

    }

}