add missing unit test for PluginMetrics.

Signed-off-by: Haidong <[email protected]>
opensearch-project · Sep 8, 2023 · 74102d2 · 74102d2
1 parent cd8ce61
commit 74102d2
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/data-prepper-api/src/test/java/org/opensearch/dataprepper/metrics/PluginMetricsTest.java b/data-prepper-api/src/test/java/org/opensearch/dataprepper/metrics/PluginMetricsTest.java
@@ -17,6 +17,7 @@
 import java.util.StringJoiner;
 import java.util.concurrent.atomic.AtomicInteger;
 
+import static java.util.Collections.emptyList;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertThrows;
@@ -125,6 +126,18 @@ public void testReferenceGauge() {
         assertEquals(3, gauge.length());
     }
 
+    @Test
+    public void testReferenceGaugeWithTags() {
+        final String testString = "abc";
+        final String gauge = objectUnderTest.gaugeWithTags("gauge", emptyList(), testString, String::length);
+        assertNotNull(
+                Metrics.globalRegistry.get(new StringJoiner(MetricNames.DELIMITER)
+                        .add(PIPELINE_NAME).add(PLUGIN_NAME)
+                        .add("gauge").toString()).meter());
+        assertEquals(3, gauge.length());
+    }
+
+
     @Test
     public void testEmptyPipelineName() {
         assertThrows(