| /******************************************************************************* |
| * Copyright (c) 2011 VMware Inc. |
| * All rights reserved. This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License v1.0 |
| * which accompanies this distribution, and is available at |
| * http://www.eclipse.org/legal/epl-v10.html |
| * |
| * Contributors: |
| * VMware Inc. - initial contribution |
| *******************************************************************************/ |
| package org.eclipse.gemini.management.integration.tests; |
| |
| import java.lang.management.ManagementFactory; |
| |
| import javax.management.MBeanServer; |
| import javax.management.MBeanServerConnection; |
| import javax.management.ObjectName; |
| import javax.management.remote.JMXConnector; |
| import javax.management.remote.JMXConnectorFactory; |
| import javax.management.remote.JMXServiceURL; |
| |
| import org.eclipse.gemini.management.Activator; |
| import org.junit.BeforeClass; |
| import org.osgi.framework.BundleContext; |
| import org.osgi.framework.Constants; |
| import org.osgi.framework.FrameworkUtil; |
| import org.osgi.framework.ServiceReference; |
| |
| public abstract class AbstractOSGiMBeanTest { |
| |
| private static BundleContext bc; |
| |
| protected String mBeanObjectName; |
| |
| protected boolean addFrameworkAndUUID = false; |
| |
| @BeforeClass |
| public static void setup(){ |
| BundleContext bc = FrameworkUtil.getBundle(Activator.class).getBundleContext(); |
| AbstractOSGiMBeanTest.bc = bc; |
| ServiceReference<MBeanServer> ref = bc.getServiceReference(MBeanServer.class); |
| if (ref == null) { |
| bc.registerService(MBeanServer.class.getCanonicalName(), ManagementFactory.getPlatformMBeanServer(), null); |
| } |
| } |
| |
| protected <T> T jmxFetchData(String operation, Object[] arguments, String[] types, Class<T> returnType) throws Exception { |
| JMXConnector connector; |
| String url = "service:jmx:rmi:///jndi/rmi://localhost:21045/jmxrmi"; |
| JMXServiceURL jmxURL = new JMXServiceURL(url); |
| connector = JMXConnectorFactory.connect(jmxURL); |
| MBeanServerConnection connection = connector.getMBeanServerConnection(); |
| ObjectName name = new ObjectName(getMBeanObjectName()); |
| Object result = connection.invoke(name, operation, arguments, types); |
| return returnType.cast(result); |
| } |
| |
| protected <T> T jmxFetchAttribute(String attribute, Class<T> returnType) throws Exception { |
| JMXConnector connector; |
| String url = "service:jmx:rmi:///jndi/rmi://localhost:21045/jmxrmi"; |
| JMXServiceURL jmxURL = new JMXServiceURL(url); |
| connector = JMXConnectorFactory.connect(jmxURL); |
| MBeanServerConnection connection = connector.getMBeanServerConnection(); |
| ObjectName name = new ObjectName(getMBeanObjectName()); |
| Object result = connection.getAttribute(name, attribute); |
| return returnType.cast(result); |
| } |
| |
| private String getMBeanObjectName(){ |
| String newObjectName = this.mBeanObjectName; |
| if(this.addFrameworkAndUUID){ |
| newObjectName = newObjectName + |
| ",framework=" + AbstractOSGiMBeanTest.bc.getBundle(0).getSymbolicName() + |
| ",uuid=" + AbstractOSGiMBeanTest.bc.getBundle(0).getBundleContext().getProperty(Constants.FRAMEWORK_UUID); |
| } |
| return newObjectName; |
| } |
| |
| } |