Extract a base class for http handler tests.

2 files changed, 105 insertions, 77 deletions

diff --git a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/HttpHandlerTestBase.java b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/HttpHandlerTestBase.java
new file mode 100644
index 00000000000..bd6fba84e35
--- /dev/null
+++ b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/HttpHandlerTestBase.java
@@ -0,0 +1,100 @@
+/*
+ * Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+ */
+
+package ai.vespa.metricsproxy.http;
+
+import ai.vespa.metricsproxy.TestUtil;
+import ai.vespa.metricsproxy.core.ConsumersConfig;
+import ai.vespa.metricsproxy.core.MetricsConsumers;
+import ai.vespa.metricsproxy.core.MetricsManager;
+import ai.vespa.metricsproxy.metric.HealthMetric;
+import ai.vespa.metricsproxy.metric.Metric;
+import ai.vespa.metricsproxy.metric.dimensions.ApplicationDimensions;
+import ai.vespa.metricsproxy.metric.dimensions.ApplicationDimensionsConfig;
+import ai.vespa.metricsproxy.metric.dimensions.NodeDimensions;
+import ai.vespa.metricsproxy.metric.dimensions.NodeDimensionsConfig;
+import ai.vespa.metricsproxy.metric.model.MetricsPacket;
+import ai.vespa.metricsproxy.service.DownService;
+import ai.vespa.metricsproxy.service.DummyService;
+import ai.vespa.metricsproxy.service.VespaService;
+import ai.vespa.metricsproxy.service.VespaServices;
+import com.google.common.collect.ImmutableList;
+import com.yahoo.container.jdisc.RequestHandlerTestDriver;
+
+import java.time.Instant;
+import java.util.List;
+
+import static ai.vespa.metricsproxy.http.ValuesFetcher.DEFAULT_PUBLIC_CONSUMER_ID;
+import static ai.vespa.metricsproxy.metric.ExternalMetrics.VESPA_NODE_SERVICE_ID;
+import static ai.vespa.metricsproxy.service.DummyService.METRIC_1;
+
+/**
+ * Base class for http handler tests.
+ *
+ * @author gjoranv
+ */
+@SuppressWarnings("UnstableApiUsage")
+public class HttpHandlerTestBase {
+
+    protected static final List<VespaService> testServices = ImmutableList.of(
+            new DummyService(0, ""),
+            new DummyService(1, ""),
+            new DownService(HealthMetric.getDown("No response")));
+
+    protected static final VespaServices vespaServices = new VespaServices(testServices);
+
+    protected static final String DEFAULT_CONSUMER = "default";
+    protected static final String CUSTOM_CONSUMER = "custom-consumer";
+
+    protected static final String CPU_METRIC = "cpu";
+
+    protected static final String URI_BASE = "http://localhost";
+
+    protected static RequestHandlerTestDriver testDriver;
+
+
+    protected static MetricsManager getMetricsManager() {
+        MetricsManager metricsManager = TestUtil.createMetricsManager(vespaServices, getMetricsConsumers(), getApplicationDimensions(), getNodeDimensions());
+        metricsManager.setExtraMetrics(ImmutableList.of(
+                new MetricsPacket.Builder(VESPA_NODE_SERVICE_ID)
+                        .timestamp(Instant.now().getEpochSecond())
+                        .putMetrics(ImmutableList.of(new Metric(CPU_METRIC, 12.345)))));
+        return metricsManager;
+    }
+
+    protected static MetricsConsumers getMetricsConsumers() {
+        var defaultConsumerDimension = new ConsumersConfig.Consumer.Metric.Dimension.Builder()
+                .key("consumer-dim").value("default-val");
+
+        var customConsumerDimension = new ConsumersConfig.Consumer.Metric.Dimension.Builder()
+                .key("consumer-dim").value("custom-val");
+
+        return new MetricsConsumers(new ConsumersConfig.Builder()
+                                            .consumer(new ConsumersConfig.Consumer.Builder()
+                                                              .name(DEFAULT_PUBLIC_CONSUMER_ID.id)
+                                                              .metric(new ConsumersConfig.Consumer.Metric.Builder()
+                                                                              .name(CPU_METRIC)
+                                                                              .outputname(CPU_METRIC))
+                                                              .metric(new ConsumersConfig.Consumer.Metric.Builder()
+                                                                              .name(METRIC_1)
+                                                                              .outputname(METRIC_1)
+                                                                              .dimension(defaultConsumerDimension)))
+                                            .consumer(new ConsumersConfig.Consumer.Builder()
+                                                              .name(CUSTOM_CONSUMER)
+                                                              .metric(new ConsumersConfig.Consumer.Metric.Builder()
+                                                                              .name(METRIC_1)
+                                                                              .outputname(METRIC_1)
+                                                                              .dimension(customConsumerDimension)))
+                                            .build());
+    }
+
+    protected static ApplicationDimensions getApplicationDimensions() {
+        return new ApplicationDimensions(new ApplicationDimensionsConfig.Builder().build());
+    }
+
+    protected static NodeDimensions getNodeDimensions() {
+        return new NodeDimensions(new NodeDimensionsConfig.Builder().build());
+    }
+
+}
diff --git a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/MetricsHandlerTest.java b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/MetricsHandlerTest.java
index ab0c6c7e6a6..83cd97bbeda 100644
--- a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/MetricsHandlerTest.java
+++ b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/http/MetricsHandlerTest.java
@@ -4,27 +4,11 @@
 
 package ai.vespa.metricsproxy.http;
 
