Java tutorial
/** * The MIT License * Copyright (c) 2010 JmxTrans team * * Permission is hereby granted, free of charge, to any person obtaining a copy * of this software and associated documentation files (the "Software"), to deal * in the Software without restriction, including without limitation the rights * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell * copies of the Software, and to permit persons to whom the Software is * furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN * THE SOFTWARE. */ package com.googlecode.jmxtrans.model.output; import com.google.common.collect.ImmutableList; import com.google.common.collect.ImmutableMap; import com.googlecode.jmxtrans.ConfigurationParser; import com.googlecode.jmxtrans.cli.JmxTransConfiguration; import com.googlecode.jmxtrans.guice.JmxTransModule; import com.googlecode.jmxtrans.model.Query; import com.googlecode.jmxtrans.model.Result; import com.googlecode.jmxtrans.model.Server; import com.googlecode.jmxtrans.model.ValidationException; import com.googlecode.jmxtrans.test.RequiresIO; import com.googlecode.jmxtrans.util.JsonUtils; import com.kaching.platform.testing.AllowDNSResolution; import org.apache.commons.pool.impl.GenericKeyedObjectPool; import org.junit.Test; import org.junit.experimental.categories.Category; import org.mockito.Matchers; import org.mockito.Mockito; import java.io.ByteArrayOutputStream; import java.io.File; import java.io.IOException; import java.io.OutputStream; import java.net.InetSocketAddress; import java.net.Socket; import java.util.ArrayList; import java.util.List; import static com.google.common.collect.ImmutableList.of; import static com.googlecode.jmxtrans.model.QueryFixtures.dummyQuery; import static com.googlecode.jmxtrans.model.QueryFixtures.queryAllowingDottedKeys; import static com.googlecode.jmxtrans.model.QueryFixtures.queryUsingDomainAsKey; import static com.googlecode.jmxtrans.model.QueryFixtures.queryWithAllTypeNames; import static com.googlecode.jmxtrans.model.ResultFixtures.dummyResults; import static com.googlecode.jmxtrans.model.ResultFixtures.numericResult; import static com.googlecode.jmxtrans.model.ResultFixtures.numericResultWithTypenames; import static com.googlecode.jmxtrans.model.ServerFixtures.dummyServer; import static org.assertj.core.api.Assertions.assertThat; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; @Category(RequiresIO.class) @AllowDNSResolution public class GraphiteWriterTests { @Test(expected = NullPointerException.class) public void hostIsRequired() throws ValidationException { try { GraphiteWriter.builder().setPort(123).build(); } catch (NullPointerException npe) { assertThat(npe).hasMessage("Host cannot be null."); throw npe; } } @Test(expected = NullPointerException.class) public void portIsRequired() throws ValidationException { try { GraphiteWriter.builder().setHost("localhost").build(); } catch (NullPointerException npe) { assertThat(npe).hasMessage("Port cannot be null."); throw npe; } } private static GraphiteWriter getGraphiteWriter(OutputStream out) throws Exception { return getGraphiteWriter(out, new ArrayList<String>()); } private static GraphiteWriter getGraphiteWriter(OutputStream out, List<String> typeNames) throws Exception { GenericKeyedObjectPool<InetSocketAddress, Socket> pool = mock(GenericKeyedObjectPool.class); Socket socket = mock(Socket.class); when(pool.borrowObject(Matchers.any(InetSocketAddress.class))).thenReturn(socket); when(socket.getOutputStream()).thenReturn(out); GraphiteWriter writer = GraphiteWriter.builder().setHost("localhost").setPort(2003).addTypeNames(typeNames) .build(); writer.setPool(pool); return writer; } private static String getOutput(Server server, Query query, Result result) throws Exception { ByteArrayOutputStream out = new ByteArrayOutputStream(); GraphiteWriter writer = getGraphiteWriter(out); writer.doWrite(server, query, of(result)); return out.toString(); } @Test public void writeSingleResult() throws Exception { // check that Graphite format is respected assertThat(getOutput(dummyServer(), dummyQuery(), numericResult())).startsWith( "servers.host_example_net_4321.ObjectPendingFinalizationCount.ObjectPendingFinalizationCount 10"); } @Test public void useObjDomainWorks() throws Exception { // check that Graphite format is respected assertThat(getOutput(dummyServer(), queryUsingDomainAsKey(), numericResult())).startsWith( "servers.host_example_net_4321.ObjectPendingFinalizationCount.ObjectPendingFinalizationCount 10 0"); } @Test public void allowDottedWorks() throws Exception { // check that Graphite format is respected assertThat(getOutput(dummyServer(), queryAllowingDottedKeys(), numericResult())).startsWith( "servers.host_example_net_4321.ObjectPendingFinalizationCount.ObjectPendingFinalizationCount 10 0"); } @Test public void useAllTypeNamesWorks() throws Exception { // Set useAllTypeNames to true String typeName = "typeName,typeNameKey1=typeNameValue1,typeNameKey2=typeNameValue2"; String typeNameReordered = "typeNameKey2=typeNameValue2,typeName,typeNameKey1=typeNameValue1"; // check that Graphite format is respected assertThat(getOutput(dummyServer(), queryWithAllTypeNames(), numericResultWithTypenames(typeName))) .startsWith( "servers.host_example_net_4321.ObjectPendingFinalizationCount.typeNameValue1_typeNameValue2.ObjectPendingFinalizationCount 10 0"); assertThat(getOutput(dummyServer(), queryWithAllTypeNames(), numericResultWithTypenames(typeNameReordered))) .startsWith( "servers.host_example_net_4321.ObjectPendingFinalizationCount.typeNameValue2_typeNameValue1.ObjectPendingFinalizationCount 10 0"); } @Test public void booleanAsNumberWorks() throws Exception { File testInput = new File(GraphiteWriterTests.class.getResource("/booleanTest.json").toURI()); boolean continueOnJsonError = true; JsonUtils jsonUtils = JmxTransModule.createInjector(new JmxTransConfiguration()) .getInstance(JsonUtils.class); ImmutableList servers = new ConfigurationParser(jsonUtils).parseServers(of(testInput), continueOnJsonError); Result result = new Result(System.currentTimeMillis(), "attributeName", "className", "objDomain", null, "typeName", ImmutableMap.of("key", (Object) true)); ByteArrayOutputStream out = new ByteArrayOutputStream(); GenericKeyedObjectPool<InetSocketAddress, Socket> pool = mock(GenericKeyedObjectPool.class); Socket socket = mock(Socket.class); when(pool.borrowObject(Matchers.any(InetSocketAddress.class))).thenReturn(socket); when(socket.getOutputStream()).thenReturn(out); Server server = ((Server) servers.get(0)); Query query = server.getQueries().asList().get(0); GraphiteWriter writer = (GraphiteWriter) (query.getOutputWriters().get(0)); writer.setPool(pool); writer.doWrite(server, query, of(result)); // check that the booleanAsNumber property was picked up from the JSON assertThat(out.toString()).startsWith("servers.host_123.objDomain.attributeName.key 1"); } @Test public void checkEmptyTypeNamesAreIgnored() throws Exception { Server server = Server.builder().setHost("host").setPort("123").build(); // Set useObjDomain to true Query query = Query.builder().setUseObjDomainAsKey(true).setAllowDottedKeys(true) .setObj("\"yammer.metrics\":name=\"uniqueName\",type=\"\"").build(); Result result = new Result(System.currentTimeMillis(), "Attribute", "com.yammer.metrics.reporting.JmxReporter$Counter", "yammer.metrics", null, "name=\"uniqueName\",type=\"\"", ImmutableMap.of("Attribute", (Object) 0)); ByteArrayOutputStream out = new ByteArrayOutputStream(); ArrayList<String> typeNames = new ArrayList<String>(); typeNames.add("name"); typeNames.add("type"); GraphiteWriter writer = getGraphiteWriter(out, typeNames); writer.doWrite(server, query, of(result)); // check that the empty type "type" is ignored when allowDottedKeys is true assertThat(out.toString()).startsWith("servers.host_123.yammer.metrics.uniqueName.Attribute 0 "); // check that this also works when literal " characters aren't included in the JMX ObjectName query = Query.builder().setUseObjDomainAsKey(true).setAllowDottedKeys(true) .setObj("yammer.metrics:name=uniqueName,type=").build(); out = new ByteArrayOutputStream(); writer = getGraphiteWriter(out, typeNames); writer.doWrite(server, query, of(result)); assertThat(out.toString()).startsWith("servers.host_123.yammer.metrics.uniqueName.Attribute 0 "); // check that the empty type "type" is ignored when allowDottedKeys is false query = Query.builder().setUseObjDomainAsKey(true).setAllowDottedKeys(false) .setObj("\"yammer.metrics\":name=\"uniqueName\",type=\"\"").build(); out = new ByteArrayOutputStream(); writer = getGraphiteWriter(out, typeNames); writer.doWrite(server, query, of(result)); assertThat(out.toString()).startsWith("servers.host_123.yammer_metrics.uniqueName.Attribute 0 "); } @Test public void socketInvalidatedWhenError() throws Exception { GenericKeyedObjectPool<InetSocketAddress, Socket> pool = mock(GenericKeyedObjectPool.class); Socket socket = mock(Socket.class); when(pool.borrowObject(Matchers.any(InetSocketAddress.class))).thenReturn(socket); UnflushableByteArrayOutputStream out = new UnflushableByteArrayOutputStream(); when(socket.getOutputStream()).thenReturn(out); GraphiteWriter writer = GraphiteWriter.builder().setHost("localhost").setPort(2003).build(); writer.setPool(pool); writer.doWrite(dummyServer(), dummyQuery(), dummyResults()); Mockito.verify(pool).invalidateObject(Matchers.any(InetSocketAddress.class), Matchers.eq(socket)); Mockito.verify(pool, Mockito.never()).returnObject(Matchers.any(InetSocketAddress.class), Matchers.eq(socket)); } private static class UnflushableByteArrayOutputStream extends ByteArrayOutputStream { @Override public void flush() throws IOException { throw new IOException(); } } }