update tests

palantir · Sep 7, 2024 · 9f4405b · 9f4405b
1 parent 63d3bd5
commit 9f4405b
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/...atch/src/test/java/com/palantir/atlasdb/autobatch/AutobatcherTelemetryComponentsTest.java b/...atch/src/test/java/com/palantir/atlasdb/autobatch/AutobatcherTelemetryComponentsTest.java
@@ -116,19 +116,25 @@ private void assertTotalTimeMetricsAreReported(TaggedMetricRegistry registry, Lo
     }
 
     private void assertWaitTimeMetricsAreNotReported(TaggedMetricRegistry registry) {
-        assertThat(getWaitTimeHistogram(registry)).isNull();
+        assertThat(getWaitTimeHistogram(registry)).isNotNull().satisfies(histogram -> assertThat(histogram.getCount())
+                .isZero());
     }
 
     private void assertRunningTimeMetricsAreNotReported(TaggedMetricRegistry registry) {
-        assertThat(getRunningTimeHistogram(registry)).isNull();
+        assertThat(getRunningTimeHistogram(registry))
+                .isNotNull()
+                .satisfies(histogram -> assertThat(histogram.getCount()).isZero());
     }
 
     private void assertWaitTimePercentageMetricsAreNotReported(TaggedMetricRegistry registry) {
-        assertThat(getWaitTimePercentageHistogram(registry)).isNull();
+        assertThat(getWaitTimePercentageHistogram(registry))
+                .isNotNull()
+                .satisfies(histogram -> assertThat(histogram.getCount()).isZero());
     }
 
     private void assertTotalTimeMetricsAreNotReported(TaggedMetricRegistry registry) {
-        assertThat(getTotalTimeHistogram(registry)).isNull();
+        assertThat(getTotalTimeHistogram(registry)).isNotNull().satisfies(histogram -> assertThat(histogram.getCount())
+                .isZero());
     }
 
     private static Histogram getWaitTimeHistogram(TaggedMetricRegistry registry) {