fix

apache · Mar 7, 2025 · 26b3240 · 26b3240
1 parent 32228d8
commit 26b3240
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/integration-test/common/src/test/java/org/apache/uniffle/test/RpcClientRetryTest.java b/integration-test/common/src/test/java/org/apache/uniffle/test/RpcClientRetryTest.java
@@ -132,7 +132,6 @@ public void testCancelGrpc() throws InterruptedException {
     assertEquals(2, result.getSuccessBlockIds().size());
 
     enableFirstNSendDataRequestsToFail(2);
-    System.out.println("1====================");
     CompletableFuture<SendShuffleDataResult> future =
         CompletableFuture.supplyAsync(
             () -> shuffleWriteClientImpl.sendShuffleData(testAppId, blocks, needCancelRequest));
@@ -225,8 +224,7 @@ public void testRpcRetryLogic(StorageType storageType) {
   }
 
   private static void enableFirstNReadRequestsToFail(int failedCount) {
-    Lists.newArrayList(grpcShuffleServers, nettyShuffleServers).stream()
-        .flatMap(List::stream)
+    grpcShuffleServers.stream()
         .forEach(
             server ->
                 ((MockedGrpcServer) server.getServer())
@@ -235,8 +233,7 @@ private static void enableFirstNReadRequestsToFail(int failedCount) {
   }
 
   private static void enableFirstNSendDataRequestsToFail(int failedCount) {
-    Lists.newArrayList(grpcShuffleServers, nettyShuffleServers).stream()
-        .flatMap(List::stream)
+    grpcShuffleServers.stream()
         .forEach(
             server ->
                 ((MockedGrpcServer) server.getServer())
@@ -245,8 +242,7 @@ private static void enableFirstNSendDataRequestsToFail(int failedCount) {
   }
 
   private static void disableFirstNReadRequestsToFail() {
-    Lists.newArrayList(grpcShuffleServers, nettyShuffleServers).stream()
-        .flatMap(List::stream)
+    grpcShuffleServers.stream()
         .forEach(
             server ->
                 ((MockedGrpcServer) server.getServer())
@@ -285,8 +281,6 @@ private void registerShuffleServer(
                 .unregisterTimeSec(10)
                 .unregisterRequestTimeSec(10));
 
-    System.out.println("============");
-    System.out.println(grpcShuffleServerInfoList);
     for (int i = 0; i < replica; i++) {
       shuffleWriteClientImpl.registerShuffle(
           grpcShuffleServerInfoList.get(i),