From 1d40d597aaeeb233c385713a2fa6e468687397d1 Mon Sep 17 00:00:00 2001 From: Min Shen Date: Sun, 25 Jul 2021 23:51:56 -0700 Subject: [PATCH] Rebase with master --- .../org/apache/spark/shuffle/ShuffleBlockPusherSuite.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/core/src/test/scala/org/apache/spark/shuffle/ShuffleBlockPusherSuite.scala b/core/src/test/scala/org/apache/spark/shuffle/ShuffleBlockPusherSuite.scala index 0baee6879b6bc..2800be1cb9e20 100644 --- a/core/src/test/scala/org/apache/spark/shuffle/ShuffleBlockPusherSuite.scala +++ b/core/src/test/scala/org/apache/spark/shuffle/ShuffleBlockPusherSuite.scala @@ -332,9 +332,9 @@ class ShuffleBlockPusherSuite extends SparkFunSuite with BeforeAndAfterEach { when(shuffleClient.pushBlocks(any(), any(), any(), any(), any())) .thenAnswer((invocation: InvocationOnMock) => { val pushedBlocks = invocation.getArguments()(2).asInstanceOf[Array[String]] - val blockFetchListener = invocation.getArguments()(4).asInstanceOf[BlockFetchingListener] + val blockPushListener = invocation.getArguments()(4).asInstanceOf[BlockPushingListener] pushedBlocks.foreach(blockId => { - blockFetchListener.onBlockFetchFailure( + blockPushListener.onBlockPushFailure( blockId, new IOException("Failed to send RPC", new FileNotFoundException("file not found"))) })