Java tutorial
/* * RHQ Management Platform * Copyright (C) 2005-2008 Red Hat, Inc. * All rights reserved. * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation version 2 of the License. * * This program 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 for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ package org.rhq.plugins.modcluster.test; import java.io.File; import java.util.HashSet; import java.util.LinkedList; import java.util.Queue; import java.util.Set; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.testng.annotations.AfterSuite; import org.testng.annotations.BeforeSuite; import org.testng.annotations.Test; import org.rhq.core.clientapi.server.discovery.InventoryReport; import org.rhq.core.domain.configuration.Configuration; import org.rhq.core.domain.configuration.PropertySimple; import org.rhq.core.domain.measurement.MeasurementData; import org.rhq.core.domain.measurement.MeasurementDefinition; import org.rhq.core.domain.measurement.MeasurementReport; import org.rhq.core.domain.measurement.MeasurementScheduleRequest; import org.rhq.core.domain.resource.Resource; import org.rhq.core.pc.PluginContainer; import org.rhq.core.pc.PluginContainerConfiguration; import org.rhq.core.pc.plugin.FileSystemPluginFinder; import org.rhq.core.pc.plugin.PluginEnvironment; import org.rhq.core.pc.plugin.PluginManager; import org.rhq.core.pluginapi.configuration.ConfigurationFacet; import org.rhq.core.pluginapi.inventory.ResourceComponent; import org.rhq.core.pluginapi.measurement.MeasurementFacet; import org.rhq.core.pluginapi.operation.OperationFacet; import org.rhq.core.pluginapi.operation.OperationResult; /** * @author Stefan Negrea */ @SuppressWarnings("rawtypes") @Test(groups = "modcluster-plugin") public class ModClusterPluginIntegrationTest { private Log log = LogFactory.getLog(this.getClass()); private static final String PLUGIN_NAME = "mod_cluster"; @BeforeSuite public void start() { try { File pluginDir = new File("target/testsetup/plugins"); PluginContainerConfiguration pcConfig = new PluginContainerConfiguration(); pcConfig.setPluginFinder(new FileSystemPluginFinder(pluginDir)); pcConfig.setPluginDirectory(pluginDir); pcConfig.setInsideAgent(false); PluginContainer.getInstance().setConfiguration(pcConfig); PluginContainer.getInstance().initialize(); log.info("PC started."); for (String plugin : PluginContainer.getInstance().getPluginManager().getMetadataManager() .getPluginNames()) { log.info("...Loaded plugin: " + plugin); } } catch (Exception e) { log.info("Error initializing the context", e); } } @AfterSuite public void stop() { PluginContainer.getInstance().shutdown(); } @Test public void testPluginLoad() { PluginManager pluginManager = PluginContainer.getInstance().getPluginManager(); PluginEnvironment pluginEnvironment = pluginManager.getPlugin(PLUGIN_NAME); assert (pluginEnvironment != null) : "Null environment, plugin not loaded"; assert (pluginEnvironment.getPluginName().equals(PLUGIN_NAME)); } @Test(dependsOnMethods = "testPluginLoad") public void testDiscovery() throws Exception { InventoryReport report = PluginContainer.getInstance().getInventoryManager().executeServerScanImmediately(); assert report != null; log.info("Discovery took: " + (report.getEndTime() - report.getStartTime()) + "ms"); Thread.sleep(1000); report = PluginContainer.getInstance().getInventoryManager().executeServiceScanImmediately(); assert report != null; log.info("Discovery took: " + (report.getEndTime() - report.getStartTime()) + "ms"); Set<Resource> resources = findResource(PluginContainer.getInstance().getInventoryManager().getPlatform()); log.info("Found " + resources.size() + " mod_cluster and mod_cluster_context instance(s)."); assert (resources.size() != 0) : "No mod_cluster or related instances found."; if (resources.size() != 0) { for (Object objectResource : resources.toArray()) { Resource resource = (Resource) objectResource; if (resource.getResourceType().getName().equals("mod_cluster")) { testMainResource(resource); } else if (resource.getResourceType().getName().equals("Webapp Context")) { testWebappContext(resource); } else if (resource.getResourceType().getName().equals("HA Service Configuration")) { testLoadServiceConfiguration(resource); } else if (resource.getResourceType().getName().equals("HA Service")) { testServiceMeasurement(resource); testServiceMethodInvocation(resource); testServiceSimpleMethodInvocation(resource); } else if (resource.getResourceType().getName().equals("Catalina Service")) { testServiceSimpleMethodInvocation(resource); } } } } private void testServiceSimpleMethodInvocation(Resource resource) throws InterruptedException, Exception { ResourceComponent resourceComponent = PluginContainer.getInstance().getInventoryManager() .getResourceComponent(resource); Configuration config = new Configuration(); OperationResult result = ((OperationFacet) resourceComponent).invokeOperation("enable", config); log.info("Result of operation enable was: " + result.getSimpleResult()); assert result.getSimpleResult().equals("true") : "The operation execution failed!"; result = ((OperationFacet) resourceComponent).invokeOperation("disable", config); log.info("Result of operation disable was: " + result.getSimpleResult()); assert result.getSimpleResult().equals("true") : "The operation execution failed!"; } private void testServiceMethodInvocation(Resource resource) throws InterruptedException, Exception { ResourceComponent resourceComponent = PluginContainer.getInstance().getInventoryManager() .getResourceComponent(resource); Configuration config = new Configuration(); config.put(new PropertySimple("p1", "1")); config.put(new PropertySimple("p2", java.util.concurrent.TimeUnit.SECONDS)); OperationResult result = ((OperationFacet) resourceComponent).invokeOperation("stop", config); log.info("Result of operation stop was: " + result.getSimpleResult()); } private void testLoadServiceConfiguration(Resource resource) throws Exception { ResourceComponent resourceComponent = PluginContainer.getInstance().getInventoryManager() .getResourceComponent(resource); ((ConfigurationFacet) resourceComponent).loadResourceConfiguration(); } private void testServiceMeasurement(Resource resource) throws Exception { ResourceComponent resourceComponent = PluginContainer.getInstance().getInventoryManager() .getResourceComponent(resource); if (resourceComponent instanceof MeasurementFacet) { for (MeasurementDefinition def : resource.getResourceType().getMetricDefinitions()) { Set<MeasurementScheduleRequest> metricList = new HashSet<MeasurementScheduleRequest>(); metricList .add(new MeasurementScheduleRequest(1, def.getName(), 1000, true, def.getDataType(), null)); MeasurementReport report = new MeasurementReport(); ((MeasurementFacet) resourceComponent).getValues(report, metricList); MeasurementData data = report.getTraitData().iterator().next(); assert data != null : "Unable to collect trait [" + def.getName() + "] on " + resource; log.info("Measurement: " + def.getName() + "=" + data.getValue()); } } } private void testMainResource(Resource resource) throws Exception { ResourceComponent resourceComponent = PluginContainer.getInstance().getInventoryManager() .getResourceComponent(resource); if (resourceComponent instanceof OperationFacet) { for (String operationName : new String[] { "reset", "refresh" }) { OperationResult result = ((OperationFacet) resourceComponent).invokeOperation(operationName, new Configuration()); log.info("Result of operation " + operationName + " was: " + result); //need to wait for the reset to finish, this is a long operation Thread.sleep(5000); } } } private void testWebappContext(Resource resource) throws Exception { ResourceComponent resourceComponent = PluginContainer.getInstance().getInventoryManager() .getResourceComponent(resource); if (resourceComponent instanceof OperationFacet) { try { OperationResult result = ((OperationFacet) resourceComponent).invokeOperation("enableContext", null); log.info("Result of operation " + "enableContext" + " was: " + result.getSimpleResult()); Configuration config = new Configuration(); config.put(new PropertySimple("timeout", "1")); config.put(new PropertySimple("unit", java.util.concurrent.TimeUnit.SECONDS)); result = ((OperationFacet) resourceComponent).invokeOperation("stopContext", config); log.info("Result of operation stopContext was: " + result.getSimpleResult()); assert result.getSimpleResult().equals("true") : "The operation execution failed!"; result = ((OperationFacet) resourceComponent).invokeOperation("enableContext", null); log.info("Result of operation enableContext was: " + result.getSimpleResult()); assert result.getSimpleResult().equals("true") : "The operation execution failed!"; result = ((OperationFacet) resourceComponent).invokeOperation("disableContext", null); log.info("Result of operation disableContext was: " + result.getSimpleResult()); assert result.getSimpleResult().equals("true") : "The operation execution failed!"; } catch (Exception e) { log.info("Operation failed. ", e); } } } private Set<Resource> findResource(Resource parent) { Set<Resource> foundResources = new HashSet<Resource>(); Queue<Resource> discoveryQueue = new LinkedList<Resource>(); discoveryQueue.add(parent); while (!discoveryQueue.isEmpty()) { Resource currentResource = discoveryQueue.poll(); log.info("Discovered resource of type: " + currentResource.getResourceType().getName()); if (currentResource.getResourceType().getPlugin().equals(PLUGIN_NAME)) { foundResources.add(currentResource); } if (currentResource.getChildResources() != null) { for (Resource child : currentResource.getChildResources()) { discoveryQueue.add(child); } } } return foundResources; } }