/*

 * Copyright (c) 2016, 2017, 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

 * @modules java.base/jdk.internal.misc

 * @library /test/lib ..

 * @build sun.hotspot.WhiteBox

 * @compile/module=java.base java/lang/ModuleHelper.java

 * @run main ClassFileInstaller sun.hotspot.WhiteBox

 *                              sun.hotspot.WhiteBox$WhiteBoxPermission

 * @run main/othervm -Xbootclasspath/a:. -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI JVMAddModulePackage

 */

import static jdk.test.lib.Asserts.*;

import java.sql.Time;

public class JVMAddModulePackage {

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

        MyClassLoader cl1 = new MyClassLoader();

        MyClassLoader cl3 = new MyClassLoader();

        Object module_one, module_two, module_three;

        boolean result;

        module_one = ModuleHelper.ModuleObject("module_one", cl1, new String[] { "mypackage" });

        assertNotNull(module_one, "Module should not be null");

        module_two = ModuleHelper.ModuleObject("module_two", cl1, new String[] { "yourpackage" });

        assertNotNull(module_two, "Module should not be null");

        module_three = ModuleHelper.ModuleObject("module_three", cl3, new String[] { "package/num3" });

        assertNotNull(module_three, "Module should not be null");

        // Simple call

        ModuleHelper.AddModulePackage(module_one, "new_package");

        // Add a package and export it

        ModuleHelper.AddModulePackage(module_one, "package/num3");

        ModuleHelper.AddModuleExportsToAll(module_one, "package/num3");

        // Null module argument, expect an NPE

        try {

            ModuleHelper.AddModulePackage(null, "new_package");

            throw new RuntimeException("Failed to get the expected NPE");

        } catch(NullPointerException e) {

            // Expected

        }

        // Bad module argument, expect an IAE

        try {

            ModuleHelper.AddModulePackage(cl1, "new_package");

            throw new RuntimeException("Failed to get the expected IAE");

        } catch(IllegalArgumentException e) {

            // Expected

        }

        // Null package argument, expect an NPE

        try {

            ModuleHelper.AddModulePackage(module_one, null);

            throw new RuntimeException("Failed to get the expected NPE");

        } catch(NullPointerException e) {

            // Expected

        }

        // Existing package, expect an ISE

        try {

            ModuleHelper.AddModulePackage(module_one, "yourpackage");

            throw new RuntimeException("Failed to get the expected ISE");

        } catch(IllegalStateException e) {

            // Expected

        }

        // Invalid package name, expect an IAE

        try {

            ModuleHelper.AddModulePackage(module_one, "your.package");

            throw new RuntimeException("Failed to get the expected IAE");

        } catch(IllegalArgumentException e) {

            // Expected

        }

        // Invalid package name, expect an IAE

        try {

            ModuleHelper.AddModulePackage(module_one, ";your/package");

            throw new RuntimeException("Failed to get the expected IAE");

        } catch(IllegalArgumentException e) {

            // Expected

        }

        // Invalid package name, expect an IAE

        try {

            ModuleHelper.AddModulePackage(module_one, "7[743");

            throw new RuntimeException("Failed to get the expected IAE");

        } catch(IllegalArgumentException e) {

            // Expected

        }

        // Empty package name, expect an IAE

        try {

            ModuleHelper.AddModulePackage(module_one, "");

            throw new RuntimeException("Failed to get the expected IAE");

        } catch(IllegalArgumentException e) {

            // Expected

        }

        // Add package named "java" to an module defined to a class loader other than the boot or platform loader.

        try {

            // module_one is defined to a MyClassLoader class loader.

            ModuleHelper.AddModulePackage(module_one, "java/foo");

            throw new RuntimeException("Failed to get the expected IAE");

        } catch(IllegalArgumentException e) {

            if (!e.getMessage().contains("prohibited package name")) {

              throw new RuntimeException("Failed to get expected IAE message for prohibited package name: " + e.getMessage());

            }

        }

        // Package "javabar" should be ok

        ModuleHelper.AddModulePackage(module_one, "javabar");

        // Package named "java" defined to the boot class loader, should be ok

        Object module_javabase = module_one.getClass().getModule();

        ModuleHelper.AddModulePackage(module_javabase, "java/foo");

        // Package named "java" defined to the platform class loader, should be ok

        // The module java.sql is defined to the platform class loader.

        java.sql.Time jst = new java.sql.Time(45000); // milliseconds

        Object module_javasql = jst.getClass().getModule();

        ModuleHelper.AddModulePackage(module_javasql, "java/foo");

    }

    static class MyClassLoader extends ClassLoader { }

}