Move MockMetric to jdisc_core test package

2 files changed, 1 insertions, 55 deletions

diff --git a/docproc/src/main/java/com/yahoo/docproc/jdisc/metric/MockMetric.java b/docproc/src/main/java/com/yahoo/docproc/jdisc/metric/MockMetric.java
deleted file mode 100644
index a3cdd7d3394..00000000000
--- a/docproc/src/main/java/com/yahoo/docproc/jdisc/metric/MockMetric.java
+++ /dev/null
@@ -1,53 +0,0 @@
-// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
-package com.yahoo.docproc.jdisc.metric;
-
-import com.yahoo.jdisc.Metric;
-
-import java.util.Map;
-import java.util.concurrent.ConcurrentHashMap;
-
-/**
- * Metric implementation for tests.
- *
- * @author jonmv
- */
-public class MockMetric implements Metric {
-
-    private final Map<String, Map<Map<String, ?>, Double>> metrics = new ConcurrentHashMap<>();
-
-    @Override
-    public void set(String key, Number val, Context ctx) {
-        metrics.computeIfAbsent(key, k -> new ConcurrentHashMap<>())
-               .put(MapContext.emptyIfNull(ctx).dimensions, val.doubleValue());
-    }
-
-    @Override
-    public void add(String key, Number val, Context ctx) {
-        metrics.computeIfAbsent(key, k -> new ConcurrentHashMap<>())
-               .merge(MapContext.emptyIfNull(ctx).dimensions, val.doubleValue(), Double::sum);
-    }
-
-    @Override
-    public Context createContext(Map<String, ?> properties) {
-        return properties == null ? MapContext.empty : new MapContext(properties);
-    }
-
-    public Map<String, Map<Map<String, ?>, Double>> metrics() { return metrics; }
-
-    private static class MapContext implements Context {
-
-        private static final MapContext empty = new MapContext(Map.of());
-
-        private final Map<String, ?> dimensions;
-
-        private MapContext(Map<String, ?> dimensions) {
-            this.dimensions = dimensions;
-        }
-
-        private static MapContext emptyIfNull(Context context) {
-            return context == null ? empty : (MapContext) context;
-        }
-
-    }
-
-}
diff --git a/docproc/src/test/java/com/yahoo/docproc/SimpleDocumentProcessorTestCase.java b/docproc/src/test/java/com/yahoo/docproc/SimpleDocumentProcessorTestCase.java
index c5fb4b75cf3..660feb49a91 100644
--- a/docproc/src/test/java/com/yahoo/docproc/SimpleDocumentProcessorTestCase.java
+++ b/docproc/src/test/java/com/yahoo/docproc/SimpleDocumentProcessorTestCase.java
@@ -2,8 +2,7 @@
 package com.yahoo.docproc;
 
 import com.yahoo.container.StatisticsConfig;
-import com.yahoo.docproc.jdisc.metric.MockMetric;
-import com.yahoo.docproc.jdisc.metric.NullMetric;
+import com.yahoo.jdisc.test.MockMetric;
 import com.yahoo.document.DataType;
 import com.yahoo.document.DocumentId;
 import com.yahoo.document.DocumentOperation;