From 0c261deb15aa58c68da772d8fd0c16ec627fe430 Mon Sep 17 00:00:00 2001
From: hexoul <crosien@gmail.com>
Date: Fri, 14 May 2021 13:20:01 +0900
Subject: [PATCH] Fix flaky test

---
 .../armeria/server/healthcheck/HealthCheckServiceTest.java   | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/core/src/test/java/com/linecorp/armeria/server/healthcheck/HealthCheckServiceTest.java b/core/src/test/java/com/linecorp/armeria/server/healthcheck/HealthCheckServiceTest.java
index 3f07b10255b..77866218904 100644
--- a/core/src/test/java/com/linecorp/armeria/server/healthcheck/HealthCheckServiceTest.java
+++ b/core/src/test/java/com/linecorp/armeria/server/healthcheck/HealthCheckServiceTest.java
@@ -317,7 +317,10 @@ void waitWithWrongMethod() throws Exception {
                                   HttpHeaderNames.PREFER, "wait=60",
                                   HttpHeaderNames.IF_NONE_MATCH, "\"healthy\"")).aggregate();
         assertThat(f.get().status()).isEqualTo(HttpStatus.METHOD_NOT_ALLOWED);
-        verify(logger).isDebugEnabled();
+        await().atMost(5, TimeUnit.SECONDS)
+               .untilAsserted(() -> {
+                   assertThatCode(() -> verify(logger).isDebugEnabled()).doesNotThrowAnyException();
+               });
         verifyNoMoreInteractions(logger);
     }