/*
 * 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 8168836
 * @summary  Basic argument validation for --add-reads
 * @library /lib/testlibrary /test/lib
 * @modules jdk.compiler
 * @build jdk.test.lib.compiler.ModuleInfoMaker
 * @build jdk.test.lib.compiler.CompilerUtils
 * @build jdk.testlibrary.*
 * @build AddReadsTestWarningError
 * @run testng AddReadsTestWarningError
 */

import java.io.BufferedOutputStream;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.stream.Stream;

import jdk.test.lib.compiler.ModuleInfoMaker;
import jdk.testlibrary.OutputAnalyzer;
import static jdk.testlibrary.ProcessTools.*;

import org.testng.annotations.BeforeTest;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
import static org.testng.Assert.*;

@Test
public class AddReadsTestWarningError {

    private static final Path MODS_DIR = Paths.get("mods");
    private static final Path SRC_DIR = Paths.get("src");
    private static final String M1_MAIN = "m1/p1.C1";
    private static final String M4_MAIN = "m4/p4.C4";

    @BeforeTest
    public void setup() throws Exception {
        ModuleInfoMaker builder = new ModuleInfoMaker(SRC_DIR);
        builder.writeJavaFiles("m1",
            "module m1 { requires m4; }",
            "package p1; public class C1 { " +
            "    public static void main(String... args) {" +
            "        p2.C2 c2 = new p2.C2();" +
            "        p3.C3 c3 = new p3.C3();" +
            "    }" +
            "}"
        );

        builder.writeJavaFiles("m2",
            "module m2 { exports p2; }",
            "package p2; public class C2 { }"
        );

        builder.writeJavaFiles("m3",
            "module m3 { exports p3; }",
            "package p3; public class C3 { }"
        );

        builder.writeJavaFiles("m4",
            "module m4 { requires m2; requires m3; }",
            "package p4; public class C4 { " +
            "    public static void main(String... args) {}" +
            "}"
        );

        builder.compile("m2", MODS_DIR);
        builder.compile("m3", MODS_DIR);
        builder.compile("m4", MODS_DIR);
        builder.compile("m1", MODS_DIR, "--add-reads", "m1=m2,m3");
    }


    @DataProvider(name = "goodcases")
    public Object[][] goodCases() {
        return new Object[][]{
            // empty items
            { "m1=,m2,m3",       null },
            { "m1=m2,,m3",       null },
            { "m1=m2,m3,",       null },

            // duplicates
            { "m1=m2,m2,m3,,",  null },

        };
    }


    @Test(dataProvider = "goodcases")
    public void test(String value, String ignore) throws Exception {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        PrintStream ps = new PrintStream(new BufferedOutputStream(baos));
        OutputAnalyzer outputAnalyzer =
            executeTestJava("--add-reads", value,
                            "--module-path", MODS_DIR.toString(),
                            "-m", M1_MAIN)
                .outputTo(ps)
                .errorTo(ps);

        assertTrue(outputAnalyzer.getExitValue() == 0);

        System.out.println(baos.toString());
        String[] output = baos.toString().split("\\R");
        assertFalse(Arrays.stream(output)
                          .filter(s -> !s.matches("WARNING: Module name .* may soon be illegal"))
                          .filter(s -> s.startsWith("WARNING:"))
                          .findAny().isPresent());
    }


    @DataProvider(name = "illFormedAddReads")
    public Object[][] illFormedAddReads() {
        return new Object[][]{
            { "m1",         "Unable to parse --add-reads <module>=<value>: m1" },

            // missing source part
            { "=m2",        "Unable to parse --add-reads <module>=<value>: =m2" },

            // empty list, missing target
            { "m1=",        "Unable to parse --add-reads <module>=<value>: m1=" },

            // empty list
            { "m1=,,",      "Target must be specified: --add-reads m1=,," },
        };
    }


    @Test(dataProvider = "illFormedAddReads")
    public void testIllFormedAddReads(String value, String msg) throws Exception {
        int exitValue =
            executeTestJava("--add-reads", value,
                            "--module-path", MODS_DIR.toString(),
                            "-m", M4_MAIN)
                .outputTo(System.out)
                .errorTo(System.out)
                .shouldContain(msg)
                .getExitValue();

        assertTrue(exitValue != 0);
    }


    @DataProvider(name = "unknownNames")
    public Object[][] unknownNames() {
        return new Object[][]{

            // source not found
            {"DoesNotExist=m2",    "WARNING: Unknown module: DoesNotExist specified to --add-reads"},

            // target not found
            {"m2=DoesNotExist",    "WARNING: Unknown module: DoesNotExist specified to --add-reads"},

            // bad names
            {"m*=m2",              "WARNING: Unknown module: m* specified to --add-reads"},
            {"m2=m!",              "WARNING: Unknown module: m! specified to --add-reads"},

        };
    }

    @Test(dataProvider = "unknownNames")
    public void testUnknownNames(String value, String msg) throws Exception {
        int exitValue =
            executeTestJava("--add-reads", value,
                            "--module-path", MODS_DIR.toString(),
                            "-m", M4_MAIN)
                .outputTo(System.out)
                .errorTo(System.out)
                .shouldContain(msg)
                .getExitValue();

        assertTrue(exitValue == 0);
    }


    @DataProvider(name = "missingArguments")
    public Object[][] missingArguments() {
        return new Object[][]{
            { new String[] {"--add-reads" },
                "Error: --add-reads requires modules to be specified"},

            { new String[] { "--add-reads=" },
                "Error: --add-reads= requires modules to be specified"},

            { new String[] { "--add-reads", "" },
                "Error: --add-reads requires modules to be specified"},
        };
    }

    @Test(dataProvider = "missingArguments")
    public void testEmptyArgument(String[] options, String msg) throws Exception {
        String[] args = Stream.concat(Arrays.stream(options), Stream.of("-version"))
                              .toArray(String[]::new);
        int exitValue = executeTestJava(args)
            .outputTo(System.out)
            .errorTo(System.out)
            .shouldContain(msg)
            .getExitValue();

        assertTrue(exitValue != 0);
    }
}