/*
 * 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);
    }

}