use mock for metricsFactory

Author: abbeyqy

src/test/java/com/arpnetworking/tsdcore/sinks/KairosDbSinkTest.java

@@ -17,8 +17,8 @@
 
 import akka.http.javadsl.model.MediaTypes;
 import com.arpnetworking.commons.jackson.databind.ObjectMapperFactory;
+import com.arpnetworking.metrics.Metrics;
 import com.arpnetworking.metrics.MetricsFactory;
-import com.arpnetworking.metrics.impl.TsdMetricsFactory;
 import com.arpnetworking.tsdcore.model.AggregatedData;
 import com.arpnetworking.tsdcore.model.Condition;
 import com.arpnetworking.tsdcore.model.FQDSN;
@@ -39,6 +39,8 @@
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.Mockito;
 
 import java.io.IOException;
 import java.net.URI;
@@ -57,12 +59,12 @@ public void setUp() {
         _wireMockServer = new WireMockServer(0);
         _wireMockServer.start();
         _wireMock = new WireMock(_wireMockServer.port());
-
         _kairosDbSinkBuilder = new KairosDbSink.Builder()
                 .setName("kairosdb_sink_test")
                 .setActorSystem(getSystem())
                 .setUri(URI.create("http://localhost:" + _wireMockServer.port() + PATH))
-                .setMetricsFactory(METRICS_FACTORY);
+                .setMetricsFactory(_mockMetricsFactory);
+        Mockito.doReturn(_mockMetrics).when(_mockMetricsFactory).create();
     }
 
     @After
@@ -116,7 +118,6 @@ public void testPost() throws InterruptedException, IOException {
                         .setPeriod(Period.minutes(1))
                         .setStart(start)
                         .build());
-
         // Allow the request/response to complete
         Thread.sleep(1000);
 
@@ -140,7 +141,9 @@ public void testPost() throws InterruptedException, IOException {
     private static final String PATH = "/kairos/post/path";
     private static final StatisticFactory STATISTIC_FACTORY = new StatisticFactory();
     private static final ObjectMapper OBJECT_MAPPER = ObjectMapperFactory.getInstance();
-    private static final MetricsFactory METRICS_FACTORY = TsdMetricsFactory.newInstance(
-                "mock_metrics_factory",
-                "mock_metrics_factory");
+
+    @Mock
+    private Metrics _mockMetrics;
+    @Mock
+    private MetricsFactory _mockMetricsFactory;
 }