/*
* JBoss, Home of Professional Open Source.
* Copyright 2011, 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 org.jboss.as.jmx;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.ADD;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.DESCRIBE;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.OP;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.OP_ADDR;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.SUBSYSTEM;
import java.util.List;
import javax.management.MBeanServerConnection;
import javax.management.remote.JMXConnectorFactory;
import javax.management.remote.JMXServiceURL;
import javax.xml.stream.XMLStreamException;
import junit.framework.Assert;
import org.jboss.as.controller.PathAddress;
import org.jboss.as.controller.PathElement;
import org.jboss.as.controller.extension.ExtensionRegistry;
import org.jboss.as.controller.registry.ManagementResourceRegistration;
import org.jboss.as.controller.registry.Resource;
import org.jboss.as.network.SocketBinding;
import org.jboss.as.remoting.EndpointService;
import org.jboss.as.remoting.RemotingServices;
import org.jboss.as.remoting.management.ManagementRemotingServices;
import org.jboss.as.subsystem.test.AbstractSubsystemTest;
import org.jboss.as.subsystem.test.AdditionalInitialization;
import org.jboss.as.subsystem.test.ControllerInitializer;
import org.jboss.as.subsystem.test.KernelServices;
import org.jboss.dmr.ModelNode;
import org.jboss.msc.service.ServiceName;
import org.jboss.msc.service.ServiceTarget;
import org.junit.Test;
import org.xnio.OptionMap;
/**
*
* @author <a href="kabir.khan@jboss.com">Kabir Khan</a>
*/
public class JMXSubsystemTestCase extends AbstractSubsystemTest {
private static final String LAUNCH_TYPE = "launch-type";
private static final String TYPE_STANDALONE = "STANDALONE";
public JMXSubsystemTestCase() {
super(JMXExtension.SUBSYSTEM_NAME, new JMXExtension());
}
@Test
public void testParseEmptySubsystem() throws Exception {
//Parse the subsystem xml into operations
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.CURRENT.getUriString() + "\">" +
"</subsystem>";
List<ModelNode> operations = super.parse(subsystemXml);
///Check that we have the expected number of operations
Assert.assertEquals(1, operations.size());
//Check that each operation has the correct content
ModelNode addSubsystem = operations.get(0);
Assert.assertEquals(ADD, addSubsystem.get(OP).asString());
PathAddress addr = PathAddress.pathAddress(addSubsystem.get(OP_ADDR));
Assert.assertEquals(1, addr.size());
PathElement element = addr.getElement(0);
Assert.assertEquals(SUBSYSTEM, element.getKey());
Assert.assertEquals(JMXExtension.SUBSYSTEM_NAME, element.getValue());
}
@Test
public void testParseSubsystemWithBadChild() throws Exception {
//Parse the subsystem xml into operations
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.CURRENT.getUriString() + "\">" +
" <invalid/>" +
"</subsystem>";
try {
super.parse(subsystemXml);
Assert.fail("Should not have parsed bad child");
} catch (XMLStreamException expected) {
}
}
@Test
public void testParseSubsystemWithBadAttribute() throws Exception {
//Parse the subsystem xml into operations
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.CURRENT.getUriString() + "\" bad=\"very_bad\">" +
"</subsystem>";
try {
super.parse(subsystemXml);
Assert.fail("Should not have parsed bad attribute");
} catch (XMLStreamException expected) {
}
}
@Test
public void testParseSubsystemWithConnector() throws Exception {
//Parse the subsystem xml into operations
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.CURRENT.getUriString() + "\">" +
" <remoting-connector use-management-endpoint=\"false\" />" +
"</subsystem>";
List<ModelNode> operations = super.parse(subsystemXml);
///Check that we have the expected number of operations
Assert.assertEquals(2, operations.size());
//Check that each operation has the correct content
ModelNode addSubsystem = operations.get(0);
Assert.assertEquals(ADD, addSubsystem.get(OP).asString());
assertJmxSubsystemAddress(addSubsystem.get(OP_ADDR));
ModelNode addConnector = operations.get(1);
Assert.assertEquals(ADD, addConnector.get(OP).asString());
assertJmxConnectorAddress(addConnector.get(OP_ADDR));
}
@Test
public void testParseSubsystemWithTwoConnectors() throws Exception {
//Parse the subsystem xml into operations
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.CURRENT.getUriString() + "\">" +
"<remoting-connector/>" +
"<remoting-connector/>" +
"</subsystem>";
try {
super.parse(subsystemXml);
Assert.fail("Should not have parsed second connector");
} catch (XMLStreamException expected) {
}
}
@Test
public void testParseSubsystemWithBadConnectorAttribute() throws Exception {
//Parse the subsystem xml into operations
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.CURRENT.getUriString() + "\">" +
"<remoting-connector bad=\"verybad\"/>" +
"</subsystem>";
try {
super.parse(subsystemXml);
Assert.fail("Should not have parsed bad attribute");
} catch (XMLStreamException expected) {
}
}
@Test
public void testInstallIntoController() throws Exception {
//Parse the subsystem xml and install into the controller
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.CURRENT.getUriString() + "\">" +
"<remoting-connector/>" +
"</subsystem>";
KernelServices services = super.installInController(new BaseAdditionalInitalization(),subsystemXml);
//Read the whole model and make sure it looks as expected
ModelNode model = services.readWholeModel();
Assert.assertTrue(model.get(SUBSYSTEM).hasDefined(JMXExtension.SUBSYSTEM_NAME));
//Make sure that we can connect to the MBean server
int port = 12345;
String urlString = System.getProperty("jmx.service.url",
"service:jmx:remoting-jmx://localhost:" + port);
JMXServiceURL serviceURL = new JMXServiceURL(urlString);
MBeanServerConnection connection = null;
long end = System.currentTimeMillis() + 10000;
do {
try {
connection = JMXConnectorFactory.connect(serviceURL, null).getMBeanServerConnection();
} catch (Exception e) {
e.printStackTrace();
Thread.sleep(500);
}
} while (connection == null && System.currentTimeMillis() < end);
Assert.assertNotNull(connection);
connection.getMBeanCount();
super.assertRemoveSubsystemResources(services);
}
@Test
public void testParseAndMarshalModel1_0() throws Exception {
//Parse the subsystem xml and install into the first controller
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.JMX_1_0.getUriString() + "\">" +
" <jmx-connector registry-binding=\"registry1\" server-binding=\"server1\" />" +
"</subsystem>";
String finishedSubsystemXml =
"<subsystem xmlns=\"" + Namespace.JMX_1_0.getUriString() + "\"/>";
AdditionalInitialization additionalInit = new BaseAdditionalInitalization();
KernelServices servicesA = super.installInController(additionalInit, subsystemXml);
//Get the model and the persisted xml from the first controller
ModelNode modelA = servicesA.readWholeModel();
String marshalled = servicesA.getPersistedSubsystemXml();
servicesA.shutdown();
Assert.assertTrue(marshalled.contains(Namespace.JMX_1_1.getUriString()));
marshalled = marshalled.replace(Namespace.JMX_1_1.getUriString(), Namespace.JMX_1_0.getUriString());
System.out.println(marshalled);
Assert.assertEquals(normalizeXML(finishedSubsystemXml), normalizeXML(marshalled));
//Install the persisted xml from the first controller into a second controller
KernelServices servicesB = super.installInController(additionalInit, marshalled);
ModelNode modelB = servicesB.readWholeModel();
//Make sure the models from the two controllers are identical
super.compare(modelA, modelB);
}
@Test
public void testParseAndMarshalModel1_1() throws Exception {
//Parse the subsystem xml and install into the first controller
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.JMX_1_1.getUriString() + "\">" +
"<remoting-connector/>" +
"</subsystem>";
AdditionalInitialization additionalInit = new BaseAdditionalInitalization();
KernelServices servicesA = super.installInController(additionalInit, subsystemXml);
//Get the model and the persisted xml from the first controller
ModelNode modelA = servicesA.readWholeModel();
String marshalled = servicesA.getPersistedSubsystemXml();
servicesA.shutdown();
Assert.assertEquals(normalizeXML(subsystemXml), normalizeXML(marshalled));
//Install the persisted xml from the first controller into a second controller
KernelServices servicesB = super.installInController(additionalInit, marshalled);
ModelNode modelB = servicesB.readWholeModel();
//Make sure the models from the two controllers are identical
super.compare(modelA, modelB);
}
@Test
public void testParseAndMarshalModel1_1WithShowModel() throws Exception {
//Parse the subsystem xml and install into the first controller
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.CURRENT.getUriString() + "\">" +
"<show-model value=\"true\"/>" +
"</subsystem>";
AdditionalInitialization additionalInit = new BaseAdditionalInitalization();
KernelServices servicesA = super.installInController(additionalInit, subsystemXml);
//Get the model and the persisted xml from the first controller
ModelNode modelA = servicesA.readWholeModel();
String marshalled = servicesA.getPersistedSubsystemXml();
servicesA.shutdown();
Assert.assertEquals(normalizeXML(subsystemXml), normalizeXML(marshalled));
//Install the persisted xml from the first controller into a second controller
KernelServices servicesB = super.installInController(additionalInit, marshalled);
ModelNode modelB = servicesB.readWholeModel();
//Make sure the models from the two controllers are identical
super.compare(modelA, modelB);
}
@Test
public void testParseAndMarshalModelWithRemoteConnectorRef1_1() throws Exception {
//Parse the subsystem xml and install into the first controller
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.JMX_1_1.getUriString() + "\">" +
"<remoting-connector/> " +
"</subsystem>";
AdditionalInitialization additionalInit = new BaseAdditionalInitalization();
KernelServices servicesA = super.installInController(additionalInit, subsystemXml);
//Get the model and the persisted xml from the first controller
ModelNode modelA = servicesA.readWholeModel();
String marshalled = servicesA.getPersistedSubsystemXml();
servicesA.shutdown();
Assert.assertEquals(normalizeXML(subsystemXml), normalizeXML(marshalled));
//Install the persisted xml from the first controller into a second controller
KernelServices servicesB = super.installInController(additionalInit, marshalled);
ModelNode modelB = servicesB.readWholeModel();
//Make sure the models from the two controllers are identical
super.compare(modelA, modelB);
}
@Test
public void testDescribeHandler() throws Exception {
//Parse the subsystem xml and install into the first controller
String subsystemXml =
"<subsystem xmlns=\"" + Namespace.CURRENT.getUriString() + "\">" +
" <remoting-connector />" +
"</subsystem>";
KernelServices servicesA = super.installInController(subsystemXml);
//Get the model and the describe operations from the first controller
ModelNode modelA = servicesA.readWholeModel();
ModelNode describeOp = new ModelNode();
describeOp.get(OP).set(DESCRIBE);
describeOp.get(OP_ADDR).set(
PathAddress.pathAddress(
PathElement.pathElement(SUBSYSTEM, JMXExtension.SUBSYSTEM_NAME)).toModelNode());
List<ModelNode> operations = checkResultAndGetContents(servicesA.executeOperation(describeOp)).asList();
servicesA.shutdown();
Assert.assertEquals(2, operations.size());
//Install the describe options from the first controller into a second controller
KernelServices servicesB = super.installInController(operations);
ModelNode modelB = servicesB.readWholeModel();
//Make sure the models from the two controllers are identical
super.compare(modelA, modelB);
}
private void assertJmxConnectorAddress(ModelNode address) {
PathAddress addr = PathAddress.pathAddress(address);
Assert.assertEquals(2, addr.size());
PathElement element = addr.getElement(0);
Assert.assertEquals(SUBSYSTEM, element.getKey());
Assert.assertEquals(JMXExtension.SUBSYSTEM_NAME, element.getValue());
element = addr.getElement(1);
Assert.assertEquals(CommonAttributes.REMOTING_CONNECTOR, element.getKey());
Assert.assertEquals(CommonAttributes.JMX, element.getValue());
}
private void assertJmxSubsystemAddress(ModelNode address) {
PathAddress addr = PathAddress.pathAddress(address);
Assert.assertEquals(1, addr.size());
PathElement element = addr.getElement(0);
Assert.assertEquals(SUBSYSTEM, element.getKey());
Assert.assertEquals(JMXExtension.SUBSYSTEM_NAME, element.getValue());
}
private static class BaseAdditionalInitalization extends AdditionalInitialization {
@Override
protected void initializeExtraSubystemsAndModel(ExtensionRegistry extensionRegistry, Resource rootResource,
ManagementResourceRegistration rootRegistration) {
rootResource.getModel().get(LAUNCH_TYPE).set(TYPE_STANDALONE);
}
@Override
protected void setupController(ControllerInitializer controllerInitializer) {
controllerInitializer.addSocketBinding("remote", 12345);
controllerInitializer.addPath("jboss.controller.temp.dir", System.getProperty("java.io.tmpdir"), null);
}
@Override
protected void addExtraServices(final ServiceTarget target) {
ManagementRemotingServices.installRemotingEndpoint(target, ManagementRemotingServices.MANAGEMENT_ENDPOINT, "loaclhost", EndpointService.EndpointType.MANAGEMENT, null, null);
ServiceName tmpDirPath = ServiceName.JBOSS.append("server", "path", "jboss.controller.temp.dir");
RemotingServices.installSecurityServices(target, "remote", null, null, tmpDirPath, null, null);
RemotingServices.installConnectorServicesForSocketBinding(target, ManagementRemotingServices.MANAGEMENT_ENDPOINT, "remote", SocketBinding.JBOSS_BINDING_NAME.append("remote"), OptionMap.EMPTY, null, null);
}
}
}