fix: forget another node test

sofastack · Jun 17, 2024 · 3ed6dcc · 3ed6dcc
1 parent e8cc1c1
commit 3ed6dcc
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 9 deletions.
diff --git a/jraft-core/src/test/java/com/alipay/sofa/jraft/core/NodeTest.java b/jraft-core/src/test/java/com/alipay/sofa/jraft/core/NodeTest.java
@@ -30,6 +30,7 @@
 import java.util.concurrent.Future;
 import java.util.concurrent.ThreadLocalRandom;
 import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
 import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicInteger;
 
@@ -3518,9 +3519,13 @@ public void testChangePeersChaosApplyTasks() throws Exception {
         for (final ChangeArg arg : args) {
             arg.stop = true;
         }
-        for (final Future<?> future : futures) {
-            future.get();
-        }
+		for (final Future<?> future : futures) {
+			try {
+				future.get(20, TimeUnit.SECONDS);
+			} catch (TimeoutException e) {
+				// ignore
+			}
+		}
 
         cluster.waitLeader();
         final SynchronizedClosure done = new SynchronizedClosure();

diff --git a/jraft-extension/java-log-storage-impl/src/test/java/com/alipay/sofa/jraft/core/NodeTest.java b/jraft-extension/java-log-storage-impl/src/test/java/com/alipay/sofa/jraft/core/NodeTest.java
@@ -3407,13 +3407,13 @@ public void testChangePeersChaosApplyTasks() throws Exception {
         for (final ChangeArg arg : args) {
             arg.stop = true;
         }
-	for (final Future<?> future : futures) {
-		try {
-			future.get(20, TimeUnit.SECONDS);
-		} catch (TimeoutException e) {
-			// ignore
+		for (final Future<?> future : futures) {
+			try {
+				future.get(20, TimeUnit.SECONDS);
+			} catch (TimeoutException e) {
+				// ignore
+			}
 		}
-	}
 
         cluster.waitLeader();
         final SynchronizedClosure done = new SynchronizedClosure();