/*

 * Copyright 2008 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 */

/*

 * @test %M% %I%

 * @bug 6675526

 * @summary Test MBeans named with @ObjectNameTemplate

 * @author Jean-Francois Denise

 * @run main/othervm ObjectNameTemplateTest

 */

import java.lang.management.ManagementFactory;

import java.lang.reflect.Constructor;

import java.util.ArrayList;

import java.util.List;

import javax.management.Attribute;

import javax.management.AttributeList;

import javax.management.AttributeNotFoundException;

import javax.management.DynamicMBean;

import javax.management.ImmutableDescriptor;

import javax.management.InvalidAttributeValueException;

import javax.management.JMX;

import javax.management.MBeanException;

import javax.management.MBeanInfo;

import javax.management.MBeanServer;

import javax.management.MXBean;

import javax.management.MBean;

import javax.management.ManagedAttribute;

import javax.management.NotCompliantMBeanException;

import javax.management.ObjectName;

import javax.management.ObjectNameTemplate;

import javax.management.ReflectionException;

import javax.management.StandardMBean;

public class ObjectNameTemplateTest {

    private static MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();

    private static final String NAME_TEMPLATE_MULTI =

            "com.example:type=MultiStdCache,name={Name}";

    private static final String NAME_TEMPLATE_MONO =

            "com.example:{Type}={TypeValue}";

    private static final String NAME_TEMPLATE_QUOTED =

            "com.example:type=Quotted,name=\"{Name}\"";

    private static final String NAME_TEMPLATE_WRAPPED =

            "com.example:type=MgtInterface,id={Id}";

    private static final String NAME_TEMPLATE_FULL =

            "{Naming}";

    private static final String FULL_NAME = "com.example:type=NotAdvised";

    private static final String NAME1 = "toto1";

    private static final String NAME2 = "toto2";

    private static final String TYPE_KEY = "thisIsTheType";

    private static final String TYPE_VALUE = "aTypeValue";

    private static final String INVALID_NAME = "?,=*,\n, ";

    private static final int ID = 999;

    private static Object[] EMPTY_PARAMS = {};

    private static String[] EMPTY_SIGNATURE = {};

    private static final ObjectName OBJECTNAME_CACHE =

            ObjectName.valueOf("com.example:type=Cache");

    private static final ObjectName OBJECTNAME_SUBCACHE =

            ObjectName.valueOf("com.example:type=SubCache");

    private static final ObjectName OBJECTNAME_CACHEMX =

            ObjectName.valueOf("com.example:type=CacheMX");

    private static final ObjectName OBJECTNAME_SUBCACHEMX =

            ObjectName.valueOf("com.example:type=SubCacheMX");

    private static final ObjectName OBJECTNAME_DYNACACHE =

            ObjectName.valueOf("com.example:type=DynaCache");

    private static final ObjectName OBJECTNAME_STDCACHE =

            ObjectName.valueOf("com.example:type=StdCache");

    private static final ObjectName OBJECTNAME_STDCACHEMX =

            ObjectName.valueOf("com.example:type=StdCacheMX");

    private static final ObjectName OBJECTNAME_MULTI_1 =

            ObjectName.valueOf("com.example:" +

            "type=MultiStdCache,name=" + NAME1);

    private static final ObjectName OBJECTNAME_MULTI_2 =

            ObjectName.valueOf("com.example:" +

            "type=MultiStdCache,name=" + NAME2);

    private static final ObjectName OBJECTNAME_MONO =

            ObjectName.valueOf("com.example:" + TYPE_KEY + "=" +

            TYPE_VALUE);

    private static final ObjectName OBJECTNAME_QUOTED =

            ObjectName.valueOf("com.example:type=Quotted," +

            "name="+ObjectName.quote(INVALID_NAME));

    private static final ObjectName OBJECTNAME_WRAPPED_RESOURCE =

            ObjectName.valueOf("com.example:type=MgtInterface,id=" + ID);

    private static final ObjectName OBJECTNAME_FULL =

            ObjectName.valueOf(FULL_NAME);

    private static void test(Class<?> mbean, Object[] params,

            String[] signature, ObjectName name, String template)

            throws Exception {

        mbs.createMBean(mbean.getName(), null, params, signature);

        test(name, template);

        List<Class<?>> parameters = new ArrayList<Class<?>>();

        for (String sig : signature) {

            parameters.add(Class.forName(sig));

        }

        Class<?> classes[] = new Class<?>[parameters.size()];

        Constructor ctr = mbean.getConstructor(parameters.toArray(classes));

        Object inst = ctr.newInstance(params);

        test(inst, name, template);

    }

    private static void test(Object obj, ObjectName name, String template)

            throws Exception {

        mbs.registerMBean(obj, null);

        test(name, template);

    }

    private static void test(ObjectName name, String template)

            throws Exception {

        if (!mbs.isRegistered(name)) {

            throw new Exception("Wrong " + name + " name");

        }

        if (template != null && !mbs.getMBeanInfo(name).getDescriptor().

                getFieldValue("objectNameTemplate").equals(template)) {

            throw new Exception("Invalid Derscriptor");

        }

        mbs.unregisterMBean(name);

    }

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

        test(Cache.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_CACHE,

                OBJECTNAME_CACHE.toString());

        test(CacheMX.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_CACHEMX,

                OBJECTNAME_CACHEMX.toString());

        test(SubCache.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_SUBCACHE,

                OBJECTNAME_SUBCACHE.toString());

