diff --git a/proxy/pom.xml b/proxy/pom.xml index 73213c3c5..6ab173e9a 100644 --- a/proxy/pom.xml +++ b/proxy/pom.xml @@ -593,6 +593,11 @@ proto-google-common-protos 2.0.1 + + org.yaml + snakeyaml + 2.0 + diff --git a/proxy/src/test/java/com/wavefront/agent/logsharvesting/LogsIngesterTest.java b/proxy/src/test/java/com/wavefront/agent/logsharvesting/LogsIngesterTest.java index 05ae35a61..1f17a8bc1 100644 --- a/proxy/src/test/java/com/wavefront/agent/logsharvesting/LogsIngesterTest.java +++ b/proxy/src/test/java/com/wavefront/agent/logsharvesting/LogsIngesterTest.java @@ -1,12 +1,23 @@ package com.wavefront.agent.logsharvesting; -import static org.easymock.EasyMock.*; +import static org.easymock.EasyMock.createMock; +import static org.easymock.EasyMock.expect; +import static org.easymock.EasyMock.expectLastCall; +import static org.easymock.EasyMock.replay; +import static org.easymock.EasyMock.reset; +import static org.easymock.EasyMock.verify; import static org.hamcrest.MatcherAssert.assertThat; -import static org.hamcrest.Matchers.*; import static org.hamcrest.Matchers.contains; +import static org.hamcrest.Matchers.containsInAnyOrder; +import static org.hamcrest.Matchers.emptyIterable; +import static org.hamcrest.Matchers.equalTo; +import static org.hamcrest.Matchers.hasSize; +import static org.hamcrest.Matchers.instanceOf; +import static org.hamcrest.Matchers.lessThan; import com.fasterxml.jackson.databind.ObjectMapper; import com.fasterxml.jackson.dataformat.yaml.YAMLFactory; +import com.fasterxml.jackson.dataformat.yaml.YAMLFactoryBuilder; import com.google.common.collect.ImmutableList; import com.google.common.collect.ImmutableMap; import com.google.common.collect.Lists; @@ -41,11 +52,15 @@ import org.junit.After; import org.junit.Test; import org.logstash.beats.Message; +import org.yaml.snakeyaml.LoaderOptions; import wavefront.report.Histogram; import wavefront.report.ReportPoint; /** @author Mori Bellamy (mori@wavefront.com) */ public class LogsIngesterTest { + private final AtomicLong now; + private final AtomicLong nanos; + private final ObjectMapper objectMapper; private LogsIngestionConfig logsIngestionConfig; private LogsIngester logsIngesterUnderTest; private FilebeatIngester filebeatIngesterUnderTest; @@ -53,9 +68,13 @@ public class LogsIngesterTest { private ReportableEntityHandlerFactory mockFactory; private ReportableEntityHandler mockPointHandler; private ReportableEntityHandler mockHistogramHandler; - private AtomicLong now = new AtomicLong((System.currentTimeMillis() / 60000) * 60000); - private AtomicLong nanos = new AtomicLong(System.nanoTime()); - private ObjectMapper objectMapper = new ObjectMapper(new YAMLFactory()); + + public LogsIngesterTest() { + this.now = new AtomicLong((System.currentTimeMillis() / 60000) * 60000); + this.nanos = new AtomicLong(System.nanoTime()); + YAMLFactoryBuilder factory = new YAMLFactoryBuilder(new YAMLFactory()); + this.objectMapper = new ObjectMapper(factory.loaderOptions(new LoaderOptions()).build()); + } private LogsIngestionConfig parseConfigFile(String configPath) throws IOException { File configFile =