MBeanExporterParserTests.java example

Explorer
spring-integration-master
/*
 * Copyright 2002-2015 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.springframework.integration.jmx.config;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;

import java.util.Properties;

import javax.management.MBeanServer;

import org.junit.Test;
import org.junit.runner.RunWith;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.integration.monitor.IntegrationMBeanExporter;
import org.springframework.integration.support.management.AbstractMessageChannelMetrics;
import org.springframework.integration.support.management.AbstractMessageHandlerMetrics;
import org.springframework.integration.support.management.DefaultMessageChannelMetrics;
import org.springframework.integration.support.management.DefaultMessageHandlerMetrics;
import org.springframework.integration.support.management.ExponentialMovingAverage;
import org.springframework.integration.support.management.ExponentialMovingAverageRate;
import org.springframework.integration.support.management.ExponentialMovingAverageRatio;
import org.springframework.integration.support.management.IntegrationManagementConfigurer;
import org.springframework.integration.support.management.MessageChannelMetrics;
import org.springframework.integration.support.management.MessageHandlerMetrics;
import org.springframework.integration.support.management.MetricsFactory;
import org.springframework.integration.test.util.TestUtils;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

/**
 * @author Mark Fisher
 * @author Oleg Zhurakousky
 * @author Gunnar Hillert
 * @author Gary Russell
 * @since 2.0
 */
@ContextConfiguration
@RunWith(SpringJUnit4ClassRunner.class)
@DirtiesContext
public class MBeanExporterParserTests {

	@Autowired
	private ApplicationContext context;

	@Test
	public void testMBeanExporterExists() throws InterruptedException {
		IntegrationMBeanExporter exporter = this.context.getBean(IntegrationMBeanExporter.class);
		MBeanServer server = this.context.getBean("mbs", MBeanServer.class);
		Properties properties = TestUtils.getPropertyValue(exporter, "objectNameStaticProperties", Properties.class);
		assertNotNull(properties);
		assertEquals(2, properties.size());
		assertTrue(properties.containsKey("foo"));
		assertTrue(properties.containsKey("bar"));
		assertEquals(server, exporter.getServer());
		assertSame(context.getBean("keyNamer"), TestUtils.getPropertyValue(exporter, "namingStrategy"));
		MessageChannelMetrics metrics = context.getBean("foo", MessageChannelMetrics.class);
		assertTrue(metrics.isCountsEnabled());
		assertFalse(metrics.isStatsEnabled());
		checkCustomized(metrics);
		MessageHandlerMetrics handlerMetrics = context.getBean("transformer.handler", MessageHandlerMetrics.class);
		checkCustomized(handlerMetrics);
		metrics = context.getBean("bar", MessageChannelMetrics.class);
		assertTrue(metrics.isCountsEnabled());
		assertFalse(metrics.isStatsEnabled());
		metrics = context.getBean("baz", MessageChannelMetrics.class);
		assertFalse(metrics.isCountsEnabled());
		assertFalse(metrics.isStatsEnabled());
		metrics = context.getBean("qux", MessageChannelMetrics.class);
		assertFalse(metrics.isCountsEnabled());
		assertFalse(metrics.isStatsEnabled());
		metrics = context.getBean("fiz", MessageChannelMetrics.class);
		assertTrue(metrics.isCountsEnabled());
		assertTrue(metrics.isStatsEnabled());
		metrics = context.getBean("buz", MessageChannelMetrics.class);
		assertTrue(metrics.isCountsEnabled());
		assertTrue(metrics.isStatsEnabled());
		metrics = context.getBean("!excluded", MessageChannelMetrics.class);
		assertFalse(metrics.isCountsEnabled());
		assertFalse(metrics.isStatsEnabled());
		checkCustomized(metrics);
		MetricsFactory factory = context.getBean(MetricsFactory.class);
		IntegrationManagementConfigurer configurer = context.getBean(IntegrationManagementConfigurer.class);
		assertSame(factory, TestUtils.getPropertyValue(configurer, "metricsFactory"));
		exporter.destroy();
	}

	private void checkCustomized(MessageChannelMetrics metrics) {
		assertFalse(metrics.isLoggingEnabled());
		assertEquals(20, TestUtils.getPropertyValue(metrics, "channelMetrics.sendDuration.window"));
		assertEquals(1000000., TestUtils.getPropertyValue(metrics, "channelMetrics.sendDuration.factor", Double.class),
				.01);
		assertEquals(30, TestUtils.getPropertyValue(metrics, "channelMetrics.sendErrorRate.window"));
		assertEquals(2000000.,
				TestUtils.getPropertyValue(metrics, "channelMetrics.sendErrorRate.period", Double.class), .01);
		assertEquals(.001 / 120000,
				TestUtils.getPropertyValue(metrics, "channelMetrics.sendErrorRate.lapse", Double.class), .01);

		assertEquals(40, TestUtils.getPropertyValue(metrics, "channelMetrics.sendSuccessRatio.window"));
		assertEquals(.001 / 130000, TestUtils.getPropertyValue(metrics, "channelMetrics.sendRate.lapse", Double.class),
				.01);

		assertEquals(50, TestUtils.getPropertyValue(metrics, "channelMetrics.sendRate.window"));
		assertEquals(3000000., TestUtils.getPropertyValue(metrics, "channelMetrics.sendRate.period", Double.class), .01);
		assertEquals(.001 / 140000, TestUtils.getPropertyValue(metrics, "channelMetrics.sendRate.lapse", Double.class),
				.01);
	}

	private void checkCustomized(MessageHandlerMetrics metrics) {
		assertEquals(20, TestUtils.getPropertyValue(metrics, "handlerMetrics.duration.window"));
		assertEquals(1000000., TestUtils.getPropertyValue(metrics, "handlerMetrics.duration.factor", Double.class),
				.01);
	}

	public static class CustomMetrics implements MetricsFactory {

		@Override
		public AbstractMessageChannelMetrics createChannelMetrics(String name) {
			return new DefaultMessageChannelMetrics(name,
					new ExponentialMovingAverage(20, 1000000.),
					new ExponentialMovingAverageRate(2000, 120000, 30, true),
					new ExponentialMovingAverageRatio(130000, 40, true),
					new ExponentialMovingAverageRate(3000, 140000, 50, true));
		}

		@Override
		public AbstractMessageHandlerMetrics createHandlerMetrics(String name) {
			return new DefaultMessageHandlerMetrics(name, new ExponentialMovingAverage(20, 1000000.));
		}

	}

}