-import ai.vespa.metricsproxy.TestUtil;
-import ai.vespa.metricsproxy.core.ConsumersConfig;
-import ai.vespa.metricsproxy.core.ConsumersConfig.Consumer;
-import ai.vespa.metricsproxy.core.MetricsConsumers;
-import ai.vespa.metricsproxy.core.MetricsManager;
-import ai.vespa.metricsproxy.metric.HealthMetric;
-import ai.vespa.metricsproxy.metric.Metric;
-import ai.vespa.metricsproxy.metric.dimensions.ApplicationDimensions;
-import ai.vespa.metricsproxy.metric.dimensions.ApplicationDimensionsConfig;
-import ai.vespa.metricsproxy.metric.dimensions.NodeDimensions;
-import ai.vespa.metricsproxy.metric.dimensions.NodeDimensionsConfig;
-import ai.vespa.metricsproxy.metric.model.MetricsPacket;
 import ai.vespa.metricsproxy.metric.model.json.GenericJsonModel;
 import ai.vespa.metricsproxy.metric.model.json.GenericMetrics;
 import ai.vespa.metricsproxy.metric.model.json.GenericService;
 import ai.vespa.metricsproxy.service.DownService;
-import ai.vespa.metricsproxy.service.DummyService;
-import ai.vespa.metricsproxy.service.VespaService;
-import ai.vespa.metricsproxy.service.VespaServices;
 import com.fasterxml.jackson.databind.ObjectMapper;
-import com.google.common.collect.ImmutableList;
 import com.yahoo.container.jdisc.RequestHandlerTestDriver;
 import org.json.JSONArray;
 import org.json.JSONObject;
@@ -33,15 +17,11 @@ import org.junit.Ignore;
 import org.junit.Test;
 
 import java.io.IOException;
-import java.time.Instant;
-import java.util.List;
 import java.util.concurrent.Executors;
 
 import static ai.vespa.metricsproxy.core.VespaMetrics.INSTANCE_DIMENSION_ID;
 import static ai.vespa.metricsproxy.http.MetricsHandler.V1_PATH;
 import static ai.vespa.metricsproxy.http.MetricsHandler.VALUES_PATH;
