/*

 * 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.

 */

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.ArrayList;

import java.util.List;

import java.util.Map;

import jdk.testlibrary.OutputAnalyzer;

import org.testng.annotations.BeforeTest;

import org.testng.annotations.DataProvider;

import org.testng.annotations.Test;

import static org.testng.Assert.assertTrue;

import static jdk.testlibrary.ProcessTools.*;

/**

 * @test

 * @bug 8168205

 * @summary Test the default class path if -Djava.class.path is set

 * @library /lib/testlibrary

 * @modules jdk.compiler

 * @build CompilerUtils jdk.testlibrary.*

 * @run testng JavaClassPathTest

 */

public class JavaClassPathTest {

    private static final Path SRC_DIR = Paths.get(System.getProperty("test.src"),

                                                  "src");

    private static final Path MODS_DIR = Paths.get("mods");

    private static final String TEST_MODULE = "m";

    private static final String TEST_MAIN = "jdk.test.Main";

    @BeforeTest

    public void setup() throws Exception {

        boolean compiled = CompilerUtils.compile(SRC_DIR.resolve(TEST_MODULE),

                                                 MODS_DIR.resolve(TEST_MODULE));

        assertTrue(compiled, "module " + TEST_MODULE + " did not compile");

        // add the class and a resource to the current working directory

        Path file = Paths.get("jdk/test/Main.class");

        Files.createDirectories(file.getParent());

        Files.copy(MODS_DIR.resolve(TEST_MODULE).resolve(file), file);

        Path res = Paths.get("jdk/test/res.properties");

        Files.createFile(res);

    }

    @DataProvider(name = "classpath")

    public Object[][] classpath() {

        return new Object[][]{

            // true indicates that class path default to current working directory

            { "",                              true  },

            { "-Djava.class.path",             true  },

            { "-Djava.class.path=",            true  },

            { "-Djava.class.path=.",           true  },

        };

    }

    @Test(dataProvider = "classpath")

    public void testUnnamedModule(String option, boolean expected) throws Throwable {

        List<String> args = new ArrayList<>();

        if (!option.isEmpty()) {

            args.add(option);

        }

        args.add(TEST_MAIN);

        args.add(Boolean.toString(expected));

        assertTrue(execute(args).getExitValue() == 0);

    }

    @DataProvider(name = "moduleAndClassPath")

    public Object[][] moduleAndClassPath() {

        return new Object[][]{

            // true indicates that class path default to current working directory

            { "",                              false  },

            { "-Djava.class.path",             false  },

            { "-Djava.class.path=",            false  },

            { "-Djava.class.path=.",           true   },

        };

    }

    @Test(dataProvider = "moduleAndClassPath")

    public void testNamedModule(String option, boolean expected) throws Throwable {

        List<String> args = new ArrayList<>();

        if (!option.isEmpty()) {

            args.add(option);

        }

        args.add("--module-path");

        args.add(MODS_DIR.toString());

        args.add("-m");

        args.add(TEST_MODULE + "/" + TEST_MAIN);

        args.add(Boolean.toString(expected));

        assertTrue(execute(args).getExitValue() == 0);

    }

    private OutputAnalyzer execute(List<String> options) throws Throwable {

        ProcessBuilder pb =

            createJavaProcessBuilder(options.toArray(new String[0]));

        Map<String,String> env = pb.environment();

        // remove CLASSPATH environment variable

        String value = env.remove("CLASSPATH");

        return executeCommand(pb)

                    .outputTo(System.out)

                    .errorTo(System.out);

    }

}