make available map for a metric name (for unit testing)

1 files changed, 10 insertions, 0 deletions

diff --git a/simplemetrics/src/main/java/com/yahoo/metrics/simple/Bucket.java b/simplemetrics/src/main/java/com/yahoo/metrics/simple/Bucket.java
index 4ae284c4304..45393ff5f77 100644
--- a/simplemetrics/src/main/java/com/yahoo/metrics/simple/Bucket.java
+++ b/simplemetrics/src/main/java/com/yahoo/metrics/simple/Bucket.java
@@ -146,6 +146,16 @@ public class Bucket {
         return singleMetric;
     }
 
+    public Map<Point, UntypedMetric> getMapForMetric(@NonNull String metricName) {
+        Map<Point, UntypedMetric> result = new HashMap<>();
+        for (Map.Entry<Identifier, UntypedMetric> entry : values.entrySet()) {
+            if (metricName.equals(entry.getKey().getName())) {
+                result.put(entry.getKey().getLocation(), entry.getValue());
+            }
+        }
+        return result;
+    }
+
     public Map<String, List<Map.Entry<Point, UntypedMetric>>> getValuesByMetricName() {
         Map<String, List<Map.Entry<Point, UntypedMetric>>> result = new HashMap<>();
         for (Map.Entry<Identifier, UntypedMetric> entry : values.entrySet()) {