Intermittent SubscriberPublMsgToMsgRetComplBean test fix

Signed-off-by: Daniel Kec <[email protected]>

microprofile/messaging/core/src/test/java/io/helidon/microprofile/messaging/inner/subscriber/SubscriberPublMsgToMsgRetComplBean.java

@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2020, 2023 Oracle and/or its affiliates.
+ * Copyright (c) 2020 Oracle and/or its affiliates.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -19,6 +19,7 @@
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.CompletionStage;
 import java.util.concurrent.CopyOnWriteArraySet;
+import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.ExecutorService;
 
 import javax.enterprise.context.ApplicationScoped;
@@ -42,6 +43,7 @@
 public class SubscriberPublMsgToMsgRetComplBean implements AssertableTestBean, AsyncTestBean {
 
     CopyOnWriteArraySet<String> resultData = new CopyOnWriteArraySet<>();
+    private final CountDownLatch countDownLatch = new CountDownLatch(TEST_DATA.size());
 
     private final ExecutorService executor = createExecutor();
 
@@ -54,11 +56,15 @@ public Publisher<Message<String>> sourceForCsVoidMessage() {
 
     @Incoming("cs-void-message")
     public CompletionStage<Void> consumeMessageAndReturnCompletionStageOfVoid(Message<String> message) {
-        return CompletableFuture.runAsync(() -> resultData.add(message.getPayload()), executor);
+        return CompletableFuture.runAsync(() -> {
+            resultData.add(message.getPayload());
+            countDownLatch.countDown();
+        }, executor);
     }
 
     @Override
     public void assertValid() {
+        await("Messages not delivered in time!", countDownLatch);
         assertWithOrigin("Result doesn't match", resultData, is(TEST_DATA));
     }