CAMEL-19684: reduce flakiness in camel-rocketmq

components/camel-rocketmq/src/test/java/org/apache/camel/component/rocketmq/RocketMQRouteTest.java

@@ -18,10 +18,12 @@
 package org.apache.camel.component.rocketmq;
 
 import java.io.IOException;
+import java.time.Duration;
 
 import org.apache.camel.CamelContext;
 import org.apache.camel.builder.RouteBuilder;
 import org.apache.camel.component.mock.MockEndpoint;
+import org.awaitility.Awaitility;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.BeforeEach;
@@ -46,14 +48,15 @@ static void beforeAll() throws Exception {
     @BeforeEach
     public void setUp() throws Exception {
         super.setUp();
-        resultEndpoint = (MockEndpoint) context.getEndpoint(RESULT_ENDPOINT_URI);
+
     }
 
     @Override
     protected CamelContext createCamelContext() throws Exception {
         CamelContext camelContext = super.createCamelContext();
         RocketMQComponent rocketMQComponent = new RocketMQComponent();
         rocketMQComponent.setNamesrvAddr(rocketMQService.nameserverAddress());
+
         camelContext.addComponent("rocketmq", rocketMQComponent);
         return camelContext;
     }
@@ -71,13 +74,14 @@ public void configure() {
 
     @Test
     public void testSimpleRoute() throws Exception {
+        resultEndpoint = (MockEndpoint) context.getEndpoint(RESULT_ENDPOINT_URI);
         resultEndpoint.expectedBodiesReceived(EXPECTED_MESSAGE);
         resultEndpoint.message(0).header(RocketMQConstants.TOPIC).isEqualTo("START_TOPIC");
         resultEndpoint.message(0).header(RocketMQConstants.TAG).isEqualTo("startTag");
 
         template.sendBody(START_ENDPOINT_URI, EXPECTED_MESSAGE);
 
-        resultEndpoint.assertIsSatisfied();
+        Awaitility.await().atMost(Duration.ofSeconds(5)).untilAsserted(() -> resultEndpoint.assertIsSatisfied());
     }
 
     @AfterAll

test-infra/camel-test-infra-rocketmq/src/test/java/org/apache/camel/test/infra/rocketmq/services/RocketMQContainer.java

@@ -22,7 +22,6 @@
 import org.apache.camel.test.infra.common.services.ContainerService;
 import org.apache.camel.test.infra.rocketmq.common.RocketMQProperties;
 import org.awaitility.Awaitility;
-import org.junit.jupiter.api.extension.ExtensionContext;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.testcontainers.containers.Container;
@@ -37,15 +36,13 @@ public class RocketMQContainer implements RocketMQService, ContainerService<Rock
 
     private final RocketMQNameserverContainer nameserverContainer;
     private final RocketMQBrokerContainer brokerContainer1;
-    private final RocketMQBrokerContainer brokerContainer2;
 
     public RocketMQContainer() {
         Network network = Network.newNetwork();
 
         nameserverContainer = new RocketMQNameserverContainer(network);
 
         brokerContainer1 = new RocketMQBrokerContainer(network, "broker1");
-        brokerContainer2 = new RocketMQBrokerContainer(network, "broker2");
     }
 
     @Override
@@ -64,14 +61,12 @@ public void initialize() {
         LOG.info("Apache RocketMQ running at address {}", nameserverAddress());
 
         brokerContainer1.start();
-        brokerContainer2.start();
     }
 
     @Override
     public void shutdown() {
         nameserverContainer.stop();
         brokerContainer1.stop();
-        brokerContainer2.stop();
     }
 
     public void createTopic(String topic) {