AbstractOSGiMBeanTest.java

/*******************************************************************************
 * 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;
	}
	
}