1 files changed, 77 insertions, 0 deletions
diff --git a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/service/VespaServiceTest.java b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/service/VespaServiceTest.java
new file mode 100644
index 00000000000..13be98db23a
--- /dev/null
+++ b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/service/VespaServiceTest.java
@@ -0,0 +1,77 @@
+/*
+ * Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+ */
+
+package ai.vespa.metricsproxy.service;
+
+import ai.vespa.metricsproxy.TestUtil;
+import ai.vespa.metricsproxy.metric.Metrics;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.io.File;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+
+/**
+ * @author Unknown
+ */
+public class VespaServiceTest {
+    private MockHttpServer service;
+    private int csPort;
+    private static final String response;
+
+    static {
+        response = TestUtil.getContents("metrics-state.json");
+        HttpMetricFetcher.CONNECTION_TIMEOUT = 60000; // 60 secs in unit tests
+    }
+
+    @Before
+    public void setupHTTPServer() {
+        csPort = 18632; // see factory/doc/port-ranges.txt
+        try {
+            service = new MockHttpServer(csPort, response, HttpMetricFetcher.METRICS_PATH);
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+
+    @Test
+    public void testService() {
+        VespaService service = new VespaService("qrserver", "container/qrserver.0");
+        assertThat(service.getServiceName(), is("qrserver"));
+        assertThat(service.getInstanceName(), is("qrserver"));
+        assertThat(service.getPid(), is(-1));
+        assertThat(service.getConfigId(), is("container/qrserver.0"));
+
+
+        service = VespaService.create("qrserver2", "container/qrserver.0", -1);
+        assertThat(service.getServiceName(), is("qrserver"));
+        assertThat(service.getInstanceName(), is("qrserver2"));
+        assertThat(service.getPid(), is(-1));
+        assertThat(service.getConfigId(), is("container/qrserver.0"));
+    }
+
+    @Test
+    // TODO: Make it possible to test this without running a HTTP server to create the response
+    public void testMetricsFetching() {
+        VespaService service = VespaService.create("service1", "id", csPort);
+        Metrics metrics = service.getMetrics();
+        assertThat(metrics.getMetric("queries.count").getValue().intValue(), is(28));
+
+        // Shutdown server and check that no metrics are returned (should use empty metrics
+        // when unable to fetch new metrics)
+        shutdown();
+
+        metrics = service.getMetrics();
+        assertThat(metrics.size(), is(0));
+    }
+
+    @After
+    public void shutdown() {
+        this.service.close();
+    }
+
+}