-import static ai.vespa.metricsproxy.http.ValuesFetcher.DEFAULT_PUBLIC_CONSUMER_ID;
-import static ai.vespa.metricsproxy.metric.ExternalMetrics.VESPA_NODE_SERVICE_ID;
 import static ai.vespa.metricsproxy.metric.model.StatusCode.DOWN;
 import static ai.vespa.metricsproxy.metric.model.json.JacksonUtil.createObjectMapper;
 import static ai.vespa.metricsproxy.service.DummyService.METRIC_1;
@@ -56,35 +36,17 @@ import static org.junit.Assert.fail;
  * @author gjoranv
  */
 @SuppressWarnings("UnstableApiUsage")
-public class MetricsHandlerTest {
+public class MetricsHandlerTest extends HttpHandlerTestBase {
 
-    private static final List<VespaService> testServices = ImmutableList.of(
-            new DummyService(0, ""),
-            new DummyService(1, ""),
-            new DownService(HealthMetric.getDown("No response")));
-
-    private static final VespaServices vespaServices = new VespaServices(testServices);
-
-    private static final String DEFAULT_CONSUMER = "default";
-    private static final String CUSTOM_CONSUMER = "custom-consumer";
-
-    private static final String CPU_METRIC = "cpu";
-
-    private static final String URI_BASE = "http://localhost";
     private static final String V1_URI = URI_BASE + V1_PATH;
     private static final String VALUES_URI = URI_BASE + VALUES_PATH;
 
-
-    private static RequestHandlerTestDriver testDriver;
-
     @BeforeClass
     public static void setup() {
-        MetricsManager metricsManager = TestUtil.createMetricsManager(vespaServices, getMetricsConsumers(), getApplicationDimensions(), getNodeDimensions());
-        metricsManager.setExtraMetrics(ImmutableList.of(
-                new MetricsPacket.Builder(VESPA_NODE_SERVICE_ID)
-                        .timestamp(Instant.now().getEpochSecond())
-                        .putMetrics(ImmutableList.of(new Metric(CPU_METRIC, 12.345)))));
-        MetricsHandler handler = new MetricsHandler(Executors.newSingleThreadExecutor(), metricsManager, vespaServices, getMetricsConsumers());
+        MetricsHandler handler = new MetricsHandler(Executors.newSingleThreadExecutor(),
+                                                    getMetricsManager(),
+                                                    vespaServices,
+                                                    getMetricsConsumers());
         testDriver = new RequestHandlerTestDriver(handler);
     }
 
@@ -233,38 +195,4 @@ public class MetricsHandlerTest {
         throw new RuntimeException();
     }
 
-    private static MetricsConsumers getMetricsConsumers() {
-        var defaultConsumerDimension = new Consumer.Metric.Dimension.Builder()
-                .key("consumer-dim").value("default-val");
-
-        var customConsumerDimension = new Consumer.Metric.Dimension.Builder()
-                .key("consumer-dim").value("custom-val");
-
-        return new MetricsConsumers(new ConsumersConfig.Builder()
-                                            .consumer(new Consumer.Builder()
-                                                              .name(DEFAULT_PUBLIC_CONSUMER_ID.id)
-                                                              .metric(new Consumer.Metric.Builder()
-                                                                              .name(CPU_METRIC)
-                                                                              .outputname(CPU_METRIC))
-                                                              .metric(new Consumer.Metric.Builder()
-                                                                              .name(METRIC_1)
-                                                                              .outputname(METRIC_1)
-                                                                              .dimension(defaultConsumerDimension)))
-                                            .consumer(new Consumer.Builder()
-                                                    .name(CUSTOM_CONSUMER)
-                                                              .metric(new Consumer.Metric.Builder()
-                                                                              .name(METRIC_1)
-                                                                              .outputname(METRIC_1)
-                                                                              .dimension(customConsumerDimension)))
-                                            .build());
-    }
-
-    private static ApplicationDimensions getApplicationDimensions() {
-        return new ApplicationDimensions(new ApplicationDimensionsConfig.Builder().build());
-    }
-
-    private static NodeDimensions getNodeDimensions() {
-        return new NodeDimensions(new NodeDimensionsConfig.Builder().build());
-    }
-
 }