/*

 * Copyright (c) 2005, 2015, 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 4716807

 * @summary Test if the repository supports correctly the use of

 *          wildcards in the ObjectName key properties value part.

 * @author Luis-Miguel Alventosa

 * @modules java.management

 * @run clean RepositoryWildcardTest

 * @run build RepositoryWildcardTest

 * @run main RepositoryWildcardTest

 */

import java.util.HashSet;

import java.util.Set;

import java.util.TreeSet;

import javax.management.InstanceNotFoundException;

import javax.management.MBeanServer;

import javax.management.MBeanServerDelegate;

import javax.management.MBeanServerFactory;

import javax.management.ObjectName;

import javax.management.RuntimeOperationsException;

public class RepositoryWildcardTest {

    private static final String classname =

            "javax.management.monitor.StringMonitor";

    private static int mbeanCreation(MBeanServer mbs, String name)

        throws Exception {

        int error = 0;

        try {

            System.out.println("Test: createMBean(" + name + ")");

            mbs.createMBean(classname, ObjectName.getInstance(name));

            error++;

            System.out.println("Didn't get expected exception!");

            System.out.println("Test failed!");

        } catch (RuntimeOperationsException e) {

            System.out.println("Got expected exception = " +

                               e.getCause().toString());

            System.out.println("Test passed!");

        }

        return error;

    }

    private static int mbeanDeletion(MBeanServer mbs, String name)

        throws Exception {

        int error = 0;

        try {

            System.out.println("Test: unregisterMBean(" + name + ")");

            mbs.unregisterMBean(ObjectName.getInstance(name));

            error++;

            System.out.println("Didn't get expected exception!");

            System.out.println("Test failed!");

        } catch (InstanceNotFoundException e) {

            System.out.println("Got expected exception = " + e.toString());

            System.out.println("Test passed!");

        }

        return error;

    }

    private static int mbeanQuery(MBeanServer mbs,

                                  String name,

                                  Set<ObjectName> expectedSet)

        throws Exception {

        int error = 0;

        System.out.println("Test: queryNames(" + name + ")");

        Set<ObjectName> returnedSet =

                mbs.queryNames(ObjectName.getInstance(name), null);

        System.out.println("ReturnedSet = " + new TreeSet(returnedSet));

        System.out.println("ExpectedSet = " + new TreeSet(expectedSet));

        if (returnedSet.equals(expectedSet)) {

            System.out.println("Test passed!");

        } else {

            error++;

            System.out.println("Test failed!");

        }

        return error;

    }

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

        int error = 0;

        MBeanServer mbs = MBeanServerFactory.newMBeanServer();

        ObjectName[] namesArray = {

            ObjectName.getInstance("d:k=abc"),

            ObjectName.getInstance("d:k=abcd"),

            ObjectName.getInstance("d:k=abcde"),

            ObjectName.getInstance("d:k=abc,k2=v2"),

            ObjectName.getInstance("d:k=abcd,k2=v2"),

            ObjectName.getInstance("d:k=abcde,k2=v2"),

            ObjectName.getInstance("d:k=\"abc\""),

            ObjectName.getInstance("d:k=\"abc\",k2=v2"),

            ObjectName.getInstance("d:p1=v1,p2=v2,p3=v3,p4=v4,p5=v5,p6=v6," +

                                   "p7=v7,p8=v8,p9=v9,p10=v10,p11=v11,p12=v12")

        };

        for (ObjectName name : namesArray)

            mbs.createMBean(classname, name);

        System.out.println("----------------------------------------------");

        System.out.println("TEST createMBean WITH PATTERNS");

        System.out.println("----------------------------------------------");

        error += mbeanCreation(mbs, "d:k=v,*");

        error += mbeanCreation(mbs, "d:k=*");

        error += mbeanCreation(mbs, "d:k=a?b");

        error += mbeanCreation(mbs, "d:k=\"?\"");

        error += mbeanCreation(mbs, "d:k=\"a*b\"");

        System.out.println("----------------------------------------------");

        System.out.println("TEST queryNames WITH PATTERNS");

        System.out.println("----------------------------------------------");

        Set<ObjectName> expectedSet = new HashSet<ObjectName>();

        for (ObjectName name : namesArray)

            expectedSet.add(name);

        expectedSet.add(MBeanServerDelegate.DELEGATE_NAME);

        Set<ObjectName> returnedSet =

            mbs.queryNames(ObjectName.getInstance("*:*"), null);

        error += mbeanQuery(mbs, "*:*", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        for (ObjectName name : namesArray)

            expectedSet.add(name);

        error += mbeanQuery(mbs, "d:*", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        expectedSet.add(namesArray[0]);

        expectedSet.add(namesArray[1]);

        expectedSet.add(namesArray[2]);

        expectedSet.add(namesArray[6]);

        error += mbeanQuery(mbs, "d:k=*", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        expectedSet.add(namesArray[3]);

        expectedSet.add(namesArray[4]);

        expectedSet.add(namesArray[5]);

        expectedSet.add(namesArray[7]);

        error += mbeanQuery(mbs, "d:k=*,k2=v2", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        expectedSet.add(namesArray[3]);

        expectedSet.add(namesArray[4]);

        expectedSet.add(namesArray[5]);

        expectedSet.add(namesArray[7]);

        error += mbeanQuery(mbs, "d:k=*,k2=v?", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        expectedSet.add(namesArray[3]);

        expectedSet.add(namesArray[4]);

        expectedSet.add(namesArray[5]);

        expectedSet.add(namesArray[7]);

        error += mbeanQuery(mbs, "d:*,k2=v2", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        expectedSet.add(namesArray[1]);

        error += mbeanQuery(mbs, "d:k=ab?d", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        expectedSet.add(namesArray[1]);

        expectedSet.add(namesArray[4]);

        error += mbeanQuery(mbs, "d:k=ab?d,*", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        expectedSet.add(namesArray[6]);

        error += mbeanQuery(mbs, "d:k=\"*\"", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        expectedSet.add(namesArray[7]);

        error += mbeanQuery(mbs, "d:k=\"*\",k2=v?", expectedSet);

        expectedSet = new HashSet<ObjectName>();

        expectedSet.add(namesArray[8]);

        error += mbeanQuery(mbs,

                            "d:p1=v?,p2=v?,p3=v?,p4=v?,p5=v?,p6=v?," +

                            "p7=v?,p8=v?,p9=v?,p10=v??,p11=v??,p12=v??",

                            expectedSet);

        System.out.println("----------------------------------------------");

        System.out.println("TEST unregisterMBean WITH PATTERNS");

        System.out.println("----------------------------------------------");

        error += mbeanDeletion(mbs, "d:k=*");

        error += mbeanDeletion(mbs, "d:k=\"*\"");

        if (error > 0) {

            final String msg = "Test FAILED! Got " + error + " error(s)";

            System.out.println(msg);

            throw new IllegalArgumentException(msg);

        } else {

            System.out.println("Test PASSED!");

        }

    }

}