Add method prepareBlocks in class BlockManagerSuite#MockBlockTransfetService

Author: biaoma-ty

common/network-shuffle/src/main/java/org/apache/spark/network/shuffle/RetryingBlockPreparer.java

@@ -152,7 +152,7 @@ public void onBlockPrepareFailure(Throwable exception) {
                 if (this == currentListener) {
                     initiateRetry();
                 } else {
-                    logger.error(String.format("BM@PrepareBlock failed to send blocks' info, " +
+                    logger.error(String.format("PrepareBlock failed to send blocks' info, " +
                             "and will not retry (%s retries)", retryCount), exception);
                     shouldForwardFailure = true;
                 }

core/src/test/scala/org/apache/spark/storage/BlockManagerSuite.scala

@@ -39,7 +39,7 @@ import org.apache.spark.memory.UnifiedMemoryManager
 import org.apache.spark.network.{BlockDataManager, BlockTransferService}
 import org.apache.spark.network.buffer.{ManagedBuffer, NioManagedBuffer}
 import org.apache.spark.network.netty.NettyBlockTransferService
-import org.apache.spark.network.shuffle.BlockFetchingListener
+import org.apache.spark.network.shuffle.{BlockFetchingListener, BlockPreparingListener}
 import org.apache.spark.rpc.RpcEnv
 import org.apache.spark.scheduler.LiveListenerBus
 import org.apache.spark.serializer.{JavaSerializer, KryoSerializer, SerializerManager}
@@ -1241,6 +1241,16 @@ class BlockManagerSuite extends SparkFunSuite with Matchers with BeforeAndAfterE
       Future {}
     }
 
+    override def prepareBlocks(
+        host: String,
+        port: Int,
+        execId: String,
+        prepareBlockIds: Array[String],
+        releaseBlocks: Array[String],
+        listener: BlockPreparingListener): Unit = {
+      listener.onBlockPrepareSuccess("mockBlockId", new NioManagedBuffer(ByteBuffer.allocate(1)))
+    }
+
     override def fetchBlockSync(
         host: String,
         port: Int,