        test(SubCacheMX.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_SUBCACHEMX,

                OBJECTNAME_SUBCACHEMX.toString());

        test(DynaCache.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_DYNACACHE,

                null);

        test(StdCacheMX.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_STDCACHEMX,

                OBJECTNAME_STDCACHEMX.toString());

        test(StdCache.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_STDCACHE,

                OBJECTNAME_STDCACHE.toString());

        String[] sig = {String.class.getName()};

        Object[] params = {NAME1};

        test(MultiStdCache.class, params, sig, OBJECTNAME_MULTI_1,

                NAME_TEMPLATE_MULTI);

        Object[] params2 = {NAME2};

        test(MultiStdCache.class, params2, sig, OBJECTNAME_MULTI_2,

                NAME_TEMPLATE_MULTI);

        test(MonoStdCache.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_MONO,

                NAME_TEMPLATE_MONO);

        test(Quoted.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_QUOTED,

                NAME_TEMPLATE_QUOTED);

        test(new StandardMBean(new WrappedResource(), MgtInterface.class),

                OBJECTNAME_WRAPPED_RESOURCE, NAME_TEMPLATE_WRAPPED);

        test(FullName.class, EMPTY_PARAMS, EMPTY_SIGNATURE, OBJECTNAME_FULL,

                NAME_TEMPLATE_FULL);

        try {

            test(Wrong.class, EMPTY_PARAMS, EMPTY_SIGNATURE, null, null);

            throw new Exception("No treceived expected Exception");

        } catch (NotCompliantMBeanException ncex) {

            if (!(ncex.getCause() instanceof AttributeNotFoundException)) {

                throw new Exception("Invalid initCause");

            }

        }

    }

    @MBean

    @ObjectNameTemplate("{Naming}")

    public static class FullName {

        @ManagedAttribute

        public String getNaming() {

            return FULL_NAME;

        }

    }

    @ObjectNameTemplate("com.example:type=MgtInterface,id={Id}")

    public interface MgtInterface {

        public int getId();

    }

    public static class WrappedResource implements MgtInterface {

        public int getId() {

            return ID;

        }

    }

    @MBean

    @ObjectNameTemplate("com.example:type=Cache")

    public static class Cache {

    }

    @ObjectNameTemplate("com.example:type=SubCache")

    public static class SubCache extends Cache {

    }

    @MXBean

    @ObjectNameTemplate("com.example:type=CacheMX")

    public static class CacheMX {

    }

    @ObjectNameTemplate("com.example:type=SubCacheMX")

    public static class SubCacheMX extends CacheMX {

    }

    @ObjectNameTemplate("com.example:type=StdCache")

    public interface StdCacheMBean {

    }

    public static class StdCache implements StdCacheMBean {

    }

    @ObjectNameTemplate("com.example:type=StdCacheMX")

    public interface StdCacheMXBean {

    }

    public static class StdCacheMX implements StdCacheMXBean {

    }

    public static class DynaCache implements DynamicMBean {

        public Object getAttribute(String attribute) throws AttributeNotFoundException, MBeanException, ReflectionException {

            throw new UnsupportedOperationException("Not supported yet.");

        }

        public void setAttribute(Attribute attribute) throws AttributeNotFoundException, InvalidAttributeValueException, MBeanException, ReflectionException {

            throw new UnsupportedOperationException("Not supported yet.");

        }

        public AttributeList getAttributes(String[] attributes) {

            throw new UnsupportedOperationException("Not supported yet.");

        }

        public AttributeList setAttributes(AttributeList attributes) {

            throw new UnsupportedOperationException("Not supported yet.");

        }

        public Object invoke(String actionName, Object[] params, String[] signature) throws MBeanException, ReflectionException {

            throw new UnsupportedOperationException("Not supported yet.");

        }

        public MBeanInfo getMBeanInfo() {

            ImmutableDescriptor d = new ImmutableDescriptor(JMX.OBJECT_NAME_TEMPLATE + "=com.example:type=DynaCache");

            return new MBeanInfo("DynaCache", "Description", null, null, null, null, d);

        }

    }

    @ObjectNameTemplate("com.example:type=MultiStdCache,name={Name}")

    public interface MultiStdCacheMXBean {

        public String getName();

    }

    public static class MultiStdCache implements MultiStdCacheMXBean {

        private String name;

        public MultiStdCache(String name) {

            this.name = name;

        }

        public String getName() {

            return name;

        }

    }

    @ObjectNameTemplate("com.example:{Type}={TypeValue}")

    public interface MonoStdCacheMXBean {

        public String getTypeValue();

        public String getType();

    }

    public static class MonoStdCache implements MonoStdCacheMXBean {

        public String getTypeValue() {

            return TYPE_VALUE;

        }

        public String getType() {

            return TYPE_KEY;

        }

    }

    @ObjectNameTemplate("com.example:type=Quotted,name=\"{Name}\"")

    public interface QuottedMXBean {

        public String getName();

    }

    public static class Quoted implements QuottedMXBean {

        public String getName() {

            return INVALID_NAME;

        }

    }

    @ObjectNameTemplate("com.example:{Type}={TypeValue}, name={Name}")

    public interface WrongMXBean {

        public String getTypeValue();

        public String getType();

    }

    public static class Wrong implements WrongMXBean {

        public String getTypeValue() {

            return TYPE_VALUE;

        }

        public String getType() {

            return TYPE_KEY;

        }

    }

}