add more tests for PolledMeter

Netflix · Dec 4, 2022 · f9d6074 · f9d6074
1 parent 10dc725
commit f9d6074
Showing 1 changed file with 79 additions and 0 deletions.
diff --git a/spectator-api/src/test/java/com/netflix/spectator/api/patterns/PolledMeterTest.java b/spectator-api/src/test/java/com/netflix/spectator/api/patterns/PolledMeterTest.java
@@ -26,9 +26,15 @@
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
 
+import java.util.ArrayList;
 import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ScheduledFuture;
 import java.util.concurrent.atomic.AtomicLong;
+import java.util.concurrent.atomic.LongAdder;
 
 public class PolledMeterTest {
 
@@ -197,4 +203,77 @@ public void poll() {
     Assertions.assertEquals(1.0, g.value(), 1e-12);
     Assertions.assertEquals(1, c.count());
   }
+
+  @Test
+  public void monitorValueLongAdder() {
+    final Registry r = new DefaultRegistry(Clock.SYSTEM, p -> null);
+    final Id id = r.createId("test");
+
+    final LongAdder v = PolledMeter.using(r)
+            .withName("test")
+            .monitorValue(new LongAdder());
+
+    Assertions.assertEquals(0.0, r.counter(id).actualCount());
+
+    v.increment();
+    PolledMeter.update(r);
+    Assertions.assertEquals(1.0, r.gauge(id).value());
+
+    v.decrement();
+    PolledMeter.update(r);
+    Assertions.assertEquals(0.0, r.gauge(id).value());
+
+    v.add(50);
+    PolledMeter.update(r);
+    Assertions.assertEquals(50.0, r.gauge(id).value());
+  }
+
+  @Test
+  public void monitorSizeOfList() {
+    final Registry r = new DefaultRegistry(Clock.SYSTEM, p -> null);
+    final Id id = r.createId("test");
+
+    final List<String> list = PolledMeter.using(r)
+            .withName("test")
+            .monitorSize(Collections.synchronizedList(new ArrayList<String>()));
+
+    Assertions.assertEquals(0.0, r.counter(id).actualCount());
+
+    list.add("a");
+    PolledMeter.update(r);
+    Assertions.assertEquals(1.0, r.gauge(id).value());
+
+    list.add("b");
+    PolledMeter.update(r);
+    Assertions.assertEquals(2.0, r.gauge(id).value());
+
+    list.clear();
+    PolledMeter.update(r);
+    Assertions.assertEquals(0.0, r.gauge(id).value());
+  }
+
+  @Test
+  public void monitorSizeOfMap() {
+    final Registry r = new DefaultRegistry(Clock.SYSTEM, p -> null);
+    final Id id = r.createId("test");
+
+    final Map<String, String> map = PolledMeter.using(r)
+            .withName("test")
+            .monitorSize(new ConcurrentHashMap<String, String>());
+
+    Assertions.assertEquals(0.0, r.counter(id).actualCount());
+
+    map.put("a", "a-value");
+    PolledMeter.update(r);
+    Assertions.assertEquals(1.0, r.gauge(id).value());
+
+    map.put("b", "b-value");
+    PolledMeter.update(r);
+    Assertions.assertEquals(2.0, r.gauge(id).value());
+
+    map.remove("a");
+    map.remove("b");
+    PolledMeter.update(r);
+    Assertions.assertEquals(0.0, r.gauge(id).value());
+  }
 }