fix: fix the simple param and load balance test

Signed-off-by: wangxye <[email protected]>
AutoMQ · Nov 14, 2023 · 24d030f · 24d030f
1 parent f546ba9
commit 24d030f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/java/e2e-v4/src/test/java/org/apache/rocketmq/cluster/LoadBalancingTest.java b/java/e2e-v4/src/test/java/org/apache/rocketmq/cluster/LoadBalancingTest.java
@@ -117,6 +117,8 @@ public Boolean call() throws Exception {
         consumer3.shutdown();
         consumer4.shutdown();
 
+        VerifyUtils.waitForLoadBalance(topic, consumer1, consumer2);
+
         producer.send(topic, tag, messageSize);
 
         await().atMost(120, SECONDS).until(new Callable<Boolean>() {

diff --git a/java/e2e/src/test/java/org/apache/rocketmq/broker/simple/SimpleParamTest.java b/java/e2e/src/test/java/org/apache/rocketmq/broker/simple/SimpleParamTest.java
@@ -131,7 +131,7 @@ public void test_waitAckException_reReceive_ack() {
                 if (messageViews.size() > 0) {
                     for (MessageView messageView : messageViews) {
                         log.info("MessageId:{}, Body:{}, tag:{}, property:{}", messageView.getMessageId(), StandardCharsets.UTF_8.decode(messageView.getBody()).toString(), messageView.getTag().get(), messageView.getProperties());
-                        TestUtils.waitForSeconds(15);
+                        TestUtils.waitForSeconds(20);
                         consumer.ack(messageView);
                         Assertions.fail("Calling changeInvisibleDuration after ack fails with an INVALID_RECEIPT_HANDLE error");
                     }