Merge pull request #2258 from ozangunalp/fix_flaky_pulsar_health_check

Fix Pulsar HealthCheckTest
smallrye · Aug 9, 2023 · 3c6e35f · 3c6e35f
2 parents 83bedaa + ff3b04e
commit 3c6e35f
Showing 1 changed file with 16 additions and 12 deletions.
diff --git a/...ng-pulsar/src/test/java/io/smallrye/reactive/messaging/pulsar/health/HealthCheckTest.java b/...ng-pulsar/src/test/java/io/smallrye/reactive/messaging/pulsar/health/HealthCheckTest.java
@@ -109,18 +109,22 @@ public void testHealthOfConsumingApplicationWithFailure() throws PulsarClientExc
 
         await().until(() -> !isAlive());
 
-        HealthReport startup = getHealth().getStartup();
-        HealthReport liveness = getHealth().getLiveness();
-        HealthReport readiness = getHealth().getReadiness();
-
-        assertThat(startup.isOk()).isFalse();
-        assertThat(readiness.isOk()).isFalse();
-        assertThat(liveness.isOk()).isFalse();
-        assertThat(startup.getChannels()).hasSize(1);
-        assertThat(readiness.getChannels()).hasSize(2);
-        assertThat(liveness.getChannels()).hasSize(2);
-        assertThat(liveness.getChannels().get(0).getChannel()).isEqualTo("input");
-        assertThat(liveness.getChannels().get(0).isOk()).isFalse();
+        await().untilAsserted(() -> {
+            HealthReport startup = getHealth().getStartup();
+            HealthReport liveness = getHealth().getLiveness();
+            HealthReport readiness = getHealth().getReadiness();
+
+            assertThat(startup.isOk()).isFalse();
+            assertThat(readiness.isOk()).isFalse();
+            assertThat(liveness.isOk()).isFalse();
+            assertThat(startup.getChannels()).hasSize(1);
+            assertThat(readiness.getChannels()).hasSize(2);
+            assertThat(liveness.getChannels()).hasSize(2);
+
+            assertThat(liveness.getChannels())
+                    .anySatisfy(channelInfo -> assertThat(channelInfo.getChannel()).isEqualTo("input"))
+                    .allSatisfy(channelInfo -> assertThat(channelInfo.isOk()).isFalse());
+        });
     }
 
     private MapBasedConfig configProducingBean() {