Increase test coverage for HealthCheckManager (#4666)

dropwizard · Jan 30, 2022 · afa7ecd · afa7ecd
1 parent 4c2560b
commit afa7ecd
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/dropwizard-health/src/test/java/io/dropwizard/health/HealthCheckManagerTest.java b/dropwizard-health/src/test/java/io/dropwizard/health/HealthCheckManagerTest.java
@@ -54,6 +54,10 @@ void shouldIgnoreUnconfiguredAddedHealthChecks() {
 
         // then
         verifyNoInteractions(scheduler);
+        assertThat(manager.healthStateView(NAME))
+            .isEmpty();
+        assertThat(manager.healthStateViews())
+            .isEmpty();
     }
 
     @Test
@@ -71,6 +75,10 @@ void shouldScheduleHealthCheckWhenConfiguredHealthCheckAdded() {
 
         // then
         verifyCheckWasScheduled(scheduler, true);
+        assertThat(manager.healthStateViews())
+            .singleElement()
+            .isEqualTo(manager.healthStateView(NAME).orElseThrow(IllegalStateException::new))
+            .satisfies(view -> assertThat(view.getName()).isEqualTo(NAME));
     }
 
     @Test
@@ -86,6 +94,11 @@ void shouldUnscheduleTaskWhenHealthCheckRemoved() {
 
         // then
         verify(scheduler).unschedule(NAME);
+        assertThat(manager.healthStateView(NAME))
+            .isEmpty();
+        assertThat(manager.healthStateViews())
+            .singleElement()
+            .isNull();
     }
 
     @Test