Remove method that was only used in tests.

2 files changed, 7 insertions, 5 deletions

diff --git a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/metric/MetricsTest.java b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/metric/MetricsTest.java
index b9e6377c27b..cce88a5bd53 100644
--- a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/metric/MetricsTest.java
+++ b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/metric/MetricsTest.java
@@ -12,6 +12,7 @@ import java.util.HashMap;
 import java.util.Map;
 
 import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertThat;
 
 /**
@@ -46,7 +47,8 @@ public class MetricsTest {
     public void testBasicMetric() {
         Metrics m = new Metrics();
         m.add(new Metric("count", 1, System.currentTimeMillis() / 1000));
-        assertThat(m.get("count").intValue(), is(1));
+        assertThat(m.getMetrics().size(), is(1));
+        assertThat(m.getMetrics().get(0).getName(), is("count"));
     }
 
     @Test
diff --git a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/service/MetricsFetcherTest.java b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/service/MetricsFetcherTest.java
index 278b2a3143a..ce33ce9f7bf 100644
--- a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/service/MetricsFetcherTest.java
+++ b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/service/MetricsFetcherTest.java
@@ -22,10 +22,10 @@ public class MetricsFetcherTest {
         String jsonData = TestUtil.getFileContents("metrics-state.json");
         RemoteMetricsFetcher fetcher = new RemoteMetricsFetcher(new DummyService(0, "dummy/id/0"), port);
         Metrics metrics = fetcher.createMetrics(jsonData, 0);
-        assertThat("Wrong number of metrics", metrics.size(), is(10));
-        assertThat("Wrong value for metric", metrics.get("query_hits.count").intValue(), is(28));
-        assertThat("Wrong value for metric ", metrics.get("queries.rate").doubleValue(), is(0.4667));
-        assertThat("Wrong timestamp", metrics.getTimeStamp(), is(1334134700L));
+        assertThat(metrics.size(), is(10));
+        assertThat(metrics.getMetric("query_hits.count").getValue().intValue(), is(28));
+        assertThat(metrics.getMetric("queries.rate").getValue().doubleValue(), is(0.4667));
+        assertThat(metrics.getTimeStamp(), is(1334134700L));
     }
 
     @Test