/*

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

 * @summary Checks access to test MXBean

 * @author Olivier Lagneau

 * @modules java.management.rmi

 * @library /lib/testlibrary

 * @compile Basic.java

 * @run main/othervm/timeout=300 -DDEBUG_STANDARD MXBeanInteropTest2

 */

import java.util.Iterator;

import java.util.Map;

import java.util.Set;

import javax.management.Attribute;

import javax.management.JMX;

import javax.management.MBeanAttributeInfo;

import javax.management.MBeanConstructorInfo;

import javax.management.MBeanServer;

import java.lang.management.ManagementFactory;

import javax.management.MBeanInfo;

import javax.management.MBeanNotificationInfo;

import javax.management.MBeanOperationInfo;

import javax.management.MBeanServerConnection;

import javax.management.ObjectName;

import javax.management.remote.JMXConnector;

import javax.management.remote.JMXConnectorFactory;

import javax.management.remote.JMXConnectorServer;

import javax.management.remote.JMXConnectorServerFactory;

import javax.management.remote.JMXServiceURL;

public class MXBeanInteropTest2 {

    private static String BASIC_MXBEAN_CLASS_NAME = "Basic";

    /*

     * First Debug properties and arguments are collect in expected

     * map  (argName, value) format, then calls original test's run method.

     */

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

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

        // Parses parameters

        Utils.parseDebugProperties();

        Map<String, Object> map = Utils.parseParameters(args) ;

        // Run test

        MXBeanInteropTest2 test = new MXBeanInteropTest2();

        test.run(map);

    }

    public void run(Map<String, Object> args) {

        System.out.println("MXBeanInteropTest2::run: Start") ;

        int errorCount = 0 ;

        try {

            // JMX MbeanServer used inside single VM as if remote.

            MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();

            JMXServiceURL url = new JMXServiceURL("rmi", null, 0);

            JMXConnectorServer cs =

                JMXConnectorServerFactory.newJMXConnectorServer(url, null, mbs);

            cs.start();

            JMXServiceURL addr = cs.getAddress();

            JMXConnector cc = JMXConnectorFactory.connect(addr);

            MBeanServerConnection mbsc = cc.getMBeanServerConnection();

            // Prints all MBeans whatever the domain is.

            printMBeans(mbsc) ;

            // Call test body

            errorCount += doBasicMXBeanTest(mbsc) ;

            // Terminate the JMX Client

            cc.close();

        } catch(Exception e) {

            Utils.printThrowable(e, true) ;

            throw new RuntimeException(e);

        }

        if ( errorCount == 0 ) {

            System.out.println("MXBeanInteropTest2::run: Done without any error") ;

        } else {

            System.out.println("MXBeanInteropTest2::run: Done with "

                    + errorCount

                    + " error(s)") ;

            throw new RuntimeException("errorCount = " + errorCount);

        }

    }

    /**

     * Prints all MBeans whatever the domain is.

     */

    private static void printMBeans(MBeanServerConnection mbsc) throws Exception {

        Set<ObjectName> set = mbsc.queryNames(null, null);

        System.out.println("---- MBeans found :");

        for (Iterator<ObjectName> iter = set.iterator(); iter.hasNext(); ) {

            System.out.println(iter.next().toString());

        }

        System.out.println("\n") ;

    }

    private final int doBasicMXBeanTest(MBeanServerConnection mbsc) {

        int errorCount = 0 ;

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

        try {

            ObjectName objName =

                    new ObjectName("sqe:type=BasicMXBean") ;

            mbsc.createMBean(BASIC_MXBEAN_CLASS_NAME, objName);

            MBeanInfo mbInfo = mbsc.getMBeanInfo(objName);

            printMBeanInfo(mbInfo);

            System.out.println("---- OK\n") ;

            System.out.println("getMBeanInfo\t\t"

                    + mbInfo);

            System.out.println("---- OK\n") ;

            System.out.println("Check mxbean field in the MBeanInfo");

            String mxbeanField =

                    (String)mbInfo.getDescriptor().getFieldValue(JMX.MXBEAN_FIELD);

            if ( mxbeanField == null || ! mxbeanField.equals("true")) {

                System.out.println("---- ERROR : Improper mxbean field value "

                        + mxbeanField);

                errorCount++;

            }

            System.out.println("---- OK\n") ;

            System.out.println("Set attribute ObjectNameAtt");

            Attribute att = new Attribute("ObjectNameAtt", objName);

            mbsc.setAttribute(objName, att);

            ObjectName value =

                    (ObjectName)mbsc.getAttribute(objName, "ObjectNameAtt");

            if ( ! value.equals(objName) ) {

                errorCount++;

                System.out.println("---- ERROR : setAttribute failed, got "

                        + value

                        + " while expecting "

                        + objName);

            }

            System.out.println("---- OK\n") ;

            System.out.println("Call operation doNothing");

            mbsc.invoke(objName,  "doNothing", null, null);

            System.out.println("---- OK\n") ;

            System.out.println("Call operation getWeather");

            Object weather = mbsc.invoke(objName,

                    "getWeather",

                    new Object[]{Boolean.TRUE},

                    new String[]{"boolean"});

            System.out.println("Weather is " + weather);

            System.out.println("---- OK\n") ;

        } catch (Exception e) {

            Utils.printThrowable(e, true) ;

            errorCount++ ;

            System.out.println("---- ERROR\n") ;

        }

        return errorCount ;

    }

    private void printMBeanInfo(MBeanInfo mbInfo) {

        System.out.println("Description " + mbInfo.getDescription());

        for (MBeanConstructorInfo ctor : mbInfo.getConstructors()) {

            System.out.println("Constructor " + ctor.getName());

        }

        for (MBeanAttributeInfo att : mbInfo.getAttributes()) {

            System.out.println("Attribute " + att.getName()

            + " [" + att.getType() + "]");

        }

        for (MBeanOperationInfo oper : mbInfo.getOperations()) {

            System.out.println("Operation " + oper.getName());

        }

        for (MBeanNotificationInfo notif : mbInfo.getNotifications()) {

            System.out.println("Notification " + notif.getName());

        }

    }

}