/*
* JBoss, Home of Professional Open Source.
* Copyright 2008, Red Hat Middleware LLC, and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package test.implementation.registry;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import javax.management.MBeanServer;
import javax.management.MBeanServerFactory;
import javax.management.ObjectName;
import junit.framework.TestCase;
import org.jboss.mx.server.ServerConstants;
import test.implementation.registry.support.Trivial;
/**
* Tests the value map processing in the managed mbean registry
*
* @author <a href="mailto:Adrian.Brock@HappeningTimes.com">Adrian Brock</a>.
*/
public class ValuesTestCase
extends TestCase
implements ServerConstants
{
// Attributes ----------------------------------------------------------------
// Constructor ---------------------------------------------------------------
/**
* Construct the test
*/
public ValuesTestCase(String s)
{
super(s);
}
// Tests ---------------------------------------------------------------------
/**
* Test classloader
*/
public void testClassLoader()
throws Exception
{
// Get the previous registry value
String previous = System.getProperty(MBEAN_REGISTRY_CLASS_PROPERTY);
if (previous == null)
previous = DEFAULT_MBEAN_REGISTRY_CLASS;
//System.setProperty(MBEAN_REGISTRY_CLASS_PROPERTY,
// "org.jboss.mx.server.registry.ManagedMBeanRegistry");
MBeanServer server = MBeanServerFactory.createMBeanServer();
try
{
HashMap valuesMap = new HashMap();
ClassLoader cl = Thread.currentThread().getContextClassLoader();
valuesMap.put(CLASSLOADER, Thread.currentThread().getContextClassLoader());
ObjectName mbean = new ObjectName("test:test=test");
server.invoke(new ObjectName(MBEAN_REGISTRY), "registerMBean",
new Object[]
{
new Trivial(),
mbean,
valuesMap
},
new String[]
{
Object.class.getName(),
ObjectName.class.getName(),
Map.class.getName()
}
);
Object result = server.invoke(new ObjectName(MBEAN_REGISTRY), "getValue",
new Object[]
{
mbean,
CLASSLOADER
},
new String[]
{
ObjectName.class.getName(),
String.class.getName()
}
);
assertEquals(cl, result);
}
finally
{
MBeanServerFactory.releaseMBeanServer(server);
System.setProperty(MBEAN_REGISTRY_CLASS_PROPERTY, previous);
}
}
/**
* Test value
*/
public void testValue()
throws Exception
{
// Get the previous registry value
String previous = System.getProperty(MBEAN_REGISTRY_CLASS_PROPERTY);
if (previous == null)
previous = DEFAULT_MBEAN_REGISTRY_CLASS;
//System.setProperty(MBEAN_REGISTRY_CLASS_PROPERTY,
// "org.jboss.mx.server.registry.ManagedMBeanRegistry");
MBeanServer server = MBeanServerFactory.createMBeanServer();
try
{
HashMap valuesMap = new HashMap();
Date date = new Date(System.currentTimeMillis());
valuesMap.put("date", date);
ObjectName mbean = new ObjectName("test:test=test");
server.invoke(new ObjectName(MBEAN_REGISTRY), "registerMBean",
new Object[]
{
new Trivial(),
mbean,
valuesMap
},
new String[]
{
Object.class.getName(),
ObjectName.class.getName(),
Map.class.getName()
}
);
Object result = server.invoke(new ObjectName(MBEAN_REGISTRY), "getValue",
new Object[]
{
mbean,
"date"
},
new String[]
{
ObjectName.class.getName(),
String.class.getName()
}
);
assertEquals(date, result);
}
finally
{
MBeanServerFactory.releaseMBeanServer(server);
System.setProperty(MBEAN_REGISTRY_CLASS_PROPERTY, previous);
}
}
/**
* Test value registered
*/
public void testValueRegistered()
throws Exception
{
// Get the previous registry value
String previous = System.getProperty(MBEAN_REGISTRY_CLASS_PROPERTY);
if (previous == null)
previous = DEFAULT_MBEAN_REGISTRY_CLASS;
//System.setProperty(MBEAN_REGISTRY_CLASS_PROPERTY,
// "org.jboss.mx.server.registry.ManagedMBeanRegistry");
MBeanServer server = MBeanServerFactory.createMBeanServer();
try
{
HashMap valuesMap = new HashMap();
Date date = new Date(System.currentTimeMillis());
valuesMap.put("date", date);
ObjectName mbean = new ObjectName("test:test=test");
server.invoke(new ObjectName(MBEAN_REGISTRY), "registerMBean",
new Object[]
{
new Trivial(),
mbean,
valuesMap
},
new String[]
{
Object.class.getName(),
ObjectName.class.getName(),
Map.class.getName()
}
);
Object result = server.invoke(new ObjectName(MBEAN_REGISTRY), "getValue",
new Object[]
{
mbean,
"date"
},
new String[]
{
ObjectName.class.getName(),
String.class.getName()
}
);
assertEquals(date, result);
// Now remove it, reregister it and make sure it returns the new value
server.unregisterMBean(mbean);
Thread.sleep(2);
date = new Date(System.currentTimeMillis());
HashMap valueMap2 = new HashMap();
valueMap2.put("date", date);
server.invoke(new ObjectName(MBEAN_REGISTRY), "registerMBean",
new Object[]
{
new Trivial(),
mbean,
valueMap2
},
new String[]
{
Object.class.getName(),
ObjectName.class.getName(),
Map.class.getName()
}
);
result = server.invoke(new ObjectName(MBEAN_REGISTRY), "getValue",
new Object[]
{
mbean,
"date"
},
new String[]
{
ObjectName.class.getName(),
String.class.getName()
}
);
assertEquals(date, result);
}
finally
{
MBeanServerFactory.releaseMBeanServer(server);
System.setProperty(MBEAN_REGISTRY_CLASS_PROPERTY, previous);
}
}
}