fix

Author: summaryzb

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),