/*
 * 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
 * @bug 5072267
 * @summary Test that a context forwarder can be created and then installed.
 * @author Eamonn McManus
 */

/* The specific thing we're testing for is that the forwarder can be created
 * with a null "next", and then installed with a real "next".  An earlier
 * defect meant that in this case the simulated jmx.context// namespace had a
 * null handler that never changed.
 */

import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;
import javax.management.ClientContext;
import javax.management.MBeanServer;
import javax.management.MBeanServerConnection;
import javax.management.MBeanServerFactory;
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;
import javax.management.remote.MBeanServerForwarder;

public class ContextForwarderTest {
    private static String failure;

    public static void main(String[] args) throws Exception {
        MBeanServer mbs = MBeanServerFactory.newMBeanServer();
        JMXServiceURL url = new JMXServiceURL("service:jmx:rmi://");
        Map<String, String> env = new HashMap<String, String>();
        env.put(JMXConnectorServer.CONTEXT_FORWARDER, "false");
        JMXConnectorServer cs = JMXConnectorServerFactory.newJMXConnectorServer(
                url, env, mbs);
        MBeanServerForwarder sysMBSF = cs.getSystemMBeanServerForwarder();
        MBeanServerForwarder mbsf = ClientContext.newContextForwarder(mbs, sysMBSF);
        sysMBSF.setMBeanServer(mbsf);

        int localCount = mbs.getMBeanCount();

        cs.start();
        try {
            JMXConnector cc = JMXConnectorFactory.connect(cs.getAddress());
            MBeanServerConnection mbsc = cc.getMBeanServerConnection();
            mbsc = ClientContext.withContext(mbsc, "foo", "bar");
            int contextCount = mbsc.getMBeanCount();
            if (localCount + 1 != contextCount) {
                fail("Local MBean count %d, context MBean count %d",
                        localCount, contextCount);
            }
            Set<ObjectName> localNames =
                    new TreeSet<ObjectName>(mbs.queryNames(null, null));
            ObjectName contextNamespaceObjectName =
                    new ObjectName(ClientContext.NAMESPACE + "//:type=JMXNamespace");
            if (!localNames.add(contextNamespaceObjectName))
                fail("Local names already contained context namespace handler");
            Set<ObjectName> contextNames = mbsc.queryNames(null, null);
            if (!localNames.equals(contextNames)) {
                fail("Name set differs locally and in context: " +
                        "local: %s; context: %s", localNames, contextNames);
            }
        } finally {
            cs.stop();
        }
        if (failure != null)
            throw new Exception("TEST FAILED: " + failure);
        else
            System.out.println("TEST PASSED");
    }

    private static void fail(String msg, Object... params) {
        failure = String.format(msg, params);
        System.out.println("FAIL: " + failure);
    }
}